SpringBoot基于post转发热更新

背景

线上出现问题想修复,但是不想因为这么一个小问题就重启服务(因为长连接会断开又或者是影响用户使用),希望能不重启服务修复bug。

原理

项目中统一都是用了post请求接口,基本都会有@PostMapping注解。我们在PostMapping做一个切片,所有进入PostMapping的函数都先检查是否有热更代码,如果有就通过Groovy执行热更代码。

实现

package com.meal.system.entity;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateDeserializer;
import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateTimeDeserializer;
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateSerializer;
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
import com.meal.BaseEntity;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.experimental.Accessors;
import org.springframework.format.annotation.DateTimeFormat;
import groovy.lang.Script;

import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.LocalDateTime;


@Data
@ApiModel(value = "热更代码")
@Accessors(chain = true)
@TableName(value = "sys_reload_code")
public class SysReloadCode extends BaseEntity {

    @ApiModelProperty("热更新路径")
    private String uri;

    @ApiModelProperty("执行函数名")
    private String funcName;

    @ApiModelProperty("执行函数")
    private String funcCode;

    //数据库不存在字段------------------------

    @ApiModelProperty(value = "代码",hidden = true)
    @TableField(exist = false)
    private Script script;

}


package com.meal.annotation.appbusiness;

import cn.hutool.core.collection.CollUtil;
import com.meal.common.vo.ResultVo;
import com.meal.constants.Constant;
import com.meal.system.entity.SysDict;
import com.meal.system.entity.SysReloadCode;
import com.meal.system.entity.SysUser;
import com.meal.utils.*;
import groovy.lang.GroovyShell;
import groovy.lang.Script;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.PostMapping;

import javax.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;

@Aspect
@Component
@Slf4j
public class HookAspect {

    public static Map<String, SysReloadCode> stringScriptMap = new ConcurrentHashMap<>();

    public static void reload() {
        List<SysReloadCode> sysReloadCodeList = EntityCrudServiceUtil.list(SysReloadCode.class, null);
        stringScriptMap.clear();
        for(SysReloadCode sysReloadCode : sysReloadCodeList) {
            //创建GroovyShell
            GroovyShell groovyShell = new GroovyShell();
            //装载解析脚本代码
            Script script = groovyShell.parse(sysReloadCode.getFuncCode());
            sysReloadCode.setScript(script);

            stringScriptMap.put(sysReloadCode.getUri(), sysReloadCode);
        }


    }

    private String getMatchKey(String uri) {
        if(ObjectUtil.isEmpty(stringScriptMap))
            return null;
        if (StringUtils.isBlank(uri)) {
            return null;
        }
        Set<String> pathList = stringScriptMap.keySet();

        for (String pattern : pathList) {
            if (UrlUtil.isMatch(pattern, uri)) {
                return pattern;
            }
        }
        return null;
    }

    @Around("@annotation(org.springframework.web.bind.annotation.PostMapping)")
    public Object methodExecution(ProceedingJoinPoint joinPoint) throws Throwable {

        HttpServletRequest request = null;
        try{
            request = ServletUtil.getRequest();
        }catch (Exception e){
            Object[] args = joinPoint.getArgs();
            return joinPoint.proceed(args);
        }

        String uri = request.getRequestURI();
        String key = this.getMatchKey(uri);

        SysReloadCode sysReloadCode = null;
        if(key != null){
            MethodSignature signature = (MethodSignature) joinPoint.getSignature();
            PostMapping annotation = signature.getMethod().getAnnotation(PostMapping.class);
            String path = annotation.value()[0];
            if(key.contains(path)){
                sysReloadCode = stringScriptMap.get(key);
            }

        }

        Object[] args = joinPoint.getArgs();
        if(sysReloadCode == null){

            // 执行目标方法
            return joinPoint.proceed(args);
        } else {
            return sysReloadCode.getScript().invokeMethod(sysReloadCode.getFuncName(), args);
        }

    }


}

使用

把需要热更的代码和对应的路径添加到数据库,然后事先暴露reload接口,开发人员执行reload函数即可热更。如果热更代码有bug,只需要把对应的数据删了重新执行reload即可。

相关推荐
吾日三省吾码4 小时前
JVM 性能调优
java
Estar.Lee4 小时前
查手机号归属地免费API接口教程
android·网络·后端·网络协议·tcp/ip·oneapi
弗拉唐5 小时前
springBoot,mp,ssm整合案例
java·spring boot·mybatis
oi776 小时前
使用itextpdf进行pdf模版填充中文文本时部分字不显示问题
java·服务器
2401_857610036 小时前
SpringBoot社团管理:安全与维护
spring boot·后端·安全
少说多做3436 小时前
Android 不同情况下使用 runOnUiThread
android·java
知兀6 小时前
Java的方法、基本和引用数据类型
java·笔记·黑马程序员
蓝黑20206 小时前
IntelliJ IDEA常用快捷键
java·ide·intellij-idea
Ysjt | 深6 小时前
C++多线程编程入门教程(优质版)
java·开发语言·jvm·c++