update

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(getLevel(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);
-            }
             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;
+    }
 }