tanghc
5 years ago
parent
c850712f8b
commit
a5a0b5c476
@ -0,0 +1,9 @@ |
|||||||
|
CREATE TABLE `config_ip_blacklist` ( |
||||||
|
`id` bigint(20) unsigned NOT NULL AUTO_INCREMENT, |
||||||
|
`ip` varchar(64) NOT NULL DEFAULT '' COMMENT 'ip', |
||||||
|
`remark` varchar(128) DEFAULT NULL COMMENT '备注', |
||||||
|
`gmt_create` datetime DEFAULT CURRENT_TIMESTAMP, |
||||||
|
`gmt_modified` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, |
||||||
|
PRIMARY KEY (`id`), |
||||||
|
UNIQUE KEY `uk_ip` (`ip`) USING BTREE |
||||||
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='IP黑名单'; |
||||||
@ -0,0 +1,184 @@ |
|||||||
|
<template> |
||||||
|
<div class="app-container"> |
||||||
|
<el-form :inline="true" :model="searchFormData" class="demo-form-inline" size="mini"> |
||||||
|
<el-form-item label="IP"> |
||||||
|
<el-input v-model="searchFormData.ip" :clearable="true" placeholder="输入IP" style="width: 250px;" /> |
||||||
|
</el-form-item> |
||||||
|
<el-form-item> |
||||||
|
<el-button type="primary" icon="el-icon-search" @click="loadTable">查询</el-button> |
||||||
|
</el-form-item> |
||||||
|
</el-form> |
||||||
|
<el-button type="primary" size="mini" icon="el-icon-plus" style="margin-bottom: 10px;" @click="onAdd">新增IP</el-button> |
||||||
|
<el-table |
||||||
|
:data="pageInfo.rows" |
||||||
|
border |
||||||
|
highlight-current-row |
||||||
|
> |
||||||
|
<el-table-column |
||||||
|
prop="ip" |
||||||
|
label="IP" |
||||||
|
width="200" |
||||||
|
/> |
||||||
|
<el-table-column |
||||||
|
prop="remark" |
||||||
|
label="备注" |
||||||
|
width="300" |
||||||
|
/> |
||||||
|
<el-table-column |
||||||
|
prop="gmtCreate" |
||||||
|
label="添加时间" |
||||||
|
width="160" |
||||||
|
/> |
||||||
|
<el-table-column |
||||||
|
prop="gmtModified" |
||||||
|
label="修改时间" |
||||||
|
width="160" |
||||||
|
/> |
||||||
|
<el-table-column |
||||||
|
label="操作" |
||||||
|
width="150" |
||||||
|
> |
||||||
|
<template slot-scope="scope"> |
||||||
|
<el-button type="text" size="mini" @click="onTableUpdate(scope.row)">修改</el-button> |
||||||
|
<el-button type="text" size="mini" @click="onTableDelete(scope.row)">删除</el-button> |
||||||
|
</template> |
||||||
|
</el-table-column> |
||||||
|
</el-table> |
||||||
|
<el-pagination |
||||||
|
background |
||||||
|
style="margin-top: 5px" |
||||||
|
:current-page="searchFormData.pageIndex" |
||||||
|
:page-size="searchFormData.pageSize" |
||||||
|
:page-sizes="[5, 10, 20, 40]" |
||||||
|
:total="pageInfo.total" |
||||||
|
layout="total, sizes, prev, pager, next" |
||||||
|
@size-change="onSizeChange" |
||||||
|
@current-change="onPageIndexChange" |
||||||
|
/> |
||||||
|
<!--dialog--> |
||||||
|
<el-dialog |
||||||
|
:title="dialogTitle" |
||||||
|
:visible.sync="dialogVisible" |
||||||
|
:close-on-click-modal="false" |
||||||
|
@close="resetForm('dialogForm')" |
||||||
|
> |
||||||
|
<el-form |
||||||
|
ref="dialogForm" |
||||||
|
:rules="dialogFormRules" |
||||||
|
:model="dialogFormData" |
||||||
|
label-width="120px" |
||||||
|
size="mini" |
||||||
|
> |
||||||
|
<el-form-item prop="ip" label="IP"> |
||||||
|
<el-input v-show="dialogFormData.id === 0" v-model="dialogFormData.ip" /> |
||||||
|
<span v-show="dialogFormData.id > 0">{{ dialogFormData.ip }}</span> |
||||||
|
</el-form-item> |
||||||
|
<el-form-item prop="remark" label="备注"> |
||||||
|
<el-input v-model="dialogFormData.remark" /> |
||||||
|
</el-form-item> |
||||||
|
</el-form> |
||||||
|
<div slot="footer" class="dialog-footer"> |
||||||
|
<el-button @click="dialogVisible = false">取 消</el-button> |
||||||
|
<el-button type="primary" @click="onDialogSave">保 存</el-button> |
||||||
|
</div> |
||||||
|
</el-dialog> |
||||||
|
</div> |
||||||
|
</template> |
||||||
|
|
||||||
|
<script> |
||||||
|
export default { |
||||||
|
data() { |
||||||
|
const ipValidator = (rule, value, callback) => { |
||||||
|
if (value === '') { |
||||||
|
callback(new Error('请输入IP')) |
||||||
|
} else { |
||||||
|
const regexIP = /^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$/ |
||||||
|
if (!regexIP.test(value)) { |
||||||
|
callback(new Error('IP格式不正确')) |
||||||
|
} |
||||||
|
callback() |
||||||
|
} |
||||||
|
} |
||||||
|
return { |
||||||
|
searchFormData: { |
||||||
|
ip: '', |
||||||
|
pageIndex: 1, |
||||||
|
pageSize: 10 |
||||||
|
}, |
||||||
|
pageInfo: { |
||||||
|
rows: [], |
||||||
|
total: 0 |
||||||
|
}, |
||||||
|
dialogVisible: false, |
||||||
|
dialogTitle: '', |
||||||
|
dialogFormData: { |
||||||
|
id: 0, |
||||||
|
ip: '', |
||||||
|
remark: '' |
||||||
|
}, |
||||||
|
dialogFormRules: { |
||||||
|
ip: [ |
||||||
|
{ validator: ipValidator, trigger: 'blur' }, |
||||||
|
{ min: 1, max: 64, message: '长度在 1 到 64 个字符', trigger: 'blur' } |
||||||
|
], |
||||||
|
remark: [ |
||||||
|
{ max: 100, message: '不能超过 100 个字符', trigger: 'blur' } |
||||||
|
] |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
created() { |
||||||
|
this.loadTable() |
||||||
|
}, |
||||||
|
methods: { |
||||||
|
loadTable: function() { |
||||||
|
this.post('ip.blacklist.page', this.searchFormData, function(resp) { |
||||||
|
this.pageInfo = resp.data |
||||||
|
}) |
||||||
|
}, |
||||||
|
onTableUpdate: function(row) { |
||||||
|
this.dialogTitle = '修改IP' |
||||||
|
this.dialogVisible = true |
||||||
|
this.$nextTick(() => { |
||||||
|
Object.assign(this.dialogFormData, row) |
||||||
|
}) |
||||||
|
}, |
||||||
|
onTableDelete: function(row) { |
||||||
|
this.confirm(`确认要移除IP【${row.ip}】吗？`, function(done) { |
||||||
|
const data = { |
||||||
|
id: row.id |
||||||
|
} |
||||||
|
this.post('ip.blacklist.del', data, function() { |
||||||
|
done() |
||||||
|
this.tip('删除成功') |
||||||
|
this.loadTable() |
||||||
|
}) |
||||||
|
}) |
||||||
|
}, |
||||||
|
onDialogSave: function() { |
||||||
|
this.$refs.dialogForm.validate((valid) => { |
||||||
|
if (valid) { |
||||||
|
const uri = this.dialogFormData.id ? 'ip.blacklist.update' : 'ip.blacklist.add' |
||||||
|
this.post(uri, this.dialogFormData, function() { |
||||||
|
this.dialogVisible = false |
||||||
|
this.loadTable() |
||||||
|
}) |
||||||
|
} |
||||||
|
}) |
||||||
|
}, |
||||||
|
onSizeChange: function(size) { |
||||||
|
this.searchFormData.pageSize = size |
||||||
|
this.loadTable() |
||||||
|
}, |
||||||
|
onAdd: function() { |
||||||
|
this.dialogTitle = '新增IP' |
||||||
|
this.dialogVisible = true |
||||||
|
this.dialogFormData.id = 0 |
||||||
|
}, |
||||||
|
onPageIndexChange: function(pageIndex) { |
||||||
|
this.searchFormData.pageIndex = pageIndex |
||||||
|
this.loadTable() |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
</script> |
||||||
@ -1,347 +0,0 @@ |
|||||||
<template> |
|
||||||
<div class="app-container"> |
|
||||||
<el-container> |
|
||||||
<el-aside style="min-height: 300px;width: 200px;"> |
|
||||||
<el-input v-model="filterText" prefix-icon="el-icon-search" placeholder="搜索服务..." style="margin-bottom:20px;" size="mini" clearable /> |
|
||||||
<el-tree |
|
||||||
ref="tree2" |
|
||||||
:data="treeData" |
|
||||||
:props="defaultProps" |
|
||||||
:filter-node-method="filterNode" |
|
||||||
:highlight-current="true" |
|
||||||
:expand-on-click-node="false" |
|
||||||
empty-text="无数据" |
|
||||||
node-key="id" |
|
||||||
class="filter-tree" |
|
||||||
default-expand-all |
|
||||||
@node-click="onNodeClick" |
|
||||||
> |
|
||||||
<span slot-scope="{ node, data }" class="custom-tree-node"> |
|
||||||
<span v-if="data.label.length < 15">{{ data.label }}</span> |
|
||||||
<span v-else> |
|
||||||
<el-tooltip :content="data.label" class="item" effect="light" placement="right"> |
|
||||||
<span>{{ data.label.substring(0, 15) + '...' }}</span> |
|
||||||
</el-tooltip> |
|
||||||
</span> |
|
||||||
</span> |
|
||||||
</el-tree> |
|
||||||
</el-aside> |
|
||||||
<el-main style="padding-top:0"> |
|
||||||
<el-form :inline="true" :model="searchFormData" class="demo-form-inline"> |
|
||||||
<el-form-item label="路由名称"> |
|
||||||
<el-input v-model="searchFormData.id" placeholder="输入接口名或版本号" size="mini" /> |
|
||||||
</el-form-item> |
|
||||||
<el-form-item> |
|
||||||
<el-button type="primary" icon="el-icon-search" size="mini" @click="onSearchTable">查询</el-button> |
|
||||||
</el-form-item> |
|
||||||
</el-form> |
|
||||||
<el-table |
|
||||||
:data="tableData" |
|
||||||
border |
|
||||||
max-height="500" |
|
||||||
> |
|
||||||
<el-table-column |
|
||||||
prop="name" |
|
||||||
label="接口名 (版本号)" |
|
||||||
width="200" |
|
||||||
> |
|
||||||
<template slot-scope="scope"> |
|
||||||
{{ scope.row.name + (scope.row.version ? ' (' + scope.row.version + ')' : '') }} |
|
||||||
</template> |
|
||||||
</el-table-column> |
|
||||||
<el-table-column |
|
||||||
prop="limitType" |
|
||||||
label="限流策略" |
|
||||||
width="120" |
|
||||||
> |
|
||||||
<template slot-scope slot="header"> |
|
||||||
限流策略 <i class="el-icon-question" style="cursor: pointer" @click="onLimitTypeTipClick"></i> |
|
||||||
</template> |
|
||||||
<template slot-scope="scope"> |
|
||||||
<span v-if="scope.row.limitType === 1">漏桶策略</span> |
|
||||||
<span v-if="scope.row.limitType === 2">令牌桶策略</span> |
|
||||||
</template> |
|
||||||
</el-table-column> |
|
||||||
<el-table-column |
|
||||||
prop="info" |
|
||||||
label="限流信息" |
|
||||||
width="500" |
|
||||||
> |
|
||||||
<template slot-scope="scope"> |
|
||||||
<span v-html="infoRender(scope.row)"></span> |
|
||||||
</template> |
|
||||||
</el-table-column> |
|
||||||
<el-table-column |
|
||||||
prop="limitStatus" |
|
||||||
label="状态" |
|
||||||
width="80" |
|
||||||
> |
|
||||||
<template slot-scope="scope"> |
|
||||||
<span v-if="scope.row.limitStatus === 1" style="color:#67C23A">已开启</span> |
|
||||||
<span v-if="scope.row.limitStatus === 0" style="color:#909399">已关闭</span> |
|
||||||
</template> |
|
||||||
</el-table-column> |
|
||||||
<el-table-column |
|
||||||
label="操作" |
|
||||||
width="80" |
|
||||||
> |
|
||||||
<template slot-scope="scope"> |
|
||||||
<el-button type="text" size="mini" @click="onTableUpdate(scope.row)">修改</el-button> |
|
||||||
</template> |
|
||||||
</el-table-column> |
|
||||||
</el-table> |
|
||||||
<!-- dialog --> |
|
||||||
<el-dialog |
|
||||||
title="设置限流" |
|
||||||
:visible.sync="limitDialogVisible" |
|
||||||
:close-on-click-modal="false" |
|
||||||
@close="onLimitDialogClose" |
|
||||||
> |
|
||||||
<el-form ref="limitDialogFormMain" :model="limitDialogFormData"> |
|
||||||
<el-form-item label="id" :label-width="formLabelWidth"> |
|
||||||
<el-input v-model="limitDialogFormData.routeId" readonly="readonly"/> |
|
||||||
</el-form-item> |
|
||||||
<el-form-item label="限流策略" :label-width="formLabelWidth"> |
|
||||||
<el-radio-group v-model="limitDialogFormData.limitType"> |
|
||||||
<el-radio :label="1">漏桶策略</el-radio> |
|
||||||
<el-radio :label="2">令牌桶策略</el-radio> |
|
||||||
</el-radio-group> |
|
||||||
</el-form-item> |
|
||||||
<el-form-item label="开启状态" :label-width="formLabelWidth"> |
|
||||||
<el-switch |
|
||||||
v-model="limitDialogFormData.limitStatus" |
|
||||||
active-color="#13ce66" |
|
||||||
inactive-color="#ff4949" |
|
||||||
:active-value="1" |
|
||||||
:inactive-value="0" |
|
||||||
> |
|
||||||
</el-switch> |
|
||||||
</el-form-item> |
|
||||||
</el-form> |
|
||||||
<el-form |
|
||||||
v-show="limitDialogFormData.limitType === 1 && limitDialogFormData.limitStatus" |
|
||||||
ref="limitDialogFormLeaky" |
|
||||||
:rules="rulesLeaky" |
|
||||||
:model="limitDialogFormData" |
|
||||||
> |
|
||||||
<el-form-item label="每秒可处理请求数" prop="execCountPerSecond" :label-width="formLabelWidth"> |
|
||||||
<el-input-number v-model="limitDialogFormData.execCountPerSecond" controls-position="right" :min="1" /> |
|
||||||
</el-form-item> |
|
||||||
<el-form-item label="错误码" prop="limitCode" :label-width="formLabelWidth"> |
|
||||||
<el-input v-model="limitDialogFormData.limitCode" /> |
|
||||||
</el-form-item> |
|
||||||
<el-form-item label="错误信息" prop="limitMsg" :label-width="formLabelWidth"> |
|
||||||
<el-input v-model="limitDialogFormData.limitMsg" /> |
|
||||||
</el-form-item> |
|
||||||
</el-form> |
|
||||||
<el-form |
|
||||||
v-show="limitDialogFormData.limitType === 2 && limitDialogFormData.limitStatus" |
|
||||||
ref="limitDialogFormToken" |
|
||||||
:rules="rulesToken" |
|
||||||
:model="limitDialogFormData" |
|
||||||
> |
|
||||||
<el-form-item label="令牌桶容量" prop="tokenBucketCount" :label-width="formLabelWidth"> |
|
||||||
<el-input-number v-model="limitDialogFormData.tokenBucketCount" controls-position="right" :min="1" /> |
|
||||||
</el-form-item> |
|
||||||
</el-form> |
|
||||||
<div slot="footer" class="dialog-footer"> |
|
||||||
<el-button @click="limitDialogVisible = false">取 消</el-button> |
|
||||||
<el-button type="primary" @click="onLimitDialogSave">保 存</el-button> |
|
||||||
</div> |
|
||||||
</el-dialog> |
|
||||||
</el-main> |
|
||||||
</el-container> |
|
||||||
</div> |
|
||||||
</template> |
|
||||||
|
|
||||||
<script> |
|
||||||
export default { |
|
||||||
data() { |
|
||||||
return { |
|
||||||
filterText: '', |
|
||||||
treeData: [], |
|
||||||
tableData: [], |
|
||||||
serviceId: '', |
|
||||||
searchFormData: {}, |
|
||||||
defaultProps: { |
|
||||||
children: 'children', |
|
||||||
label: 'label' |
|
||||||
}, |
|
||||||
// dialog |
|
||||||
limitDialogFormData: { |
|
||||||
routeId: '', |
|
||||||
execCountPerSecond: 5, |
|
||||||
limitCode: '', |
|
||||||
limitMsg: '', |
|
||||||
tokenBucketCount: 5, |
|
||||||
limitStatus: 0, // 0: 停用，1：启用 |
|
||||||
limitType: 1 |
|
||||||
}, |
|
||||||
rulesLeaky: { |
|
||||||
execCountPerSecond: [ |
|
||||||
{ required: true, message: '不能为空', trigger: 'blur' } |
|
||||||
], |
|
||||||
limitCode: [ |
|
||||||
{ required: true, message: '不能为空', trigger: 'blur' }, |
|
||||||
{ min: 1, max: 64, message: '长度在 1 到 64 个字符', trigger: 'blur' } |
|
||||||
], |
|
||||||
limitMsg: [ |
|
||||||
{ required: true, message: '不能为空', trigger: 'blur' }, |
|
||||||
{ min: 1, max: 100, message: '长度在 1 到 100 个字符', trigger: 'blur' } |
|
||||||
] |
|
||||||
}, |
|
||||||
rulesToken: { |
|
||||||
tokenBucketCount: [ |
|
||||||
{ required: true, message: '不能为空', trigger: 'blur' } |
|
||||||
] |
|
||||||
}, |
|
||||||
formLabelWidth: '150px', |
|
||||||
limitDialogVisible: false |
|
||||||
|
|
||||||
} |
|
||||||
}, |
|
||||||
watch: { |
|
||||||
filterText(val) { |
|
||||||
this.$refs.tree2.filter(val) |
|
||||||
} |
|
||||||
}, |
|
||||||
created() { |
|
||||||
this.loadTree() |
|
||||||
}, |
|
||||||
methods: { |
|
||||||
// 加载树 |
|
||||||
loadTree: function() { |
|
||||||
this.post('zookeeper.service.list', {}, function(resp) { |
|
||||||
const respData = resp.data |
|
||||||
this.treeData = this.convertToTreeData(respData, 0) |
|
||||||
}) |
|
||||||
}, |
|
||||||
// 树搜索 |
|
||||||
filterNode(value, data) { |
|
||||||
if (!value) return true |
|
||||||
return data.label.indexOf(value) !== -1 |
|
||||||
}, |
|
||||||
// 树点击事件 |
|
||||||
onNodeClick(data, node, tree) { |
|
||||||
if (data.parentId) { |
|
||||||
this.serviceId = data.label |
|
||||||
this.searchFormData.serviceId = this.serviceId |
|
||||||
this.loadTable() |
|
||||||
} |
|
||||||
}, |
|
||||||
/** |
|
||||||
* 数组转成树状结构 |
|
||||||
* @param data 数据结构 [{ |
|
||||||
"_parentId": 14, |
|
||||||
"gmtCreate": "2019-01-15 09:44:38", |
|
||||||
"gmtUpdate": "2019-01-15 09:44:38", |
|
||||||
"id": 15, |
|
||||||
"isShow": 1, |
|
||||||
"name": "用户注册", |
|
||||||
"orderIndex": 10000, |
|
||||||
"parentId": 14 |
|
||||||
},...] |
|
||||||
* @param pid 初始父节点id，一般是0 |
|
||||||
* @return 返回结果 [{ |
|
||||||
label: '一级 1', |
|
||||||
children: [{ |
|
||||||
label: '二级 1-1', |
|
||||||
children: [{ |
|
||||||
label: '三级 1-1-1' |
|
||||||
}] |
|
||||||
}] |
|
||||||
} |
|
||||||
*/ |
|
||||||
convertToTreeData(data, pid) { |
|
||||||
const result = [] |
|
||||||
const root = { |
|
||||||
label: '服务列表', |
|
||||||
parentId: pid |
|
||||||
} |
|
||||||
const children = [] |
|
||||||
for (let i = 0; i < data.length; i++) { |
|
||||||
const item = { label: data[i].serviceId, parentId: 1 } |
|
||||||
children.push(item) |
|
||||||
} |
|
||||||
root.children = children |
|
||||||
result.push(root) |
|
||||||
return result |
|
||||||
}, |
|
||||||
// table |
|
||||||
loadTable: function() { |
|
||||||
this.post('route.limit.list', this.searchFormData, function(resp) { |
|
||||||
this.tableData = resp.data |
|
||||||
}) |
|
||||||
}, |
|
||||||
onSearchTable: function() { |
|
||||||
this.loadTable() |
|
||||||
}, |
|
||||||
onTableUpdate: function(row) { |
|
||||||
this.limitDialogVisible = true |
|
||||||
this.$nextTick(() => { |
|
||||||
Object.assign(this.limitDialogFormData, row) |
|
||||||
}) |
|
||||||
}, |
|
||||||
resetForm(formName) { |
|
||||||
const frm = this.$refs[formName] |
|
||||||
frm && frm.resetFields() |
|
||||||
}, |
|
||||||
onLimitDialogClose: function() { |
|
||||||
this.resetForm('limitDialogFormLeaky') |
|
||||||
this.resetForm('limitDialogFormToken') |
|
||||||
this.limitDialogVisible = false |
|
||||||
}, |
|
||||||
infoRender: function(row) { |
|
||||||
if (!row.hasRecord) { |
|
||||||
return '--' |
|
||||||
} |
|
||||||
const html = [] |
|
||||||
if (row.limitType === 1) { |
|
||||||
html.push('每秒可处理请求数：' + row.execCountPerSecond) |
|
||||||
html.push('subCode：' + row.limitCode) |
|
||||||
html.push('subMsg：' + row.limitMsg) |
|
||||||
} else if (row.limitType === 2) { |
|
||||||
html.push('令牌桶容量：' + row.tokenBucketCount) |
|
||||||
} |
|
||||||
return html.join('，') |
|
||||||
}, |
|
||||||
onLimitDialogSave: function() { |
|
||||||
this.doValidate(function() { |
|
||||||
this.limitDialogFormData.serviceId = this.serviceId |
|
||||||
this.post('route.limit.update', this.limitDialogFormData, function(resp) { |
|
||||||
this.limitDialogVisible = false |
|
||||||
this.loadTable() |
|
||||||
}) |
|
||||||
}) |
|
||||||
}, |
|
||||||
doValidate: function(callback) { |
|
||||||
const that = this |
|
||||||
if (this.limitDialogFormData.limitStatus === 0) { |
|
||||||
callback.call(this) |
|
||||||
return |
|
||||||
} |
|
||||||
if (this.limitDialogFormData.limitType === 1) { |
|
||||||
this.$refs['limitDialogFormLeaky'].validate((valid) => { |
|
||||||
if (valid) { |
|
||||||
callback.call(that) |
|
||||||
} |
|
||||||
}) |
|
||||||
} else { |
|
||||||
this.$refs['limitDialogFormToken'].validate((valid) => { |
|
||||||
if (valid) { |
|
||||||
callback.call(that) |
|
||||||
} |
|
||||||
}) |
|
||||||
} |
|
||||||
}, |
|
||||||
onLimitTypeTipClick: function() { |
|
||||||
const leakyRemark = '漏桶策略：每秒处理固定数量的请求，超出请求返回错误信息。' |
|
||||||
const tokenRemark = '令牌桶策略：每秒放置固定数量的令牌数，每个请求进来后先去拿令牌，拿到了令牌才能继续，拿不到则等候令牌重新生成了再拿。' |
|
||||||
const content = leakyRemark + '<br>' + tokenRemark |
|
||||||
this.$alert(content, '限流策略', { |
|
||||||
dangerouslyUseHTMLString: true |
|
||||||
}) |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
</script> |
|
||||||
@ -0,0 +1,64 @@ |
|||||||
|
package com.gitee.sop.gateway.manager; |
||||||
|
|
||||||
|
import com.alibaba.fastjson.JSON; |
||||||
|
import com.gitee.sop.gateway.mapper.IPBlacklistMapper; |
||||||
|
import com.gitee.sop.gatewaycommon.bean.ChannelMsg; |
||||||
|
import com.gitee.sop.gatewaycommon.manager.DefaultIPBlacklistManager; |
||||||
|
import com.gitee.sop.gatewaycommon.manager.ZookeeperContext; |
||||||
|
import lombok.Data; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||||
|
import org.springframework.core.env.Environment; |
||||||
|
|
||||||
|
import javax.annotation.PostConstruct; |
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* 限流配置管理 |
||||||
|
* @author tanghc |
||||||
|
*/ |
||||||
|
@Slf4j |
||||||
|
public class DbIPBlacklistManager extends DefaultIPBlacklistManager { |
||||||
|
|
||||||
|
@Autowired |
||||||
|
IPBlacklistMapper ipBlacklistMapper; |
||||||
|
|
||||||
|
@Autowired |
||||||
|
Environment environment; |
||||||
|
|
||||||
|
@Override |
||||||
|
public void load() { |
||||||
|
List<String> ipList = ipBlacklistMapper.listAllIP(); |
||||||
|
log.info("加载IP黑名单, size:{}", ipList.size()); |
||||||
|
ipList.stream().forEach(this::add); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@PostConstruct |
||||||
|
protected void after() throws Exception { |
||||||
|
ZookeeperContext.setEnvironment(environment); |
||||||
|
String path = ZookeeperContext.getIpBlacklistChannelPath(); |
||||||
|
ZookeeperContext.listenPath(path, nodeCache -> { |
||||||
|
String nodeData = new String(nodeCache.getCurrentData().getData()); |
||||||
|
ChannelMsg channelMsg = JSON.parseObject(nodeData, ChannelMsg.class); |
||||||
|
final IPDto ipDto = JSON.parseObject(channelMsg.getData(), IPDto.class); |
||||||
|
String ip = ipDto.getIp(); |
||||||
|
switch (channelMsg.getOperation()) { |
||||||
|
case "add": |
||||||
|
log.info("添加IP黑名单，ip:{}", ip); |
||||||
|
add(ip); |
||||||
|
break; |
||||||
|
case "delete": |
||||||
|
log.info("移除IP黑名单，ip:{}", ip); |
||||||
|
remove(ip); |
||||||
|
break; |
||||||
|
} |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
@Data |
||||||
|
private static class IPDto { |
||||||
|
private String ip; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,22 @@ |
|||||||
|
package com.gitee.sop.gateway.mapper; |
||||||
|
|
||||||
|
import org.apache.ibatis.annotations.Mapper; |
||||||
|
import org.apache.ibatis.annotations.Select; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* IP黑名单 |
||||||
|
* @author tanghc |
||||||
|
*/ |
||||||
|
@Mapper |
||||||
|
public interface IPBlacklistMapper { |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取所有IP |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
@Select("SELECT ip FROM config_ip_blacklist") |
||||||
|
List<String> listAllIP(); |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue