auto rights regit

src/main/java/com/zglc/fm/service/RightsRegitService.java

@@ -0,0 +1,69 @@
+package com.zglc.fm.service;
+
+import com.zglc.fm.entity.RightsEntity;
+import io.swagger.models.HttpMethod;
+import io.swagger.models.Operation;
+import io.swagger.models.Swagger;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.ApplicationArguments;
+import org.springframework.boot.ApplicationRunner;
+import org.springframework.stereotype.Component;
+import springfox.documentation.service.Documentation;
+import springfox.documentation.spring.web.DocumentationCache;
+import springfox.documentation.swagger2.mappers.ServiceModelToSwagger2Mapper;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+
+@Component
+public class RightsRegitService implements ApplicationRunner {
+    private RightsService rightsService;
+    private DocumentationCache documentationCache;
+    private ServiceModelToSwagger2Mapper serviceModelToSwagger2Mapper;
+
+    private static Log log = LogFactory.getLog(RightsRegitService.class);
+
+    @Autowired
+    public RightsRegitService(RightsService rightsService, DocumentationCache documentationCache, ServiceModelToSwagger2Mapper serviceModelToSwagger2Mapper)
+    {
+       this.rightsService = rightsService;
+       this.documentationCache = documentationCache;
+       this.serviceModelToSwagger2Mapper = serviceModelToSwagger2Mapper;
+    }
+
+    @Override
+    public void run(ApplicationArguments args) throws Exception {
+        String groupName = "default";
+        Map<String, String> result = new HashMap();
+        Documentation documentation = this.documentationCache.documentationByGroup(groupName);
+        if (documentation != null) {
+            Swagger swagger = this.serviceModelToSwagger2Mapper.mapDocumentation(documentation);
+            swagger.getPaths().forEach((s, path) -> {
+                Map<HttpMethod, Operation> operationMap = path.getOperationMap();
+                operationMap.forEach((httpMethod, operation) -> {
+                    String var10000 = (String)result.put(s,operation.getTags().get(0)+ "," + operation.getSummary());
+                });
+            });
+            Iterator var13 = result.entrySet().iterator();
+
+            while(var13.hasNext()) {
+                Map.Entry<String, String> entry = (Map.Entry)var13.next();
+                String apiPath = (String)entry.getKey();
+                String apiSummary = (String)entry.getValue();
+                RightsEntity rightsEntity =  (this.rightsService.getRightsByPath(apiPath));
+                if (rightsEntity == null) {
+                    rightsEntity = new RightsEntity();
+                    rightsEntity.setRightsPath(apiPath);
+                    rightsEntity.setRightsName(apiSummary);
+                    rightsEntity.setRemark(apiSummary);
+                    rightsEntity.setId(-1);
+                    rightsService.add(rightsEntity);
+                    log.info("add rights :" + rightsEntity.getRightsPath()+","+rightsEntity.getRightsName());
+                }
+            }
+        }
+    }
+}