143 lines
4.2 KiB
Java
143 lines
4.2 KiB
Java
package com.ecep.contract.ui;
|
||
|
||
import java.util.Locale;
|
||
import java.util.function.BiConsumer;
|
||
import java.util.logging.Level;
|
||
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.beans.BeansException;
|
||
|
||
import com.ecep.contract.Message;
|
||
import com.ecep.contract.MessageHolder;
|
||
import com.ecep.contract.ds.company.service.CompanyService;
|
||
import com.ecep.contract.ds.other.service.EmployeeService;
|
||
import com.ecep.contract.ds.other.service.SysConfService;
|
||
import com.ecep.contract.model.Employee;
|
||
import com.ecep.contract.util.BeanContext;
|
||
import com.ecep.contract.util.DefaultBeanContext;
|
||
|
||
import lombok.Getter;
|
||
import lombok.Setter;
|
||
|
||
/**
|
||
* Tasker基类，提供任务执行和通信的核心功能
|
||
* <p>
|
||
* 该基类实现了任务执行框架，提供了以下核心功能：
|
||
* 1. 任务执行环境（call方法）
|
||
* 2. 消息处理和进度更新机制
|
||
* 3. Spring Bean获取和缓存功能
|
||
* 4. 当前用户和本地化支持
|
||
* <p>
|
||
* <b>主要组件：</b>
|
||
* - messageHandler: 处理消息通知
|
||
* - titleHandler: 处理标题更新
|
||
* - progressHandler: 处理进度更新
|
||
* - propertyHandler: 处理属性更新
|
||
* <p>
|
||
* <b>使用说明：</b>
|
||
* 1. 子类应继承此类并实现execute方法，实现具体业务逻辑
|
||
* 2. 使用updateMessage、updateProgress等方法进行状态反馈
|
||
* 3. 对于WebSocket任务，还应实现WebSocketServerTasker接口
|
||
* 4. 子类通常不需要覆盖setter方法，这些由框架自动设置
|
||
*/
|
||
public abstract class Tasker<T> implements java.util.concurrent.Callable<T>, BeanContext {
|
||
private static final Logger logger = LoggerFactory.getLogger(Tasker.class);
|
||
@Setter
|
||
protected java.util.function.Predicate<Message> messageHandler;
|
||
@Setter
|
||
private java.util.function.Predicate<String> titleHandler;
|
||
@Setter
|
||
private BiConsumer<String, Object> propertyHandler;
|
||
@Setter
|
||
private BiConsumer<Long, Long> progressHandler;
|
||
@Getter
|
||
@Setter
|
||
private Locale locale = Locale.getDefault();
|
||
private BeanContext beanContext;
|
||
@Setter
|
||
private Employee currentUser;
|
||
@Setter
|
||
@Getter
|
||
private boolean cancelled = false;
|
||
|
||
public Tasker() {
|
||
beanContext = new DefaultBeanContext();
|
||
}
|
||
|
||
public SysConfService getConfService() {
|
||
return getCachedBean(SysConfService.class);
|
||
}
|
||
|
||
public CompanyService getCompanyService() {
|
||
return getCachedBean(CompanyService.class);
|
||
}
|
||
|
||
public EmployeeService getEmployeeService() {
|
||
return getCachedBean(EmployeeService.class);
|
||
}
|
||
|
||
public <K> K getBean(Class<K> requiredType) throws BeansException {
|
||
return beanContext.getBean(requiredType);
|
||
}
|
||
|
||
public <K> K getCachedBean(Class<K> requiredType) {
|
||
return beanContext.getCachedBean(requiredType);
|
||
}
|
||
|
||
public Employee getCurrentUser() {
|
||
return currentUser;
|
||
}
|
||
|
||
@Override
|
||
public T call() throws Exception {
|
||
MessageHolderImpl holder = new MessageHolderImpl(this);
|
||
try {
|
||
return execute(holder);
|
||
} catch (Exception e) {
|
||
holder.error(e.getMessage());
|
||
logger.error(e.getMessage(), e);
|
||
throw e;
|
||
}
|
||
}
|
||
|
||
protected abstract T execute(MessageHolder holder) throws Exception;
|
||
|
||
protected void updateMessage(String message) {
|
||
updateMessage(Level.INFO, message);
|
||
}
|
||
|
||
public void updateTitle(String title) {
|
||
if (titleHandler != null) {
|
||
if (titleHandler.test(title)) {
|
||
return;
|
||
}
|
||
}
|
||
updateMessage(Level.INFO, title);
|
||
}
|
||
|
||
protected void updateMessage(Level level, String message) {
|
||
if (messageHandler != null) {
|
||
if (messageHandler.test(new Message(level, message))) {
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
public void updateProgress(long current, long total) {
|
||
if (progressHandler != null) {
|
||
progressHandler.accept(current, total);
|
||
}
|
||
}
|
||
|
||
public void updateProgress(double current, double total) {
|
||
updateProgress((long) (current * 1000), (long) (total * 1000));
|
||
}
|
||
|
||
public void updateProperty(String property, Object value) {
|
||
if (propertyHandler != null) {
|
||
propertyHandler.accept(property, value);
|
||
}
|
||
}
|
||
}
|