diff --git a/chsm-web-server/src/main/java/com/sunyard/chsm/controller/KeyManageController.java b/chsm-web-server/src/main/java/com/sunyard/chsm/controller/KeyManageController.java index 775985c..080f7fc 100644 --- a/chsm-web-server/src/main/java/com/sunyard/chsm/controller/KeyManageController.java +++ b/chsm-web-server/src/main/java/com/sunyard/chsm/controller/KeyManageController.java @@ -39,6 +39,7 @@ public class KeyManageController { * @return 分页列表 */ @PostMapping("/pageList") + @AuthCode(AuthCodeConst.key_list) public R> queryPageList(@RequestBody KeyInfoQuery query) { Page page = keyManageService.queryPageList(query); return R.data(page); @@ -51,6 +52,7 @@ public class KeyManageController { * @return 分页列表 */ @PostMapping("/info") + @AuthCode(AuthCodeConst.key_info) public R queryInfo(@RequestBody Long id) { Assert.notNull(id, "密钥id不能为空"); KeyInfoResp resp = keyManageService.queryInfo(id); @@ -63,6 +65,7 @@ public class KeyManageController { * @return id */ @PostMapping("/gen") + @AuthCode(AuthCodeConst.key_create) public R save(@Valid @RequestBody KeyCreateReq req) { Long id = keyManageService.create(req); return R.data(id); @@ -75,6 +78,7 @@ public class KeyManageController { * @return id */ @PostMapping("/update") + @AuthCode(AuthCodeConst.key_update) public R update(@Valid @RequestBody KeyUpdateReq req) { keyManageService.update(req); return R.ok(); @@ -100,6 +104,7 @@ public class KeyManageController { * @return id */ @PostMapping("/disable") + @AuthCode(AuthCodeConst.key_disable) public R disableKey(@Valid @RequestBody KeyManageReq req) { keyManageService.disableKey(req.getIds()); return R.ok(); @@ -112,6 +117,7 @@ public class KeyManageController { * @return id */ @PostMapping("/archive") + @AuthCode(AuthCodeConst.key_archive) public R archiveKey(@Valid @RequestBody KeyManageReq req) { keyManageService.archiveKey(req.getIds()); return R.ok(); @@ -124,6 +130,7 @@ public class KeyManageController { * @return id */ @PostMapping("/destroy") + @AuthCode(AuthCodeConst.key_destroy) public R destroyKey(@Valid @RequestBody KeyManageReq req) { keyManageService.destroyKey(req.getIds()); return R.ok(); diff --git a/chsm-web-server/src/main/java/com/sunyard/chsm/controller/SymKeyController.java b/chsm-web-server/src/main/java/com/sunyard/chsm/controller/SymKeyController.java index ce68555..555755f 100644 --- a/chsm-web-server/src/main/java/com/sunyard/chsm/controller/SymKeyController.java +++ b/chsm-web-server/src/main/java/com/sunyard/chsm/controller/SymKeyController.java @@ -1,5 +1,7 @@ package com.sunyard.chsm.controller; +import com.sunyard.chsm.auth.AuthCode; +import com.sunyard.chsm.constant.AuthCodeConst; import com.sunyard.chsm.model.R; import com.sunyard.chsm.param.SymDecryptReq; import com.sunyard.chsm.param.SymDecryptResp; @@ -42,6 +44,7 @@ public class SymKeyController { * @return */ @PostMapping("/encrypt") + @AuthCode(AuthCodeConst.sym_enc) public R encrypt(@Valid @RequestBody SymEncryptReq req) { SymEncryptResp resp = symKeyService.encrypt(req); return R.data(resp); @@ -54,6 +57,7 @@ public class SymKeyController { * @return */ @PostMapping("/decrypt") + @AuthCode(AuthCodeConst.sym_dec) public R decrypt(@Valid @RequestBody SymDecryptReq req) { SymDecryptResp resp = symKeyService.decrypt(req); return R.data(resp); @@ -66,6 +70,7 @@ public class SymKeyController { * @return */ @PostMapping("/hmac") + @AuthCode(AuthCodeConst.cal_hmac) public R hmac(@Valid @RequestBody SymHmacReq req) { SymHmacResp resp = symKeyService.hmac(req); return R.data(resp); @@ -78,6 +83,7 @@ public class SymKeyController { * @return */ @PostMapping("/hmac/check") + @AuthCode(AuthCodeConst.check_hmac) public R macCheck(@Valid @RequestBody SymHmacCheckReq req) { SymHmacCheckResp resp = symKeyService.hmacCheck(req); return R.data(resp); @@ -90,6 +96,7 @@ public class SymKeyController { * @return */ @PostMapping("/mac") + @AuthCode(AuthCodeConst.cal_mac) public R mac(@Valid @RequestBody SymMacReq req) { SymMacResp resp = symKeyService.mac(req); return R.data(resp); @@ -102,6 +109,7 @@ public class SymKeyController { * @return */ @PostMapping("/mac/check") + @AuthCode(AuthCodeConst.check_mac) public R macCheck(@Valid @RequestBody SymMacCheckReq req) { SymMacCheckResp resp = symKeyService.macCheck(req); return R.data(resp);