risk

王水玲
Commit 8f9b2e755e3c56b3c43048e8ccb6dbf0581a6723 1 parent 4986d8e1
Showing 5 changed files with 152 additions and 61 deletions
apps/web/actions/risk_management.js
apps/web/models/zookeeperModel.js
apps/web/routers.js
apps/web/views/action/add_risk.hbs
apps/web/views/action/risk_management.hbs
--- a/apps/web/actions/risk_management.js
View file @8f9b2e7
+++ b/apps/web/actions/risk_management.js
View file @8f9b2e7
@@ -4,72 +4,121 @@ const Router = require('koa-router');
 const ApiCache = require('../../ci/api_cache');
 const _ = require('lodash');
-const getter = require('../../zookeeper/getter');
-const setter = require('../../zookeeper/setter');
-const tester = require('../../zookeeper/tester');
+const ZookeeperModel = require('../models/zookeeperModel');
+let zookeeperModel = new ZookeeperModel();
 let r = new Router();
+var process = {
+    TO_VALIDATE_CODE: '跳转全局验证码解锁',
+    REDIRECT_403: '返回403'
+};
 const risk = {
     index: async(ctx, next) => {
-        // const regexp = /pc:limiter:faker:(.*)/;
-        // const threshold = ctx.request.query.threshold || 100;
-        // const limit = ctx.request.query.limit || 10;
-
-        // let hosts = await MemcachedHost.findAll();
-
-        // const selectedHosts = _.filter(hosts, host => {
-        //     const isCurrent = host.host === ctx.request.query.node;
-        //     if (isCurrent) {
-        //         host.isCurrent = true;
-        //     }
-        //     return isCurrent;
-        // });
-
-        // let results = await Promise.all(_.map(selectedHosts, (h) => {
-        //     return (new ApiCache(h.host)).find((key) => {
-        //         return regexp.test(key);
-        //     });
-        // }));
-
-        // let list = [];
-
-        // if (results && results[0]) {
-        //     Object.keys(results[0]).forEach((key) => {
-        //         const index = results[0][key];
-
-        //         if (index > threshold) {
-        //             list.push({
-        //                 ip: ((key) => {
-        //                     const m = key.match(regexp);
-
-        //                     return m && m.length > 0 ? m[1] : 'Unknown';
-        //                 })(key),
-        //                 index: index
-        //             })
-        //         }
-        //     });
-        // }
-
-        // list = _.orderBy(list, (item) => {
-        //     return item.index;
-        // }, 'desc').slice(0, limit);
-        console.log(11111)
+        let ret = await zookeeperModel.getPath('/risk');
+
+        if (ret) {
+            ret = JSON.parse(ret);
+        }
+
+        _.each(ret, function(o) {
+            o.process = process[o.process];
+        });
         await ctx.render('action/risk_management', {
+            list: ret
         });
     },
     add: async(ctx, next) => {
+        var data = {};
+        var key = ctx.request.query.key;
+        var oldRet = '';
+
+        if (key) {
+            oldRet = await zookeeperModel.getPath('/risk');
+
+            if (oldRet) {
+                oldRet = JSON.parse(oldRet);
+            }
+    
+            data = _.find(oldRet, function(o) {
+                return `${o.app}/${o.route}` === key;
+            });
+
+            data.state = data.state === 'on';
+            data.isJump = data.process === 'TO_VALIDATE_CODE';
+        }
+
         await ctx.render('action/add_risk', {
+            data: data
         });
     },
     setter: async(ctx) => {
+        let params = ctx.request.body;
+        let path = '/risk';
+        let str = `${params.app}/${params.route}`;
+        let oldRet = await zookeeperModel.getPath('/risk');
+
+        if (!oldRet) {
+            oldRet = [];
+        } else {
+            oldRet = JSON.parse(oldRet);
+        }
+
+        _.remove(oldRet, function(o) {
+            return `${o.app}/${o.route}` === str;
+        });
+
+        oldRet.push(params);
+
+        let result = await zookeeperModel.setPath(path, JSON.stringify(oldRet));
+        
+
+        if (result) {
+            ctx.body = {
+                code: 200,
+                message: 'update success'
+            };
+        } else {
+            ctx.body = {
+                code: 500,
+                message: 'update fail,Please retry'
+            }
+        }
+    },
+    remove: async(ctx) => {
+        let key = ctx.request.body.key;
+        let path = '/risk';
+        let oldRet = await zookeeperModel.getPath(path);
+
+        if (oldRet) {
+            oldRet = JSON.parse(oldRet);
+        }
+
+        _.remove(oldRet, function(o) {
+            return `${o.app}/${o.route}` === key;
+        });
+        let result = await zookeeperModel.setPath(path, JSON.stringify(oldRet));
+        
+
+        if (result) {
+            ctx.body = {
+                code: 200,
+                message: 'remove success'
+            };
+        } else {
+            ctx.body = {
+                code: 500,
+                message: 'remove fail,Please retry'
+            }
+        }
     }
 };
 r.get('/risk_management', risk.index);
 r.get('/add_risk', risk.add);
-r.post('/add_risk_data', risk.setter);
+r.post('/setter', risk.setter);
+r.post('/remove', risk.remove);
 module.exports = r;
--- a/apps/web/models/zookeeperModel.js
View file @8f9b2e7
+++ b/apps/web/models/zookeeperModel.js
View file @8f9b2e7
@@ -27,6 +27,13 @@ class ZookeeperModel extends model {
         return null;
     }
+    async removePath(path) {
+        let degradeServer = await this.getConfigAll();
+
+        if (degradeServer[0]) {
+            return await zookeeperHelpers.remove(`${degradeServer[0].ip}:${degradeServer[0].port}`, path);
+        }
+    }
 }
 module.exports = ZookeeperModel;
--- a/apps/web/routers.js
View file @8f9b2e7
+++ b/apps/web/routers.js
View file @8f9b2e7
@@ -28,8 +28,6 @@ const base = new Router();
 const file = require('./actions/file');
 const riskManagement = require('./actions/risk_management');
-console.log(riskManagement)
-
 module.exports = function(app) {
--- a/apps/web/views/action/add_risk.hbs
View file @8f9b2e7
+++ b/apps/web/views/action/add_risk.hbs
View file @8f9b2e7
@@ -17,29 +17,30 @@
 <!-- pageheader -->
 <div class="contentpanel servers-page form-horizontal">
+    {{#data}}
     <div class="form-group">
         <label for="app" class="col-sm-2 control-label text-center">应用名称：</label>
         <div class="col-sm-6">
-            <input type="text" class="form-control" id="app" placeholder="请输入应用名称">
+            <input type="text" class="form-control" id="app" placeholder="请输入应用名称" value="{{app}}">
         </div>
     </div>
     <div class="form-group">
         <label for="router" class="col-sm-2 control-label text-center">路由路径：</label>
         <div class="col-sm-6">
-            <input type="text" class="form-control" id="router" placeholder="请输入路由路径">
+            <input type="text" class="form-control" id="router" placeholder="请输入路由路径" value="{{route}}">
         </div>
     </div>
     <div class="form-group">
         <label for="exampleFormControlInput1" class="col-sm-2 control-label text-center">时间间隔：</label>
         <div class="col-sm-6">
-            <input type="text" class="form-control" id="router" placeholder="请输入时间间隔">
+            <input type="text" class="form-control" id="interval" placeholder="请输入时间间隔" value="{{interval}}">
         </div>
         <div class="col-sm-2" style="line-height: 40px;">秒</div>
     </div>
     <div class="form-group">
         <label for="exampleFormControlInput1" class="col-sm-2 control-label text-center">请求次数：</label>
         <div class="col-sm-6">
-            <input type="text" class="form-control" id="router" placeholder="请输入请求次数">
+            <input type="text" class="form-control" id="requests" placeholder="请输入请求次数" value="{{requests}}">
         </div>
         <div class="col-sm-2" style="line-height: 40px;">次</div>
     </div>
@@ -47,8 +48,8 @@
         <label for="process" class="col-sm-2 control-label text-center">处理方式：</label>
         <div class="col-sm-6">
             <select class="form-control" style="height: 40px; line-height: 40px;" id="process">
-                <option value="TO_VALIDATE_CODE">跳转全局验证码解锁</option>
-                <option value="REDIRECT_403">返回403</option>
+                <option value="TO_VALIDATE_CODE" {{#if isJump}}selected{{/if}}>跳转全局验证码解锁</option>
+                <option value="REDIRECT_403" {{#unless isJump}}selected{{/unless}}>返回403</option>
             </select>
         </div>
     </div>
@@ -56,13 +57,14 @@
         <label class="col-sm-2 control-label text-center">状　　态：</label>
         <div class="col-sm-10">
             <label class="radio-inline">
-                <input type="radio" name="status" id="status-on" value="on" checked>开启
+                <input type="radio" name="status" id="status-on" value="on" {{#if state}}checked{{/if}}>开启
             </label>
             <label class="radio-inline">
-                <input type="radio" name="status" id="status-off" value="off">关闭
+                <input type="radio" name="status" id="status-off" value="off"{{#unless state}}checked{{/unless}}>关闭
             </label>
         </div>
     </div>
+    {{/data}}
     <div class="row">
         <label class="col-sm-2"></label>
         <div class="col-sm-10">
@@ -73,7 +75,21 @@
 </div>
 <script>
     $('.save').on('click', function() {
-        $.post('/abuse_protection/unlock', {remoteIp: $(this).data('ip')}, function() {
+        $.post('/risk_management/setter', {
+            app: $('#app').val(),
+            route: $('#router').val(),
+            interval: $('#interval').val(),
+            requests: $('#requests').val(),
+            process: $('#process').val(),
+            state: $('input[type=radio]:checked').val(),
+            regRoute: $('#router').val().replace('/', '\\/')
+        }, function(ret) {
+            if (ret.code === 200) {
+                alert('添加成功！');
+                location.href = '/risk_management/risk_management';
+            } else {
+                alert('添加失败');
+            }
         });
     });
 </script>
--- a/apps/web/views/action/risk_management.hbs
View file @8f9b2e7
+++ b/apps/web/views/action/risk_management.hbs
View file @8f9b2e7
@@ -32,13 +32,13 @@
         <tbody>
             {{#list}}
                 <tr>
-                    <td>{{app}}</td>
-                    <td>{{route}}</td>
+                    <td class="app">{{app}}</td>
+                    <td class="route">{{route}}</td>
                     <td>{{interval}}</td>
                     <td>{{requests}}</td>
                     <td>{{process}}</td>
                     <td>{{state}}</td>
-                    <td><button>编辑</button><button>删除</button></td>
+                    <td><button type="button" class="btn btn-success edit">编辑</button><button  type="button" class="btn btn-danger ml10 del">删除</button></td>
                 </tr>
             {{/list}}
@@ -55,4 +55,25 @@
     $('.add').on('click', function() {
         location.href = '/risk_management/add_risk';
     });
+
+    $('.del').on('click', function() {
+        var key = $(this).parents('tr').find('.app').html() + '/' + $(this).parents('tr').find('.route').html();
+
+        $.post('/risk_management/remove', {
+            key: key
+        }, function(ret) {
+            if (ret.code === 200) {
+                alert('删除成功！');
+                location.reload();
+            } else {
+                alert('删除失败！');
+            }
+        });
+    });
+
+    $('.edit').on('click', function() {
+        var key = $(this).parents('tr').find('.app').html() + '/' + $(this).parents('tr').find('.route').html();
+
+        location.href = '/risk_management/add_risk?key=' + key;
+    });
 </script>