package com.huaheng.api.Weighing.server;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.huaheng.api.Weighing.service.BusinessService;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.pc.config.weight.domain.KuaidiWeight;
import com.huaheng.pc.config.weight.service.KuaidiWeightService;
import com.huaheng.pc.shipment.kuaidiHeader.domain.KuaidiHeader;
import com.huaheng.pc.shipment.kuaidiHeader.service.KuaidiHeaderService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.integration.annotation.Filter;
import org.springframework.integration.annotation.Router;
import org.springframework.integration.annotation.ServiceActivator;
import org.springframework.integration.annotation.Transformer;
import org.springframework.integration.dsl.IntegrationFlow;
import org.springframework.integration.dsl.IntegrationFlows;
import org.springframework.integration.ip.dsl.Udp;
import org.springframework.messaging.handler.annotation.Headers;
import org.springframework.messaging.handler.annotation.Payload;
import javax.annotation.Resource;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Map;
/**
* @ClassName UDPServer
* @Description TODO
* @Author Administrator
* @Date 2020/12/215:04
*/
@Configuration
public class UDPServer {
private static final Logger
logger = LoggerFactory.getLogger(UDPServer.class);
/**测试地址**/
// private static final Integer udpPort = 8016;
/**正式地址**/
private static final Integer udpPort = 8800;
@Resource
private BusinessService
businessService;
@Resource
KuaidiWeightService kuaidiWeightService;
@Resource
KuaidiHeaderService kuaidiHeaderService;
/**
* UDP消息接收服务写法一
* https://docs.spring.io/spring-integration/reference/html/ip.html#inbound-udp-adapters-java-configuration
*
* @param
* @return org.springframework.integration.ip.udp.UnicastReceivingChannelAdapter
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/14 11:00
*/
/*@Bean
public UnicastReceivingChannelAdapter unicastReceivingChannelAdapter() {
// 实例化一个UDP消息接收服务
UnicastReceivingChannelAdapter unicastReceivingChannelAdapter = new UnicastReceivingChannelAdapter(udpPort);
// unicastReceivingChannelAdapter.setOutputChannel(new DirectChannel());
unicastReceivingChannelAdapter.setOutputChannelName("udpChannel");
logger.info("UDP服务启动成功，端口号为: {}", udpPort);
return unicastReceivingChannelAdapter;
}*/
/**
* UDP消息接收服务写法二
* https://docs.spring.io/spring-integration/reference/html/ip.html#inbound-udp-adapters-java-dsl-configuration
*
* @param
* @return org.springframework.integration.dsl.IntegrationFlow
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/20 16:08
*/
@Bean
public IntegrationFlow integrationFlow() {
logger.info("UDP服务启动成功，端口号为: {}" + udpPort);
return IntegrationFlows.from(Udp.inboundAdapter(udpPort)).channel("udpChannel").get();
}
/**
* 转换器
*
* @param payload
* @param headers
* @return java.lang.String
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/20 15:30
*/
@Transformer(inputChannel = "udpChannel", outputChannel = "udpFilter")
public String transformer(@Payload byte[] payload, @Headers Map<String, Object> headers) {
String message = new String(payload);
// 转换为大写
// message = message.toUpperCase();
return message;
}
/**
* 过滤器
*
* @param message
* @param headers
* @return boolean
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/20 15:30
*/
@Filter(inputChannel = "udpFilter", outputChannel = "udpRouter")
public boolean filter(String message, @Headers Map<String, Object> headers) {
// 获取来源Id
String id = headers.get("id").toString();
// 获取来源IP，可以进行IP过滤
String ip = headers.get("ip_address").toString();
// 获取来源Port
String port = headers.get("ip_port").toString();
// 信息数据过滤
/*if (message.indexOf("-") < 0) {
// 没有-的数据会被过滤
return false;
}*/
return true;
}
/**
* 路由分发处理器
*
* @param message
* @param headers
* @return java.lang.String
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/20 15:35
*/
@Router(inputChannel = "udpRouter")
public String router(String message, @Headers Map<String, Object> headers) {
// 获取来源Id
String id = headers.get("id").toString();
// 获取来源IP，可以进行IP过滤
String ip = headers.get("ip_address").toString();
// 获取来源Port
String port = headers.get("ip_port").toString();
// 筛选，走那个处理器
// if (false) {
// return "udpHandle2";
// }
return "udpHandle1";
}
/**
* 最终处理器1
*
* @param message
* @return void
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/20 15:12
*/
@ServiceActivator(inputChannel = "udpHandle1")
public void udpMessageHandle(String message, @Headers Map<String, Object> headers) throws Exception {
// 可以进行异步处理
// businessService.udpHandleMethod(message);
logger.info("UDP1：{}" + message);
List<String>
code = Arrays.asList(message.split(","));
String trackCode = code.get(0);
String weight = code.get(1);
KuaidiWeight kuaidiWeight =new KuaidiWeight();
kuaidiWeight.setTrackCode(trackCode);
kuaidiWeight.setWeight(weight);
kuaidiWeight.setCreated(new Date());
kuaidiWeightService.insertWeight(kuaidiWeight);
KuaidiHeader kuaidiHeader = new KuaidiHeader();
kuaidiHeader = kuaidiHeaderService.getOne(new LambdaQueryWrapper<KuaidiHeader>().eq(KuaidiHeader::getTrackCode,trackCode));
if(kuaidiHeader != null){
if(StringUtils.isEmpty(kuaidiHeader.getTotalWeight())) {
kuaidiHeader.setTotalWeight(weight);
}else {
if(kuaidiHeader.getTotalWeight().compareTo(weight)< 0 ) {
kuaidiHeader.setOpenWeight(weight);
}else {
kuaidiHeader.setOpenWeight(kuaidiHeader.getTotalWeight());
kuaidiHeader.setTotalWeight(weight);
}
}
kuaidiHeaderService.updateById(kuaidiHeader);
}else {
KuaidiHeader model = new KuaidiHeader();
model.setWarehouseId(2);
model.setWarehouseCode("KS0001");
model.setTotalWeight(weight);
model.setTrackCode(trackCode);
model.setDeleted(true);
kuaidiHeaderService.save(model);
}
}
/**
* 最终处理器2
*
* @param message
* @return void
* @throws
* @author wliduo[i@dolyw.com]
* @date 2020/5/14 11:02
*/
@ServiceActivator(inputChannel = "udpHandle2")
public void udpMessageHandle2(String message, @Headers Map<String, Object> headers) throws Exception {
logger.info("UDP2:" + message);
System.out.println("UDP2:" + message);
}
}