lts admin: 完成 WEB 管理后台 节点的subscribe ，展示 搜索

Author: 胡贵

lts-admin/pom.xml

@@ -82,6 +82,13 @@
             <version>3.1.0</version>
             <scope>provided</scope>
         </dependency>
+
+        <dependency>
+            <groupId>javax.servlet</groupId>
+            <artifactId>jstl</artifactId>
+            <version>1.2</version>
+        </dependency>
+
         <!-- Just the annotations; use this dependency if you want to attach annotations
              to classes without connecting them to the code. -->
         <dependency>
@@ -120,5 +127,13 @@
             <artifactId>jackson-core</artifactId>
             <version>${jackson-2-version}</version>
         </dependency>
+        <dependency>
+            <groupId>commons-dbutils</groupId>
+            <artifactId>commons-dbutils</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.hsqldb</groupId>
+            <artifactId>hsqldb</artifactId>
+        </dependency>
     </dependencies>
 </project>

lts-admin/src/main/java/com/lts/job/web/cluster/RegistryService.java

@@ -4,38 +4,39 @@
 import com.lts.job.core.cluster.Config;
 import com.lts.job.core.cluster.Node;
 import com.lts.job.core.cluster.NodeType;
-import com.lts.job.core.cluster.SubscribedNodeManager;
 import com.lts.job.core.registry.NotifyEvent;
 import com.lts.job.core.registry.NotifyListener;
 import com.lts.job.core.registry.Registry;
 import com.lts.job.core.registry.RegistryFactory;
 import com.lts.job.core.util.CollectionUtils;
-import com.lts.job.core.util.Holder;
+import com.lts.job.core.util.ConcurrentHashSet;
 import com.lts.job.core.util.StringUtils;
+import com.lts.job.web.request.NodeRequest;
 import com.lts.job.web.support.AppConfigurer;
+import com.lts.job.web.support.db.NodeMemDB;
 import org.springframework.stereotype.Component;
 
 import java.util.ArrayList;
-import java.util.Collection;
 import java.util.List;
-import java.util.concurrent.ConcurrentHashMap;
 
 /**
  * Robert HG (254963746@qq.com) on 6/5/15.
  */
 @Component
 public class RegistryService {
 
-    private final ConcurrentHashMap<String/*clusterName*/, SubscribedNodeManager> NODE_MANAGER_MAP;
+    private final ConcurrentHashSet<String/*clusterName*/> map;
+
+    private NodeMemDB nodeMemDB;
 
     public RegistryService() {
-        this.NODE_MANAGER_MAP = new ConcurrentHashMap<String, SubscribedNodeManager>();
+        this.map = new ConcurrentHashSet<String>();
+        this.nodeMemDB = new NodeMemDB();
     }
 
     public synchronized void register(String clusterName) {
-        SubscribedNodeManager subscribedNodeManager = NODE_MANAGER_MAP.get(clusterName);
-        if (subscribedNodeManager != null) {
-            //  already registered
+
+        if (map.contains(clusterName)) {
             return;
         }
 
@@ -44,7 +45,6 @@ public synchronized void register(String clusterName) {
         node.addListenNodeType(NodeType.JOB_CLIENT);
         node.addListenNodeType(NodeType.TASK_TRACKER);
         node.addListenNodeType(NodeType.JOB_TRACKER);
-        node.setNodeType(NodeType.JOB_ADMIN);
 
         Config config = new Config();
         config.setIdentity(node.getIdentity());
@@ -54,13 +54,9 @@ public synchronized void register(String clusterName) {
 
         Application application = new AdminApplication();
         application.setConfig(config);
-        subscribedNodeManager = new SubscribedNodeManager(application);
 
         Registry registry = RegistryFactory.getRegistry(config);
 
-        final Holder<SubscribedNodeManager> subscribedNodeManagerHolder =
-                new Holder<SubscribedNodeManager>(subscribedNodeManager);
-
         registry.subscribe(node, new NotifyListener() {
             @Override
             public void notify(NotifyEvent event, List<Node> nodes) {
@@ -69,37 +65,23 @@ public void notify(NotifyEvent event, List<Node> nodes) {
                 }
                 switch (event) {
                     case ADD:
-                        subscribedNodeManagerHolder.get().addNodes(nodes);
+                        nodeMemDB.addNode(nodes);
                         break;
                     case REMOVE:
-                        subscribedNodeManagerHolder.get().removeNodes(nodes);
+                        nodeMemDB.removeNode(nodes);
                         break;
                 }
             }
         });
 
-        NODE_MANAGER_MAP.put(clusterName, subscribedNodeManagerHolder.get());
+        map.add(clusterName);
     }
 
-    public List<Node> getAllNodes(String clusterName) {
-        SubscribedNodeManager subscribedNodeManager = NODE_MANAGER_MAP.get(clusterName);
-        if (subscribedNodeManager == null) {
-            return new ArrayList<Node>(0);
-        }
-        return subscribedNodeManager.getNodeList();
+    public List<String> getAllClusterNames() {
+        return new ArrayList<String>(map.list());
     }
 
-    public List<Node> getAllNodes() {
-        List<Node> result = new ArrayList<Node>();
-        Collection<SubscribedNodeManager> subscribedNodeManagers = NODE_MANAGER_MAP.values();
-        if (CollectionUtils.isNotEmpty(subscribedNodeManagers)) {
-            for (SubscribedNodeManager subscribedNodeManager : subscribedNodeManagers) {
-                List<Node> nodes = subscribedNodeManager.getNodeList();
-                if (CollectionUtils.isNotEmpty(nodes)) {
-                    result.addAll(nodes);
-                }
-            }
-        }
-        return result;
+    public List<Node> getNodes(NodeRequest request) {
+        return nodeMemDB.search(request);
     }
 }

lts-admin/src/main/java/com/lts/job/web/controller/AbstractController.java

@@ -1,9 +1,22 @@
 package com.lts.job.web.controller;
 
+import com.lts.job.web.support.DateEditor;
+import org.springframework.web.bind.ServletRequestDataBinder;
+import org.springframework.web.bind.annotation.InitBinder;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Date;
+
 /**
  * @author Robert HG (254963746@qq.com) on 5/9/15.
  */
 public class AbstractController {
 
+    @InitBinder
+    protected void initBinder(HttpServletRequest request,
+                              ServletRequestDataBinder binder) throws Exception {
+        //对于需要转换为Date类型的属性，使用DateEditor进行处理
+        binder.registerCustomEditor(Date.class, new DateEditor());
+    }
 
 }

lts-admin/src/main/java/com/lts/job/web/controller/NodeController.java

@@ -0,0 +1,63 @@
+package com.lts.job.web.controller.api;
+
+import com.lts.job.core.cluster.Node;
+import com.lts.job.core.util.CollectionUtils;
+import com.lts.job.core.util.StringUtils;
+import com.lts.job.web.cluster.RegistryService;
+import com.lts.job.web.controller.AbstractController;
+import com.lts.job.web.request.NodeRequest;
+import com.lts.job.web.vo.RestfulResponse;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
+
+/**
+ * @author Robert HG (254963746@qq.com) on 5/11/15.
+ */
+@RestController
+@RequestMapping("/node")
+public class NodeApiController extends AbstractController {
+
+    @Autowired
+    RegistryService registryService;
+
+    @RequestMapping("node-list-get")
+    public RestfulResponse getNodeList(NodeRequest request) {
+        RestfulResponse response = new RestfulResponse();
+
+        List<Node> nodes = registryService.getNodes(request);
+
+        response.setSuccess(true);
+        response.setResults(CollectionUtils.sizeOf(nodes));
+        response.setRows(nodes);
+
+        return response;
+    }
+
+    @RequestMapping("node-cluster-register")
+    public RestfulResponse registerClusterNameSub(NodeRequest request) {
+        RestfulResponse response = new RestfulResponse();
+
+        if (StringUtils.isEmpty(request.getClusterName())) {
+            response.setSuccess(false);
+            response.setMsg("clusterName can not be null");
+            return response;
+        }
+        registryService.register(request.getClusterName());
+        response.setSuccess(true);
+        return response;
+    }
+
+    @RequestMapping("node-cluster-register-get")
+    public RestfulResponse getAllRegisterClusterName() {
+        RestfulResponse response = new RestfulResponse();
+        List<String> clusterNames = registryService.getAllClusterNames();
+        response.setSuccess(true);
+        response.setResults(CollectionUtils.sizeOf(clusterNames));
+        response.setRows(clusterNames);
+        return response;
+    }
+
+}

lts-admin/src/main/java/com/lts/job/web/controller/api/NodeApiController.java

@@ -0,0 +1,27 @@
+package com.lts.job.web.controller.ui;
+
+import com.lts.job.web.cluster.RegistryService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.ui.Model;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import java.util.List;
+
+/**
+ * @author Robert HG (254963746@qq.com) on 6/6/15.
+ */
+@Controller
+public class UIController {
+
+    @Autowired
+    RegistryService registryService;
+
+    @RequestMapping("node/node-manager")
+    public String nodeManagerUI(Model model) {
+        List<String> clusterNames = registryService.getAllClusterNames();
+        model.addAttribute("clusterNames", clusterNames);
+        return "node-manager";
+    }
+
+}

lts-admin/src/main/java/com/lts/job/web/controller/ui/UIController.java

@@ -0,0 +1,84 @@
+package com.lts.job.web.request;
+
+import com.lts.job.core.cluster.NodeType;
+
+import java.util.Date;
+
+/**
+ * @author Robert HG (254963746@qq.com) on 6/6/15.
+ */
+public class NodeRequest extends PageRequest {
+
+    private String clusterName;
+    private String identity;
+    private String ip;
+    private String nodeGroup;
+    private NodeType nodeType;
+    private Boolean available;
+    private Date startDate;
+    private Date endDate;
+
+    public String getNodeGroup() {
+        return nodeGroup;
+    }
+
+    public void setNodeGroup(String nodeGroup) {
+        this.nodeGroup = nodeGroup;
+    }
+
+    public String getClusterName() {
+        return clusterName;
+    }
+
+    public void setClusterName(String clusterName) {
+        this.clusterName = clusterName;
+    }
+
+    public String getIdentity() {
+        return identity;
+    }
+
+    public void setIdentity(String identity) {
+        this.identity = identity;
+    }
+
+    public String getIp() {
+        return ip;
+    }
+
+    public void setIp(String ip) {
+        this.ip = ip;
+    }
+
+    public NodeType getNodeType() {
+        return nodeType;
+    }
+
+    public void setNodeType(NodeType nodeType) {
+        this.nodeType = nodeType;
+    }
+
+    public Boolean getAvailable() {
+        return available;
+    }
+
+    public void setAvailable(Boolean available) {
+        this.available = available;
+    }
+
+    public Date getStartDate() {
+        return startDate;
+    }
+
+    public void setStartDate(Date startDate) {
+        this.startDate = startDate;
+    }
+
+    public Date getEndDate() {
+        return endDate;
+    }
+
+    public void setEndDate(Date endDate) {
+        this.endDate = endDate;
+    }
+}