背景

在開發(fā)中有時會遇到長耗時的任務(wù)，我們又不想傻傻的等待任務(wù)的執(zhí)行完成后才可能進行其他的操作。這時我們首先想到的是使用多線程進行異步處理，那么如何才能知道任務(wù)的執(zhí)行狀態(tài)呢，如執(zhí)行了多長時間，執(zhí)行的結(jié)果是成功了還是失敗了。

此處提到的長耗時任務(wù)是指本身已經(jīng)經(jīng)過優(yōu)化了不能在拆分的情況，如果任務(wù)還可以在拆分請先拆分，不要為難你自己。

如果項目中只是一次性或者很少量是這種情況，怎么著也能忍了；此處面對的是大量的，并且是需要經(jīng)常處理的情況。

如果是定時任務(wù)或者在控制臺執(zhí)行的，就忍著吧，我們總是對自己很殘忍，對他人很友善的。

總之此處要面對的就是項目中有大量的長耗時任務(wù)，需要在頁面上點擊某個按鈕后立即返回，還要知道任務(wù)執(zhí)行的情況。

解決方法

辦法總比問題多，總是有解決的方法的。我用到的技術(shù)主要是自定義注解、AOP、@Async。不太了解的請自行掃盲，此處不做展開。

具體代碼

自定義注解

import java.lang.annotation.*;

/**
 * 后續(xù)操作
 * @author 李慶海
 */
@Documented
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Aftermath {

    /**
     * 運行結(jié)果保存的路徑
     * 相對路徑是指相對prefix返回值而言
     * @return
     */
    String path() default "";

    /**
     * 運行結(jié)果保存的根路徑
     * @return
     */
    String prefix() default "/devops/ydhxg/";
}

善后處理攔截器

import cn.com.yd.cygp4.webinterface.YuandayunHelper;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.cglib.core.ReflectUtils;
import org.springframework.stereotype.Component;
import org.springframework.util.StopWatch;

import java.lang.reflect.Method;

/**
 * 后續(xù)操作攔截器
 *
 * @author 李慶海
 */
@Aspect
@Component
@Slf4j
public class AftermathInterceptor {

    @Pointcut("@annotation(cn.com.yd.cygp4.webinterface.annotations.Aftermath)")
    public void aspect() {

    }

    @Around(value = "aspect()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        log.debug("進入事后處理");
        Object reO = null;
        // 返回目標(biāo)對象
        Object target = point.getTarget();
        String targetName = target.getClass().getName();
        // 返回當(dāng)前連接點簽名
        String methodName = point.getSignature().getName();
        // 獲得參數(shù)列表
        Object[] arguments = point.getArgs();
        Class<?> targetClass = Class.forName(targetName);
        // 獲取參數(shù)類型數(shù)組
        Class<?>[] argTypes = ReflectUtils.getClasses(arguments);
        // 獲取目標(biāo)method,考慮方法的重載等問題
        Method method = targetClass.getDeclaredMethod(methodName, argTypes);
        // 獲取目標(biāo)method上的注解
        Aftermath annotation = method.getAnnotation(Aftermath.class);
        if (null != annotation) {
            String status = null;
            StopWatch clock = new StopWatch();
            //計時開始
            clock.start();
            try {
                reO = point.proceed();
                status = "成功";
            }catch(Exception e){
                status = "失敗";
            }finally {
                clock.stop();
                JSONObject json = new JSONObject();
                //執(zhí)行狀態(tài)
                json.put("status",status);
                // 花費的時間
                json.put("take",clock.getTotalTimeSeconds());
                String path = annotation.path();
                if(null == path || "".equals(path)){
                    path = targetName+"/"+methodName;
                }
                path = annotation.prefix()+path;
                System.out.println("path:"+path);
                System.out.println(json);
                YuandayunHelper.delete(path);
                YuandayunHelper.add(path,json);
            }
        } else {
            reO = point.proceed();
        }
        log.debug("退出事后處理");
        return reO;
    }
}

異步執(zhí)行

開啟異步執(zhí)行

/**
 * Springboot 程序入口
 *
 * @author 李慶海
 */
@SpringBootApplication
@EnableAsync
public class Application {

    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }

    @Bean(name = "threadPoolTaskExecutor")
    public Executor threadPoolTaskExecutor() {
        return new ThreadPoolTaskExecutor();
    }
}

在目標(biāo)方法上添加@Async

@Async
public Object chushihuagupiaogainian() {
    //do something;
    return something;
}

長耗時異步執(zhí)行示例

import cn.com.yd.cygp4.webinterface.annotations.Aftermath;
import org.springframework.scheduling.annotation.Async;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * 龍虎榜手動管理Controller
 *
 * @author 李慶海
 */
@RestController
@RequestMapping("${admin-url}")
@Api(tags = "后臺管理")
public class AdminController {

    /**
     * 初始化
     *
     * @return
     */
    @GetMapping("/init")
    @ApiOperation("后臺管理 > 數(shù)據(jù)初始化")
    @Aftermath
    @Async
    public Object init() {
        //此處耗時5m;
        return something;
    }

最后

這些代碼都是從實際開發(fā)中提煉出來的，與業(yè)務(wù)具有一定的綁定。為了跳出局限使其具有通用性，可以進行一些優(yōu)化，比如將寫入操作抽象一個接口如IWrite，定義一個方法write（String path，String content），然后根據(jù)場景進行實現(xiàn)即可達到通用的目的。關(guān)于執(zhí)行結(jié)果的獲取就更好辦了，可以是監(jiān)聽、可以是輪詢。