Skip to content

谢明辉 / fanxing3

Go to a project
Commit 14e8731e by xmh
Options

fanxing-forward: 

1. 修改 CacheUtils 使用 guava cache
2. 添加是否启用转发的配置
3. 转发结果持久化
4. 转发时3次重发机制

fanxing-task-manager
1. 实现对任务的创建更新和删除动作
2. 修复对scheduling的调用

fanxing-task-scheduling
优化
1 parent dd570351
Inline Side-by-side
Showing 27 changed files with 327 additions and 139 deletions
fanxing-code/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
......
fanxing-commons/pom.xml
...@@ -106,5 +106,10 @@ ...@@ -106,5 +106,10 @@
<artifactId>fastjson</artifactId> <artifactId>fastjson</artifactId>
<version>1.2.76</version> <version>1.2.76</version>
</dependency> </dependency>
<dependency>
<groupId>com.github.ben-manes.caffeine</groupId>
<artifactId>caffeine</artifactId>
<version>3.0.3</version>
</dependency>
</dependencies> </dependencies>
</project> </project>
\ No newline at end of file \ No newline at end of file
fanxing-commons/src/main/java/com/viontech/fanxing/commons/constant/RedisKeys.java
...@@ -29,7 +29,6 @@ public class RedisKeys { ...@@ -29,7 +29,6 @@ public class RedisKeys {
/** 用来接收所有数据的队列,用于转发 */ /** 用来接收所有数据的队列,用于转发 */
public static final String FORWARD_FORWARD_QUEUE = "forward:forwardQueue"; public static final String FORWARD_FORWARD_QUEUE = "forward:forwardQueue";
/** /**
* 用来获取 vaServer 的心跳在 redis 中对应的 key * 用来获取 vaServer 的心跳在 redis 中对应的 key
* *
...@@ -41,5 +40,9 @@ public class RedisKeys { ...@@ -41,5 +40,9 @@ public class RedisKeys {
return devId == null ? "scheduling:keepalive" : "scheduling:keepalive" + ":" + devId; return devId == null ? "scheduling:keepalive" : "scheduling:keepalive" + ":" + devId;
} }
public static String getForwardResultMap(Long forwardId) {
return "forward:result:" + forwardId;
}
} }
fanxing-commons/src/main/java/com/viontech/fanxing/commons/feing/TaskSchedulingTasksAdapter.java
...@@ -16,6 +16,6 @@ public interface TaskSchedulingTasksAdapter { ...@@ -16,6 +16,6 @@ public interface TaskSchedulingTasksAdapter {
JsonMessageUtil.JsonMessage update(Task task); JsonMessageUtil.JsonMessage update(Task task);
JsonMessageUtil.JsonMessage delete(Task task); JsonMessageUtil.JsonMessage delete(String taskUnid);
} }
fanxing-forward/src/main/java/com/viontech/fanxing/forward/batch/writer/ForwardWriter.java
package com.viontech.fanxing.forward.batch.writer; package com.viontech.fanxing.forward.batch.writer;
import com.viontech.fanxing.commons.config.ApplicationContextProvider;
import com.viontech.fanxing.commons.constant.RedisKeys;
import com.viontech.fanxing.commons.model.Forward; import com.viontech.fanxing.commons.model.Forward;
import com.viontech.fanxing.forward.ForwardApp; import com.viontech.fanxing.forward.ForwardApp;
import com.viontech.fanxing.forward.model.ForwardContent; import com.viontech.fanxing.forward.model.ForwardContent;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RMap;
import org.redisson.api.RedissonClient;
import org.springframework.batch.item.ItemWriter; import org.springframework.batch.item.ItemWriter;
import org.springframework.boot.web.client.RestTemplateBuilder;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate; import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
import javax.annotation.Resource; import java.time.Duration;
import java.util.Date;
import java.util.List; import java.util.List;
import java.util.concurrent.Future;
/** /**
* . * .
...@@ -23,21 +26,14 @@ import java.util.concurrent.Future; ...@@ -23,21 +26,14 @@ import java.util.concurrent.Future;
@Component @Component
public class ForwardWriter implements ItemWriter<ForwardContent> { public class ForwardWriter implements ItemWriter<ForwardContent> {
@Resource
private RestTemplateBuilder restTemplateBuilder;
@Override @Override
public void write(List<? extends ForwardContent> items) { public void write(List<? extends ForwardContent> items) {
RestTemplate restTemplate = restTemplateBuilder.build();
// List<ImmutablePair<Future<ResponseEntity<String>>, HttpCallable>> data = new ArrayList<>();
for (ForwardContent item : items) { for (ForwardContent item : items) {
List<Forward> forwardList = item.getForwardList(); List<Forward> forwardList = item.getForwardList();
String json = item.getJson(); String json = item.getJson();
for (Forward forward : forwardList) { for (Forward forward : forwardList) {
HttpCallable httpCallable = new HttpCallable(restTemplate, forward, json); HttpCallable httpCallable = new HttpCallable(forward, json);
ForwardApp.THREAD_POOL_EXECUTOR.submit(httpCallable); ForwardApp.THREAD_POOL_EXECUTOR.submit(httpCallable);
// ImmutablePair<Future<ResponseEntity<String>>, HttpCallable> pair = ImmutablePair.of(future, httpCallable);
// data.add(pair);
} }
} }
...@@ -47,28 +43,44 @@ public class ForwardWriter implements ItemWriter<ForwardContent> { ...@@ -47,28 +43,44 @@ public class ForwardWriter implements ItemWriter<ForwardContent> {
private @Slf4j private @Slf4j
static class HttpCallable implements Runnable { static class HttpCallable implements Runnable {
private final RestTemplate restTemplate;
private final Forward forward; private final Forward forward;
private final String json; private final String json;
private final RedissonClient redissonClient;
private int failed = 0; private int failed = 0;
public HttpCallable(RestTemplate restTemplate, Forward forward, String json) { public HttpCallable(Forward forward, String json) {
this.restTemplate = restTemplate;
this.forward = forward; this.forward = forward;
this.json = json; this.json = json;
this.redissonClient = ApplicationContextProvider.getBean(RedissonClient.class);
} }
@Override @Override
public void run() { public void run() {
try { try {
ResponseEntity<String> responseEntity = restTemplate.postForEntity(forward.getUrl(), json, String.class); Mono<String> response = WebClient.create()
.post()
.uri(forward.getUrl())
.bodyValue(json)
.retrieve()
.bodyToMono(String.class);
String block = response.block(Duration.ofSeconds(20));
} catch (Exception e) { } catch (Exception e) {
failed += 1; failed += 1;
} }
if (failed < 3) { RMap<String, Object> map = redissonClient.getMap(RedisKeys.getForwardResultMap(forward.getId()));
if (failed == 0) {
map.put("id", forward.getId());
map.addAndGet("total", 1);
map.put("lastSendTime", new Date());
} else if (failed < 3) {
ForwardApp.THREAD_POOL_EXECUTOR.submit(this); ForwardApp.THREAD_POOL_EXECUTOR.submit(this);
} else { } else {
log.info("失败次数超过三次,不再发送,forwardId:{}", forward.getId()); log.info("失败次数超过三次,不再发送,forwardId:{}", forward.getId());
map.put("id", forward.getId());
map.addAndGet("total", 1);
map.addAndGet("failed", 1);
map.put("lastSendTime", new Date());
} }
} }
} }
......
fanxing-forward/src/main/java/com/viontech/fanxing/forward/controller/DataReceiveController.java
...@@ -5,8 +5,8 @@ import com.alibaba.fastjson.JSONObject; ...@@ -5,8 +5,8 @@ import com.alibaba.fastjson.JSONObject;
import com.viontech.fanxing.commons.constant.RedisKeys; import com.viontech.fanxing.commons.constant.RedisKeys;
import com.viontech.keliu.util.JsonMessageUtil; import com.viontech.keliu.util.JsonMessageUtil;
import org.redisson.api.RBlockingDeque; import org.redisson.api.RBlockingDeque;
import org.redisson.api.RBlockingQueue;
import org.redisson.api.RedissonClient; import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController; import org.springframework.web.bind.annotation.RestController;
...@@ -24,6 +24,8 @@ import javax.annotation.Resource; ...@@ -24,6 +24,8 @@ import javax.annotation.Resource;
public class DataReceiveController { public class DataReceiveController {
@Resource @Resource
private RedissonClient redissonClient; private RedissonClient redissonClient;
@Value("${vion.forward.enable:false}")
private Boolean enableForward;
@PostMapping("/result") @PostMapping("/result")
public Object result(@RequestBody String analysisResultStr) { public Object result(@RequestBody String analysisResultStr) {
...@@ -49,7 +51,10 @@ public class DataReceiveController { ...@@ -49,7 +51,10 @@ public class DataReceiveController {
if (queue != null) { if (queue != null) {
queue.offerFirst(jsonObject); queue.offerFirst(jsonObject);
} }
// forwardQueue.offer(jsonObject);
if (enableForward) {
forwardQueue.offer(jsonObject);
}
return JsonMessageUtil.getSuccessJsonMsg("OK"); return JsonMessageUtil.getSuccessJsonMsg("OK");
} }
......
fanxing-forward/src/main/java/com/viontech/fanxing/forward/feign/OpsFeignClient.java
...@@ -5,6 +5,9 @@ import com.viontech.keliu.util.JsonMessageUtil; ...@@ -5,6 +5,9 @@ import com.viontech.keliu.util.JsonMessageUtil;
import org.springframework.cloud.openfeign.FeignClient; import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
/** /**
* . * .
...@@ -20,4 +23,7 @@ public interface OpsFeignClient { ...@@ -20,4 +23,7 @@ public interface OpsFeignClient {
@GetMapping("/forwards") @GetMapping("/forwards")
JsonMessageUtil.JsonMessage<Forward> getForwards(); JsonMessageUtil.JsonMessage<Forward> getForwards();
@PostMapping("/forwards/{id}")
JsonMessageUtil.JsonMessage<Forward> updateById(@PathVariable("id") Long id, @RequestBody Forward forward);
} }
fanxing-forward/src/main/java/com/viontech/fanxing/forward/runner/ForwardResultPersistenceRunner.java 0 → 100644
package com.viontech.fanxing.forward.runner;
import com.viontech.fanxing.commons.model.Forward;
import com.viontech.fanxing.forward.feign.OpsFeignClient;
import com.viontech.keliu.util.JsonMessageUtil;
import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RKeys;
import org.redisson.api.RLock;
import org.redisson.api.RMap;
import org.redisson.api.RedissonClient;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.util.Date;
/**
* .
*
* @author 谢明辉
* @date 2021/8/3
*/
@Component
@Slf4j
public class ForwardResultPersistenceRunner {
@Resource
private RedissonClient redissonClient;
@Resource
private OpsFeignClient opsFeignClient;
@Scheduled(cron = "1 0/5 * * * ?")
public void forwardResultPersistenceRunner() {
RLock lock = redissonClient.getLock("lock:forwardResultPersistence");
boolean success = lock.tryLock();
if (!success) {
return;
}
try {
RKeys keys = redissonClient.getKeys();
Iterable<String> forwardResultKeys = keys.getKeysByPattern("forward:result:*");
for (String forwardResultKey : forwardResultKeys) {
Forward forward = new Forward();
RMap<String, Object> map = redissonClient.getMap(forwardResultKey);
Object total = map.get("total");
forward.setTotal((Long) total);
Object lastSendTime = map.get("lastSendTime");
forward.setLastSendTime((Date) lastSendTime);
Object failed = map.get("failed");
forward.setFailed((Long) failed);
Long id = (Long) map.get("id");
JsonMessageUtil.JsonMessage<Forward> forwardJsonMessage = opsFeignClient.updateById(id, forward);
}
} catch (Exception e) {
log.error("ForwardResultPersistenceRunner failed", e);
} finally {
lock.unlock();
}
}
}
fanxing-forward/src/main/java/com/viontech/fanxing/forward/util/CacheUtils.java
package com.viontech.fanxing.forward.util; package com.viontech.fanxing.forward.util;
import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import com.viontech.fanxing.commons.model.Forward; import com.viontech.fanxing.commons.model.Forward;
import com.viontech.fanxing.commons.model.Task; import com.viontech.fanxing.commons.model.Task;
import com.viontech.fanxing.forward.feign.OpsFeignClient; import com.viontech.fanxing.forward.feign.OpsFeignClient;
...@@ -9,10 +11,11 @@ import lombok.extern.slf4j.Slf4j; ...@@ -9,10 +11,11 @@ import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import javax.annotation.Resource; import javax.annotation.Resource;
import java.time.Duration;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ExecutionException;
import java.util.stream.Collectors; import java.util.stream.Collectors;
/** /**
...@@ -21,60 +24,63 @@ import java.util.stream.Collectors; ...@@ -21,60 +24,63 @@ import java.util.stream.Collectors;
* @author 谢明辉 * @author 谢明辉
* @date 2021/7/16 * @date 2021/7/16
*/ */
@SuppressWarnings("unchecked")
@Component @Component
@Slf4j @Slf4j
public class CacheUtils { public class CacheUtils {
private static final ConcurrentHashMap<String, Object> CACHE_MAP = new ConcurrentHashMap<>(); private static final Cache<Object, Object> CACHE = CacheBuilder.newBuilder().expireAfterWrite(Duration.ofMinutes(10)).build();
private static final ConcurrentHashMap<String, Long> EXPIRE_MAP = new ConcurrentHashMap<>();
@Resource @Resource
private TaskManagerFeignClient taskManagerFeignClient; private TaskManagerFeignClient taskManagerFeignClient;
@Resource @Resource
private OpsFeignClient opsFeignClient; private OpsFeignClient opsFeignClient;
public synchronized Map<String, Task> getTaskMap() { public synchronized Map<String, Task> getTaskMap() {
String key = "task_map"; Map<String, Task> result;
Long expire = EXPIRE_MAP.get(key); try {
if (expire == null || expire < System.currentTimeMillis()) { result = (Map<String, Task>) CACHE.get("task_map", () -> {
JsonMessageUtil.JsonMessage<Task> response = null; JsonMessageUtil.JsonMessage<Task> response = null;
try { try {
response = taskManagerFeignClient.getAllTask(); response = taskManagerFeignClient.getAllTask();
} catch (Exception e) { } catch (Exception e) {
log.info("获取 task_map 失败:", e); log.info("获取 task_map 失败:", e);
} }
if (response != null && response.getData() != null) { if (response != null && response.getData() != null) {
List<Task> data = (List<Task>) response.getData(); List<Task> data = (List<Task>) response.getData();
Map<String, Task> taskMap = data.stream().collect(Collectors.toMap(Task::getUnid, x -> x, (x, y) -> x)); return data.stream().collect(Collectors.toMap(Task::getUnid, x -> x, (x, y) -> x));
CACHE_MAP.put(key, taskMap); } else {
} else { return Collections.emptyMap();
CACHE_MAP.put(key, Collections.emptyMap()); }
} });
EXPIRE_MAP.put(key, System.currentTimeMillis()); } catch (ExecutionException e) {
log.error("", e);
result = Collections.emptyMap();
} }
return result;
return (Map<String, Task>) CACHE_MAP.get(key);
} }
public synchronized List<Forward> getAllForward() { public synchronized List<Forward> getAllForward() {
String key = "forward_list"; List<Forward> result;
Long expire = EXPIRE_MAP.get(key); try {
if (expire == null || expire < System.currentTimeMillis()) { result = (List<Forward>) CACHE.get("forward_list", () -> {
JsonMessageUtil.JsonMessage<Forward> response = null; JsonMessageUtil.JsonMessage<Forward> response = null;
try { try {
response = opsFeignClient.getForwards(); response = opsFeignClient.getForwards();
} catch (Exception e) { } catch (Exception e) {
log.info("获取 forward_list 失败:", e); log.info("获取 forward_list 失败:", e);
} }
if (response != null && response.getData() != null) { if (response != null && response.getData() != null) {
List<Forward> data = (List<Forward>) response.getData(); return response.getData();
CACHE_MAP.put(key, data); } else {
} else { return Collections.emptyList();
CACHE_MAP.put(key, Collections.emptyMap()); }
} });
EXPIRE_MAP.put(key, System.currentTimeMillis()); } catch (ExecutionException e) {
log.error("", e);
result = Collections.emptyList();
} }
return result;
return (List<Forward>) CACHE_MAP.get(key);
} }
......
fanxing-forward/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
...@@ -51,6 +54,8 @@ vion: ...@@ -51,6 +54,8 @@ vion:
pic: pic:
path: G:\data path: G:\data
keep: 1 keep: 1
forward:
enable: false
redisson: redisson:
path: F:\myIDEAworkspace\jt\fanxing3\fanxing-commons\src\main\resources\redisson.yml path: F:\myIDEAworkspace\jt\fanxing3\fanxing-commons\src\main\resources\redisson.yml
debug: true debug: true
\ No newline at end of file \ No newline at end of file
fanxing-gateway/src/main/java/com/viontech/fanxing/forward/AuthorizationFilter.java
...@@ -44,7 +44,7 @@ public class AuthorizationFilter implements GlobalFilter { ...@@ -44,7 +44,7 @@ public class AuthorizationFilter implements GlobalFilter {
response.getHeaders().setContentType(MediaType.APPLICATION_JSON); response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
response.setStatusCode(HttpStatus.UNAUTHORIZED); response.setStatusCode(HttpStatus.UNAUTHORIZED);
return response.writeWith(Flux.just(msg).map(x -> { return response.writeWith(Mono.just(msg).map(x -> {
byte[] bytes = x.getBytes(StandardCharsets.UTF_8); byte[] bytes = x.getBytes(StandardCharsets.UTF_8);
return response.bufferFactory().wrap(bytes); return response.bufferFactory().wrap(bytes);
})); }));
......
fanxing-gateway/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
gateway: gateway:
enabled: true
globalcors: globalcors:
cors-configurations: cors-configurations:
'[/**]': '[/**]':
allow-credentials: true
allowedOrigins: "*" allowedOrigins: "*"
allowedMethods: allowedMethods: "*"
- ALL allowedHeaders: "*"
discovery: discovery:
locator: locator:
enabled: true enabled: true
......
fanxing-ops/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
......
fanxing-query/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
......
fanxing-task-manager/src/main/java/com/viontech/fanxing/task/manager/controller/web/TaskController.java
...@@ -6,10 +6,7 @@ import com.viontech.fanxing.commons.vo.TaskVo; ...@@ -6,10 +6,7 @@ import com.viontech.fanxing.commons.vo.TaskVo;
import com.viontech.fanxing.task.manager.controller.base.TaskBaseController; import com.viontech.fanxing.task.manager.controller.base.TaskBaseController;
import com.viontech.keliu.util.JsonMessageUtil; import com.viontech.keliu.util.JsonMessageUtil;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.*;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController @RestController
@RequestMapping("/tasks") @RequestMapping("/tasks")
...@@ -31,4 +28,18 @@ public class TaskController extends TaskBaseController { ...@@ -31,4 +28,18 @@ public class TaskController extends TaskBaseController {
return JsonMessageUtil.getSuccessJsonMsg(taskVo); return JsonMessageUtil.getSuccessJsonMsg(taskVo);
} }
@Override
@PostMapping("/{id}")
public Object update(@PathVariable(value = "id") Long id, @RequestBody TaskVo taskVo) {
taskVo.setId(id);
taskVo = taskService.updateTask(taskVo);
return JsonMessageUtil.getSuccessJsonMsg("success", taskVo);
}
@Override
@DeleteMapping("/{id}")
public Object del(@PathVariable(value = "id") Long id) {
taskService.removeTask(id);
return JsonMessageUtil.getSuccessJsonMsg("success");
}
} }
\ No newline at end of file \ No newline at end of file
fanxing-task-manager/src/main/java/com/viontech/fanxing/task/manager/feign/TaskSchedulingClient.java
...@@ -4,11 +4,8 @@ import com.viontech.fanxing.commons.feing.TaskSchedulingTasksAdapter; ...@@ -4,11 +4,8 @@ import com.viontech.fanxing.commons.feing.TaskSchedulingTasksAdapter;
import com.viontech.fanxing.commons.model.Task; import com.viontech.fanxing.commons.model.Task;
import com.viontech.keliu.util.JsonMessageUtil; import com.viontech.keliu.util.JsonMessageUtil;
import org.springframework.cloud.openfeign.FeignClient; import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.DeleteMapping; import org.springframework.web.bind.annotation.*;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
/** /**
* . * .
...@@ -17,19 +14,19 @@ import org.springframework.web.bind.annotation.RequestBody; ...@@ -17,19 +14,19 @@ import org.springframework.web.bind.annotation.RequestBody;
* @date 2021/7/12 * @date 2021/7/12
*/ */
@Service @Component
@FeignClient(value = "fanxing-task-scheduling") @FeignClient(value = "fanxing-task-scheduling")
public interface TaskSchedulingClient extends TaskSchedulingTasksAdapter { public interface TaskSchedulingClient extends TaskSchedulingTasksAdapter {
@Override @Override
@PostMapping @PostMapping("/tasks")
JsonMessageUtil.JsonMessage add(@RequestBody Task task); JsonMessageUtil.JsonMessage add(@RequestBody Task task);
@Override @Override
@PutMapping @PutMapping("/tasks")
JsonMessageUtil.JsonMessage update(@RequestBody Task task); JsonMessageUtil.JsonMessage update(@RequestBody Task task);
@Override @Override
@DeleteMapping @DeleteMapping("/tasks")
JsonMessageUtil.JsonMessage delete(@RequestBody Task task); JsonMessageUtil.JsonMessage delete(@RequestParam("taskUnid") String taskUnid);
} }
fanxing-task-manager/src/main/java/com/viontech/fanxing/task/manager/service/adapter/TaskService.java
...@@ -6,4 +6,8 @@ import com.viontech.fanxing.commons.vo.TaskVo; ...@@ -6,4 +6,8 @@ import com.viontech.fanxing.commons.vo.TaskVo;
public interface TaskService extends BaseService<Task> { public interface TaskService extends BaseService<Task> {
TaskVo addTask(Task task); TaskVo addTask(Task task);
TaskVo updateTask(Task task);
void removeTask(Long id);
} }
\ No newline at end of file \ No newline at end of file
fanxing-task-manager/src/main/java/com/viontech/fanxing/task/manager/service/impl/TaskServiceImpl.java
...@@ -29,11 +29,40 @@ public class TaskServiceImpl extends BaseServiceImpl<Task> implements TaskServic ...@@ -29,11 +29,40 @@ public class TaskServiceImpl extends BaseServiceImpl<Task> implements TaskServic
@Override @Override
public TaskVo addTask(Task task) { public TaskVo addTask(Task task) {
task = insertSelective(task); task = insertSelective(task);
task = selectByPrimaryKey(task.getId());
JsonMessageUtil.JsonMessage add = taskSchedulingClient.add(task); JsonMessageUtil.JsonMessage add = taskSchedulingClient.add(task);
if (add.isSuccess()) { if (add.isSuccess()) {
return new TaskVo(task); return new TaskVo(task);
} else { } else {
throw new RuntimeException("添加任务失败"); throw new RuntimeException(add.getMsg());
}
}
@Transactional(rollbackFor = Exception.class)
@Override
public TaskVo updateTask(Task task) {
task = insertSelective(task);
task = selectByPrimaryKey(task.getId());
JsonMessageUtil.JsonMessage update = taskSchedulingClient.update(task);
if (update.isSuccess()) {
updateByPrimaryKey(task);
return new TaskVo(task);
} else {
throw new RuntimeException(update.getMsg());
}
}
@Override
@Transactional(rollbackFor = Exception.class)
public void removeTask(Long id) {
Task task = selectByPrimaryKey(id);
JsonMessageUtil.JsonMessage delete = taskSchedulingClient.delete(task.getUnid());
if (delete.isSuccess()) {
deleteByPrimaryKey(id);
} else {
throw new RuntimeException(delete.getMsg());
} }
} }
} }
\ No newline at end of file \ No newline at end of file
fanxing-task-manager/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/controller/TaskController.java
...@@ -35,7 +35,7 @@ public class TaskController implements TaskSchedulingTasksAdapter { ...@@ -35,7 +35,7 @@ public class TaskController implements TaskSchedulingTasksAdapter {
@Override @Override
@PostMapping @PostMapping
public JsonMessageUtil.JsonMessage add(Task task) { public JsonMessageUtil.JsonMessage add(@RequestBody Task task) {
TaskData taskData = new TaskData(task); TaskData taskData = new TaskData(task);
// 获取存储配置 // 获取存储配置
Long storeConfigId = task.getStoreConfigId(); Long storeConfigId = task.getStoreConfigId();
...@@ -53,7 +53,9 @@ public class TaskController implements TaskSchedulingTasksAdapter { ...@@ -53,7 +53,9 @@ public class TaskController implements TaskSchedulingTasksAdapter {
Long nextTerminateTime = nextTime.right; Long nextTerminateTime = nextTime.right;
if (nextExecuteTime != null) { if (nextExecuteTime != null) {
taskService.addExecutableTaskData(unid, nextExecuteTime); taskService.addExecutableTaskData(unid, nextExecuteTime);
taskService.addTerminatableTaskData(unid, nextTerminateTime); if (nextTerminateTime != null) {
taskService.addTerminatableTaskData(unid, nextTerminateTime);
}
taskService.addTaskData(taskData); taskService.addTaskData(taskData);
return JsonMessageUtil.getSuccessJsonMsg("success"); return JsonMessageUtil.getSuccessJsonMsg("success");
} else { } else {
...@@ -63,13 +65,13 @@ public class TaskController implements TaskSchedulingTasksAdapter { ...@@ -63,13 +65,13 @@ public class TaskController implements TaskSchedulingTasksAdapter {
@Override @Override
@PutMapping @PutMapping
public JsonMessageUtil.JsonMessage update(Task task) { public JsonMessageUtil.JsonMessage update(@RequestBody Task task) {
String taskUnid = task.getUnid(); String taskUnid = task.getUnid();
VaServerInfo vaServerInfo = taskService.taskRunOn(taskUnid); VaServerInfo vaServerInfo = taskService.taskRunOn(taskUnid);
// vaServerId 为空说明任务未执行可以先删除再建立新任务 // vaServerId 为空说明任务未执行可以先删除再建立新任务
if (vaServerInfo == null) { if (vaServerInfo == null) {
JsonMessageUtil.JsonMessage delete = delete(task); JsonMessageUtil.JsonMessage delete = delete(taskUnid);
if (delete.isSuccess()) { if (delete.isSuccess()) {
JsonMessageUtil.JsonMessage add = add(task); JsonMessageUtil.JsonMessage add = add(task);
return add; return add;
...@@ -84,8 +86,7 @@ public class TaskController implements TaskSchedulingTasksAdapter { ...@@ -84,8 +86,7 @@ public class TaskController implements TaskSchedulingTasksAdapter {
@Override @Override
@DeleteMapping @DeleteMapping
public JsonMessageUtil.JsonMessage delete(Task task) { public JsonMessageUtil.JsonMessage delete(@RequestParam("taskUnid") String taskUnid) {
String taskUnid = task.getUnid();
vaServerService.terminateTask(taskUnid); vaServerService.terminateTask(taskUnid);
taskService.removeTaskDataAll(taskUnid); taskService.removeTaskDataAll(taskUnid);
// todo // todo
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/controller/VAServerController.java
...@@ -9,6 +9,7 @@ import org.springframework.beans.factory.annotation.Value; ...@@ -9,6 +9,7 @@ import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource; import javax.annotation.Resource;
import java.util.Collection;
import java.util.HashMap; import java.util.HashMap;
/** /**
...@@ -44,6 +45,7 @@ public class VAServerController { ...@@ -44,6 +45,7 @@ public class VAServerController {
result.put("code", 200); result.put("code", 200);
result.put("msg", "success"); result.put("msg", "success");
result.put("resultRecvUrl", "http://" + vionGatewayIp + ":" + vionGatewayPort + "/fanxing-forward/result"); result.put("resultRecvUrl", "http://" + vionGatewayIp + ":" + vionGatewayPort + "/fanxing-forward/result");
result.put("videoUploadUrl", "http://" + vionGatewayIp + ":" + vionGatewayPort + "/fanxing-forward/result");
return result; return result;
} }
...@@ -59,7 +61,8 @@ public class VAServerController { ...@@ -59,7 +61,8 @@ public class VAServerController {
@GetMapping("/vaServerInfo") @GetMapping("/vaServerInfo")
public Object getVaServerInfo() { public Object getVaServerInfo() {
return vaServerService.getVaServerInfo(); Collection<VaServerInfo> vaServerInfo = vaServerService.getVaServerInfo();
return JsonMessageUtil.getSuccessJsonMsg(vaServerInfo);
} }
/** /**
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/model/RuntimeConfig.java
...@@ -116,7 +116,7 @@ public class RuntimeConfig { ...@@ -116,7 +116,7 @@ public class RuntimeConfig {
executeTime = singleConfig.getStartTime(); executeTime = singleConfig.getStartTime();
terminateTime = singleConfig.getEndTime(); terminateTime = singleConfig.getEndTime();
if (terminateTime.isBefore(now)) { if (terminateTime.isBefore(now)) {
return null; return ImmutablePair.nullPair();
} }
executeTimestamp = executeTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli(); executeTimestamp = executeTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
terminateTimestamp = terminateTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli(); terminateTimestamp = terminateTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
...@@ -145,14 +145,14 @@ public class RuntimeConfig { ...@@ -145,14 +145,14 @@ public class RuntimeConfig {
dayOfWeek = (dayOfWeek + 1) % 7; dayOfWeek = (dayOfWeek + 1) % 7;
} }
if (!success) { if (!success) {
return null; return ImmutablePair.nullPair();
} }
break; break;
case 3: case 3:
// 随机执行,待完善 // 随机执行,待完善
break; break;
default: default:
return null; return ImmutablePair.nullPair();
} }
return ImmutablePair.of(executeTimestamp, terminateTimestamp); return ImmutablePair.of(executeTimestamp, terminateTimestamp);
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/model/vaserver/VaServerInfo.java
...@@ -23,6 +23,8 @@ public class VaServerInfo implements Serializable { ...@@ -23,6 +23,8 @@ public class VaServerInfo implements Serializable {
private String proxy; private String proxy;
/** 可用资源数量 */ /** 可用资源数量 */
private Float availableResources; private Float availableResources;
/** 状态 1在线,2离线 */
private int status = 1;
public VaServerInfo setVideoResource(Float videoResource) { public VaServerInfo setVideoResource(Float videoResource) {
this.videoResource = videoResource; this.videoResource = videoResource;
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/runner/TaskRunner.java
package com.viontech.fanxing.task.scheduling.runner; package com.viontech.fanxing.task.scheduling.runner;
import com.viontech.fanxing.commons.constant.TaskStatus;
import com.viontech.fanxing.commons.model.Task;
import com.viontech.fanxing.commons.vo.TaskVo;
import com.viontech.fanxing.task.scheduling.feign.TaskClient; import com.viontech.fanxing.task.scheduling.feign.TaskClient;
import com.viontech.fanxing.task.scheduling.model.RuntimeConfig;
import com.viontech.fanxing.task.scheduling.model.TaskData; import com.viontech.fanxing.task.scheduling.model.TaskData;
import com.viontech.fanxing.task.scheduling.model.vaserver.VaServerInfo; import com.viontech.fanxing.task.scheduling.model.vaserver.VaServerInfo;
import com.viontech.fanxing.task.scheduling.service.RedisCacheService; import com.viontech.fanxing.task.scheduling.service.RedisCacheService;
import com.viontech.fanxing.task.scheduling.service.TaskService; import com.viontech.fanxing.task.scheduling.service.TaskService;
import com.viontech.fanxing.task.scheduling.service.VAServerService; import com.viontech.fanxing.task.scheduling.service.VAServerService;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.redisson.api.RLock; import org.redisson.api.RLock;
import org.redisson.api.RMap; import org.redisson.api.RMap;
import org.redisson.api.RScoredSortedSet; import org.redisson.api.RScoredSortedSet;
...@@ -42,10 +47,10 @@ public class TaskRunner { ...@@ -42,10 +47,10 @@ public class TaskRunner {
@Scheduled(fixedDelay = 5000) @Scheduled(fixedDelay = 5000)
public void executedTaskListener() { public void executedTaskListener() {
RLock lock = redissonClient.getLock("executedTaskListener"); RLock lock = redissonClient.getLock("lock:executedTaskListener");
boolean isLock; boolean isLock;
try { try {
isLock = lock.tryLock(10, 8, TimeUnit.SECONDS); isLock = lock.tryLock(30, 25, TimeUnit.SECONDS);
} catch (InterruptedException e) { } catch (InterruptedException e) {
return; return;
} }
...@@ -58,35 +63,35 @@ public class TaskRunner { ...@@ -58,35 +63,35 @@ public class TaskRunner {
Collection<String> entryCollection = set.valueRange(0, true, System.currentTimeMillis(), true); Collection<String> entryCollection = set.valueRange(0, true, System.currentTimeMillis(), true);
for (String taskUnid : entryCollection) { for (String taskUnid : entryCollection) {
log.info("开始任务 : {}", taskUnid); log.info("开始任务 : {}", taskUnid);
// TaskData taskData = taskDataMap.get(taskUnid); TaskData taskData = taskDataMap.get(taskUnid);
// Task task = taskData.getTask(); Task task = taskData.getTask();
// String taskVaType = task.getVaType(); String taskVaType = task.getVaType();
// Float resourceNeed = task.getResourceNeed(); Float resourceNeed = task.getResourceNeed();
// Collection<VaServerInfo> vaServerInfos = vaServerMap.values(); Collection<VaServerInfo> vaServerInfos = vaServerMap.values();
// // todo 暂时先找有可用资源的vaserver,以后再进行算法优化 // todo 暂时先找有可用资源的vaserver,以后再进行算法优化
// VaServerInfo server = null; VaServerInfo server = null;
// for (VaServerInfo vaServerInfo : vaServerInfos) { for (VaServerInfo vaServerInfo : vaServerInfos) {
// if (vaServerInfo.getAvailableResources() > resourceNeed) { if (vaServerInfo.getAvailableResources() > resourceNeed) {
// server = vaServerInfo; server = vaServerInfo;
// break; break;
// } }
// } }
//
// // 找不到可以用来执行的设备,需要修改状态 // 找不到可以用来执行的设备,需要修改状态
// if (server == null) { if (server == null) {
// TaskVo taskVo = new TaskVo(); TaskVo taskVo = new TaskVo();
// taskVo.setStatus(TaskStatus.CAN_NOT_RUN.val); taskVo.setStatus(TaskStatus.CAN_NOT_RUN.val);
// taskClient.updateTask(task.getId(), taskVo); taskClient.updateTask(task.getId(), taskVo);
// continue; continue;
// } }
//
// boolean success = vaServerService.executeTask(taskData, server); boolean success = vaServerService.executeTask(taskData, server);
//
// // 修改任务状态 // 修改任务状态
// TaskVo taskVo = new TaskVo(); TaskVo taskVo = new TaskVo();
// taskVo.setStatus(TaskStatus.RUNNING.val); taskVo.setStatus(TaskStatus.RUNNING.val);
// taskClient.updateTask(task.getId(), taskVo); taskClient.updateTask(task.getId(), taskVo);
// // 移除任务 // 移除任务
set.remove(taskUnid); set.remove(taskUnid);
} }
...@@ -104,10 +109,10 @@ public class TaskRunner { ...@@ -104,10 +109,10 @@ public class TaskRunner {
@Scheduled(fixedDelay = 5000) @Scheduled(fixedDelay = 5000)
public void terminatedTaskListener() { public void terminatedTaskListener() {
RLock lock = redissonClient.getLock("terminatedTaskListener"); RLock lock = redissonClient.getLock("lock:terminatedTaskListener");
boolean isLock; boolean isLock;
try { try {
isLock = lock.tryLock(10, 8, TimeUnit.SECONDS); isLock = lock.tryLock(30, 25, TimeUnit.SECONDS);
} catch (InterruptedException e) { } catch (InterruptedException e) {
return; return;
} }
...@@ -120,15 +125,19 @@ public class TaskRunner { ...@@ -120,15 +125,19 @@ public class TaskRunner {
for (String taskUnid : entryCollection) { for (String taskUnid : entryCollection) {
log.info("停止任务 : {}", taskUnid); log.info("停止任务 : {}", taskUnid);
// TaskData taskData = taskDataMap.get(taskUnid); TaskData taskData = taskDataMap.get(taskUnid);
// RuntimeConfig runtimeConfig = taskData.getRuntimeConfig(); RuntimeConfig runtimeConfig = taskData.getRuntimeConfig();
// boolean success = vaServerService.terminateTask(taskUnid); // 获取可用的 vaserver ,执行任务终止动作,如果成功,解除 taskData和 vaServer 的关联,并且从 zset 中移除任务,恢复vaserver资源数,计算下次任务执行时间,放入zset中
// // todo 获取可用的 vaserver ,执行任务终止动作,如果成功,解除 taskData和 vaServer 的关联,并且从 zset 中移除任务,恢复vaserver资源数,计算下次任务执行时间,放入zset中 boolean success = vaServerService.terminateTask(taskUnid);
// if (success) { if (success) {
// ImmutablePair<Long, Long> nextTime = runtimeConfig.getNextTimeOfExecutionAndTerminal(); ImmutablePair<Long, Long> nextTime = runtimeConfig.getNextTimeOfExecutionAndTerminal();
// taskService.addExecutableTaskData(taskUnid, nextTime.left); Long nextExecuteTime = nextTime.left;
// taskService.addTerminatableTaskData(taskUnid, nextTime.right); Long nextTerminateTime = nextTime.right;
// } if (nextExecuteTime != null) {
taskService.addExecutableTaskData(taskUnid, nextExecuteTime);
taskService.addTerminatableTaskData(taskUnid, nextTerminateTime);
}
}
set.remove(taskUnid); set.remove(taskUnid);
} }
} catch (Exception e) { } catch (Exception e) {
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/service/TaskService.java
...@@ -75,10 +75,6 @@ public class TaskService { ...@@ -75,10 +75,6 @@ public class TaskService {
return ImmutablePair.of(taskUnid, devId); return ImmutablePair.of(taskUnid, devId);
} }
public void linkTaskAndVaServer(String taskUnid, String devId) {
RMap<String, String> map = redisCacheService.getTaskVaServerMap();
map.put(taskUnid, devId);
}
public void updateTask(Task task) { public void updateTask(Task task) {
RMap<String, TaskData> taskDataMap = redisCacheService.getTaskDataMap(); RMap<String, TaskData> taskDataMap = redisCacheService.getTaskDataMap();
......
fanxing-task-scheduling/src/main/java/com/viontech/fanxing/task/scheduling/service/VAServerService.java
...@@ -10,7 +10,9 @@ import org.redisson.api.RMap; ...@@ -10,7 +10,9 @@ import org.redisson.api.RMap;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import javax.annotation.Resource; import javax.annotation.Resource;
import java.util.Collection;
import java.util.Date; import java.util.Date;
import java.util.Map;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
/** /**
...@@ -47,8 +49,17 @@ public class VAServerService { ...@@ -47,8 +49,17 @@ public class VAServerService {
return map.get(devId); return map.get(devId);
} }
public Object getVaServerInfo() { public Collection<VaServerInfo> getVaServerInfo() {
RMap<String, VaServerInfo> map = redisCacheService.getVaServerMap(); RMap<String, VaServerInfo> map = redisCacheService.getVaServerMap();
for (Map.Entry<String, VaServerInfo> entry : map.entrySet()) {
String devId = entry.getKey();
RBucket<Date> bucket = redisCacheService.getValue(RedisKeys.getVAServerKeepAliveKey(devId));
if (!bucket.isExists()) {
VaServerInfo value = entry.getValue();
value.setStatus(0);
map.put(devId, value);
}
}
return map.values(); return map.values();
} }
...@@ -67,11 +78,12 @@ public class VAServerService { ...@@ -67,11 +78,12 @@ public class VAServerService {
*/ */
public boolean executeTask(TaskData taskData, VaServerInfo server) { public boolean executeTask(TaskData taskData, VaServerInfo server) {
Task task = taskData.getTask(); Task task = taskData.getTask();
// todo 执行任务下发,成功后关联任务和vaServer,减少vaServer的可用资源数量 // 执行任务下发,成功后关联任务和vaServer,减少vaServer的可用资源数量
vaServerHttpService.addTask(taskData, server); vaServerHttpService.addTask(taskData, server);
taskService.linkTaskAndVaServer(task.getUnid(), server.getDevID()); RMap<String, String> map = redisCacheService.getTaskVaServerMap();
map.put(task.getUnid(), server.getDevID());
server.setAvailableResources(server.getAvailableResources() - task.getResourceNeed()); server.setAvailableResources(server.getAvailableResources() - task.getResourceNeed());
RMap<String, VaServerInfo> vaServerMap = redisCacheService.getVaServerMap(); RMap<String, VaServerInfo> vaServerMap = redisCacheService.getVaServerMap();
vaServerMap.put(server.getDevID(), server); vaServerMap.put(server.getDevID(), server);
......
fanxing-task-scheduling/src/main/resources/application.yml
spring: spring:
cloud: cloud:
loadbalancer:
ribbon:
enabled: false
consul: consul:
# 服务发现配置 # 服务发现配置
discovery: discovery:
......
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!