AdminController.java 4.15 KB
package com.huaheng.control.management.controller;

import java.util.Arrays;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;

import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTVerifier;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;
import com.huaheng.control.management.dto.AcsStatus;
import com.huaheng.control.management.dto.GenerateRequestToken;
import com.huaheng.control.management.utils.HuaHengJwtUtils;
import com.huaheng.control.management.utils.HuahengRedisUtils;
import com.huaheng.control.management.utils.config.ApplicationConfig;
import com.huaheng.control.management.utils.constant.RSA256Key;
import com.huaheng.control.management.utils.support.PassRequestAuthentication;
import com.huaheng.control.management.utils.support.RequestAuthentication;
import com.huaheng.control.management.vo.Result;

import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@RestController
@RequestMapping("/admin")
public class AdminController {

    @Resource
    private ApplicationConfig applicationConfig;

    @Resource
    private HuahengRedisUtils huahengRedisUtils;

    @ResponseBody
    @PostMapping("/testPostMapping")
    @Operation(summary = "测试PostMapping")
    public Result<?> testPostMapping(@RequestBody AcsStatus acsStatus) {
        huahengRedisUtils.set("testPostMapping_test", acsStatus);
        return Result.OK("执行成功", huahengRedisUtils.get("testPostMapping_test", AcsStatus.class));
    }

    @PassRequestAuthentication
    @ResponseBody
    @PostMapping("/generateRequestToken")
    @Operation(summary = "生成请求TOKEN")
    public Result<?> generateRequestToken(@RequestBody @Validated GenerateRequestToken generateRequestToken, HttpServletRequest request) {
        Result<String> result = new Result<String>();
        // 生成API TOKEN
        RequestAuthentication requestAuthentication = new RequestAuthentication();
        // 生成TOKEN必填参数
        requestAuthentication.setOperator(generateRequestToken.getOperator()); // Token提供方
        requestAuthentication.setAudience(generateRequestToken.getAudience()); // Token使用方
        requestAuthentication.setIssuer(applicationConfig.getArtifactId()); // Token签发方
        requestAuthentication.setExpireDateTime(DateUtil.parse(generateRequestToken.getExpirationTime(), DatePattern.NORM_DATETIME_PATTERN)); // Token失效时间

        String tokenString = HuaHengJwtUtils.sign(requestAuthentication);
        result.setResult(tokenString);
        Algorithm algorithm = Algorithm.RSA256(new RSA256Key().getPublicKey(), new RSA256Key().getPrivateKey());
        // Reusable verifier instance 可复用的验证实例
        JWTVerifier verifier = JWT.require(algorithm).withIssuer().build();
        DecodedJWT jwt = verifier.verify(tokenString);
        log.info("-------------------------------------REQUEST TOKEN-------------------------------------");
        log.info("API Token:" + tokenString);
        log.info("jwt.getId():" + jwt.getId());
        log.info("jwt.getClaim(operator):" + jwt.getClaim("operator").asString());
        log.info("jwt.getIssuer():" + jwt.getIssuer());
        log.info("jwt.getAudience():" + Arrays.toString(jwt.getAudience().toArray()));
        log.info("jwt.getIssuedAt():" + DateUtil.format(jwt.getIssuedAt(), DatePattern.NORM_DATETIME_PATTERN));
        log.info("jwt.getExpiresAt():" + DateUtil.format(jwt.getExpiresAt(), DatePattern.NORM_DATETIME_PATTERN));
        log.info("-------------------------------------REQUEST TOKEN-------------------------------------");
        return result;
    }
}