新增下载平台的test方法，调用下载调度器的文件
linrf
2 years ago
parent
fd55f7cfae
commit
6e34870174
@ -1,57 +0,0 @@
|
|||||||
package com.docus.server.common.netty.server;
|
|
||||||
|
|
||||||
import io.netty.channel.Channel;
|
|
||||||
import io.netty.util.AttributeKey;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 客户端IP和通信信道的映射
|
|
||||||
*/
|
|
||||||
@Component
|
|
||||||
@Slf4j
|
|
||||||
public class ChannelRepository {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* <collectorId-Channel>
|
|
||||||
*/
|
|
||||||
private final static Map<String, Channel> COLLECTOR_ID_CHANNEL_CACHE_MAP = new ConcurrentHashMap<>();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 终端上线
|
|
||||||
*/
|
|
||||||
public void put(String collectorId, Channel channel) {
|
|
||||||
|
|
||||||
//缓存
|
|
||||||
COLLECTOR_ID_CHANNEL_CACHE_MAP.put(collectorId, channel);
|
|
||||||
|
|
||||||
AttributeKey<String> attributeKey = AttributeKey.valueOf("ip");
|
|
||||||
channel.attr(attributeKey).set(collectorId);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
public String getClientKey(Channel channel) {
|
|
||||||
|
|
||||||
AttributeKey<String> key = AttributeKey.valueOf("ip");
|
|
||||||
|
|
||||||
if (channel.hasAttr(key)) {
|
|
||||||
return channel.attr(key).get();
|
|
||||||
}
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Channel get(String key) {
|
|
||||||
return COLLECTOR_ID_CHANNEL_CACHE_MAP.get(key);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 终端离线
|
|
||||||
*/
|
|
||||||
public void remove(String key) {
|
|
||||||
COLLECTOR_ID_CHANNEL_CACHE_MAP.remove(key);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -0,0 +1,106 @@
|
|||||||
|
package com.docus.server.common.netty.server;
|
||||||
|
|
||||||
|
import com.docus.core.util.json.JSON;
|
||||||
|
import com.docus.server.common.netty.CommMsg;
|
||||||
|
import com.docus.server.common.netty.state.DeviceStateContext;
|
||||||
|
import io.netty.buffer.Unpooled;
|
||||||
|
import io.netty.channel.Channel;
|
||||||
|
import io.netty.channel.ChannelHandlerContext;
|
||||||
|
import io.netty.util.Attribute;
|
||||||
|
import io.netty.util.AttributeKey;
|
||||||
|
import io.netty.util.CharsetUtil;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 采集器id和终端
|
||||||
|
* 通信信道的映射
|
||||||
|
*/
|
||||||
|
@Component
|
||||||
|
@Slf4j
|
||||||
|
public class CollectorChannelCacheMap {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <collectorId-Channel>
|
||||||
|
*/
|
||||||
|
private final static Map<String, Channel> COLLECTOR_ID_CHANNEL_CACHE_MAP = new ConcurrentHashMap<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 采集器上线
|
||||||
|
*/
|
||||||
|
public void put(String collectorId, Channel channel) {
|
||||||
|
|
||||||
|
//缓存
|
||||||
|
COLLECTOR_ID_CHANNEL_CACHE_MAP.put(collectorId, channel);
|
||||||
|
|
||||||
|
AttributeKey<String> attributeKey = AttributeKey.valueOf("collectorId");
|
||||||
|
channel.attr(attributeKey).set(collectorId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据channel获取采集器id
|
||||||
|
*/
|
||||||
|
public String getClientKey(Channel channel) {
|
||||||
|
|
||||||
|
AttributeKey<String> key = AttributeKey.valueOf("collectorId");
|
||||||
|
|
||||||
|
if (channel.hasAttr(key)) {
|
||||||
|
return channel.attr(key).get();
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void writeAndFlush(String collectorId, CommMsg commMsg) {
|
||||||
|
Channel channel = this.get(collectorId);
|
||||||
|
if (null != channel && channel.isOpen()) {
|
||||||
|
channel.writeAndFlush(Unpooled.copiedBuffer(JSON.toJSON(commMsg), CharsetUtil.UTF_8));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Channel get(String key) {
|
||||||
|
return COLLECTOR_ID_CHANNEL_CACHE_MAP.get(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 终端离线
|
||||||
|
*/
|
||||||
|
public void remove(String key) {
|
||||||
|
COLLECTOR_ID_CHANNEL_CACHE_MAP.remove(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置属性
|
||||||
|
*
|
||||||
|
* @param ctx 通道上下文
|
||||||
|
* @param value 值
|
||||||
|
*/
|
||||||
|
public void setAttribute(ChannelHandlerContext ctx, DeviceStateContext value) {
|
||||||
|
Attribute<DeviceStateContext> attribute = getAttribute(ctx);
|
||||||
|
attribute.set(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取属性值
|
||||||
|
*
|
||||||
|
* @param ctx 通道上下文
|
||||||
|
* @return 属性值 有可能为 null
|
||||||
|
*/
|
||||||
|
public DeviceStateContext getAttributeValue(ChannelHandlerContext ctx) {
|
||||||
|
Attribute<DeviceStateContext> attribute = getAttribute(ctx);
|
||||||
|
DeviceStateContext value = attribute.get();
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 永远不会返回 null
|
||||||
|
*
|
||||||
|
* @param ctx channel 上下
|
||||||
|
* @return ctx 关联的 channle 的 attribute
|
||||||
|
*/
|
||||||
|
private Attribute<DeviceStateContext> getAttribute(ChannelHandlerContext ctx) {
|
||||||
|
return ctx.channel().attr(AttributeKey.valueOf("state"));
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,36 +0,0 @@
|
|||||||
//package com.docus.server.api.scheduling.management;
|
|
||||||
//
|
|
||||||
//import io.swagger.annotations.Api;
|
|
||||||
//import io.swagger.annotations.ApiImplicitParam;
|
|
||||||
//import io.swagger.annotations.ApiImplicitParams;
|
|
||||||
//import io.swagger.annotations.ApiOperation;
|
|
||||||
//import org.springframework.cloud.openfeign.FeignClient;
|
|
||||||
//import org.springframework.web.bind.annotation.*;
|
|
||||||
//import org.springframework.web.multipart.MultipartFile;
|
|
||||||
//
|
|
||||||
//import javax.servlet.http.HttpServletResponse;
|
|
||||||
//
|
|
||||||
//
|
|
||||||
///**
|
|
||||||
// * 文件上传下载 API
|
|
||||||
// *
|
|
||||||
// * @author AutoGenerator
|
|
||||||
// * @since 2023-07-15
|
|
||||||
// */
|
|
||||||
//@Api(value = "通用文件上传下载接口", tags = "通用文件上传下载接口")
|
|
||||||
//@FeignClient(value = "collector-scheduling-management", contextId = "collector-scheduling-management.FileApi")
|
|
||||||
//@RequestMapping("/sch/file")
|
|
||||||
//public interface FileApi {
|
|
||||||
//
|
|
||||||
// @ApiOperation("文件下载")
|
|
||||||
// @GetMapping("/download")
|
|
||||||
// void downloadFile(@RequestParam(value = "filePath") String filePath, HttpServletResponse response) throws Exception;
|
|
||||||
//
|
|
||||||
// @ApiOperation("文件上传")
|
|
||||||
// @PostMapping("/upload")
|
|
||||||
// @ApiImplicitParams({
|
|
||||||
// @ApiImplicitParam(name = "files", value = "文件", required = true, dataTypeClass = MultipartFile.class)
|
|
||||||
// })
|
|
||||||
// void uploadFile(@RequestPart MultipartFile[] files, String pathKey) throws Exception;
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
Loading…
Reference in New Issue