TRX 1 год назад
Родитель
Сommit
2aed861d7b

+ 4 - 2
OneCardIotServer/src/main/java/com/zhongshu/iot/server/core/controller/common/CommonController.java

@@ -3,6 +3,8 @@ package com.zhongshu.iot.server.core.controller.common;
 import com.github.microservice.auth.security.annotations.ResourceAuth;
 import com.github.microservice.auth.security.annotations.ResourceAuth;
 import com.github.microservice.auth.security.type.AuthType;
 import com.github.microservice.auth.security.type.AuthType;
 import com.github.microservice.net.ResultContent;
 import com.github.microservice.net.ResultContent;
+import com.github.microservice.types.deviceUse.DeviceCategory;
+import com.github.microservice.types.deviceUse.DeviceType;
 import com.zhongshu.iot.server.core.service.base.CommonService;
 import com.zhongshu.iot.server.core.service.base.CommonService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import io.swagger.v3.oas.annotations.tags.Tag;
@@ -31,13 +33,13 @@ public class CommonController {
     @Operation(summary = "设备类型 选择项")
     @Operation(summary = "设备类型 选择项")
     @RequestMapping(value = "getDeviceType", method = {RequestMethod.GET})
     @RequestMapping(value = "getDeviceType", method = {RequestMethod.GET})
     public ResultContent getDeviceType() {
     public ResultContent getDeviceType() {
-        return ResultContent.buildSuccess(commonService.getEnumListModel());
+        return ResultContent.buildSuccess(commonService.getEnums(DeviceType.values()));
     }
     }
 
 
     @ResourceAuth(value = "user", type = AuthType.User)
     @ResourceAuth(value = "user", type = AuthType.User)
     @Operation(summary = "设备节点 选择项")
     @Operation(summary = "设备节点 选择项")
     @RequestMapping(value = "getDeviceCategory", method = {RequestMethod.GET})
     @RequestMapping(value = "getDeviceCategory", method = {RequestMethod.GET})
     public ResultContent getDeviceCategory() {
     public ResultContent getDeviceCategory() {
-        return ResultContent.buildSuccess(commonService.getDeviceCategory());
+        return ResultContent.buildSuccess(commonService.getEnums(DeviceCategory.values()));
     }
     }
 }
 }

+ 11 - 13
OneCardIotServer/src/main/java/com/zhongshu/iot/server/core/service/base/CommonService.java

@@ -1,8 +1,7 @@
 package com.zhongshu.iot.server.core.service.base;
 package com.zhongshu.iot.server.core.service.base;
 
 
 import com.github.microservice.models.common.EnumListModel;
 import com.github.microservice.models.common.EnumListModel;
-import com.github.microservice.types.deviceUse.DeviceCategory;
-import com.github.microservice.types.deviceUse.DeviceType;
+import com.github.microservice.types.common.CommonEnum;
 import com.mongodb.client.result.UpdateResult;
 import com.mongodb.client.result.UpdateResult;
 import com.zhongshu.iot.server.core.util.CommonUtil;
 import com.zhongshu.iot.server.core.util.CommonUtil;
 import lombok.extern.slf4j.Slf4j;
 import lombok.extern.slf4j.Slf4j;
@@ -28,18 +27,17 @@ public class CommonService {
     @Autowired
     @Autowired
     private MongoTemplate mongoTemplate;
     private MongoTemplate mongoTemplate;
 
 
-    public EnumListModel getEnumListModel() {
+    public EnumListModel getEnums(Enum[] enums) {
         EnumListModel model = new EnumListModel();
         EnumListModel model = new EnumListModel();
-        for (DeviceType deviceType1 : DeviceType.values()) {
-            model.add(deviceType1.getRemark(), deviceType1.name());
-        }
-        return model;
-    }
-
-    public EnumListModel getDeviceCategory() {
-        EnumListModel model = new EnumListModel();
-        for (DeviceCategory deviceType1 : DeviceCategory.values()) {
-            model.add(deviceType1.getRemark(), deviceType1.name());
+        if (enums != null) {
+            for (Enum _enum : enums) {
+                String value = _enum.name();
+                String name = "";
+                if (_enum instanceof CommonEnum) {
+                    name = ((CommonEnum) _enum).getRemark();
+                }
+                model.add(name, value);
+            }
         }
         }
         return model;
         return model;
     }
     }