diff --git a/jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/mini/service/MiniUserService.java b/jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/mini/service/MiniUserService.java index 3579321..07877be 100644 --- a/jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/mini/service/MiniUserService.java +++ b/jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/mini/service/MiniUserService.java @@ -65,6 +65,7 @@ public class MiniUserService { skuCode =yxVipCard.getSkuCode(); validTime = yxVipCard.getValidDate(); activationTime = yxVipCard.getUsedTime(); + vipDTO.setVipLevel(getLevel(skuCode)); } } @@ -91,17 +92,7 @@ public class MiniUserService { //获取到了vip信息 if (StringUtils.isNotBlank(skuCode)) { - if ("1001".equals(skuCode)) { - vipDTO.setVipLevel(1); - }else if("1002".equals(skuCode)){ - vipDTO.setVipLevel(2); - }else if("1003".equals(skuCode)){ - vipDTO.setVipLevel(3); - }else if("1004".equals(skuCode)){ - vipDTO.setVipLevel(4); - }else if("9999".equals(skuCode)){ - vipDTO.setVipLevel(9); - } + vipDTO.setVipLevel(getLevel(skuCode)); YxVipSku vipSku = yxVipSkuService.getBySkuCode(skuCode); if (null!=vipSku) { vipDTO.setSkuName(vipSku.getSkuName()); @@ -114,4 +105,19 @@ public class MiniUserService { } return vipDTO; } + + public static int getLevel(String skuCode) { + if ("1001".equals(skuCode)) { + return 1; + }else if("1002".equals(skuCode)){ + return 2; + }else if("1003".equals(skuCode)){ + return 3; + }else if("1004".equals(skuCode)){ + return 4; + }else if("9999".equals(skuCode)){ + return 5; + } + return 0; + } }