From f62c9fcd0acfa540ef74d5259324dfd3ceb0ad2d Mon Sep 17 00:00:00 2001 From: whai Date: Sun, 24 Nov 2024 11:05:52 +0800 Subject: [PATCH] =?UTF-8?q?feat(springdemo):=20=E6=B7=BB=E5=8A=A0=20UDP?= =?UTF-8?q?=E9=80=9A=E4=BF=A1=E7=9B=B8=E5=85=B3=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 TargetDown、TargetImage 和 TargetStorage 类实现 ProcessStrategy 接口 - 添加 UDPConfig 配置类，实现 UDP 消息接收和发送功能 - 新增 UDPSendUtils 工具类，用于发送 UDP 消息 - 实现了 UDP 消息的解密和处理逻辑 --- .../controller/TS/UDP/UDPConfig.java | 197 ++++++++++++++++++ .../controller/TS/UDP/UDPSendUtils.java | 30 +++ .../controller/TS/common/TargetDown.java | 27 +++ .../controller/TS/common/TargetImage.java | 78 +++++++ .../controller/TS/common/TargetStorage.java | 26 +++ 5 files changed, 358 insertions(+) create mode 100644 SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPConfig.java create mode 100644 SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPSendUtils.java create mode 100644 SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetDown.java create mode 100644 SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetImage.java create mode 100644 SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetStorage.java diff --git a/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPConfig.java b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPConfig.java new file mode 100644 index 0000000..7213d51 --- /dev/null +++ b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPConfig.java @@ -0,0 +1,197 @@ +package cn.whaifree.springdemo.controller.TS.UDP; + +import cn.whaifree.springdemo.controller.TS.common.ProcessStrategy; +import cn.whaifree.springdemo.controller.TS.common.ProcessTarget; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Qualifier; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.integration.annotation.BridgeFrom; +import org.springframework.integration.annotation.Transformer; +import org.springframework.integration.channel.DirectChannel; +import org.springframework.integration.channel.PublishSubscribeChannel; +import org.springframework.integration.dispatcher.LoadBalancingStrategy; +import org.springframework.integration.dsl.*; +import org.springframework.integration.handler.MessageProcessor; +import org.springframework.integration.ip.udp.UnicastReceivingChannelAdapter; +import org.springframework.integration.ip.udp.UnicastSendingMessageHandler; +import org.springframework.integration.transformer.AbstractMessageProcessingTransformer; +import org.springframework.messaging.Message; +import org.springframework.messaging.MessageChannel; +import org.springframework.messaging.MessageHandler; +import org.springframework.messaging.MessagingException; +import org.springframework.messaging.handler.annotation.Headers; +import org.springframework.messaging.handler.annotation.Payload; +import org.springframework.stereotype.Component; + +import javax.crypto.Cipher; +import javax.crypto.NoSuchPaddingException; +import javax.crypto.spec.SecretKeySpec; +import java.security.NoSuchAlgorithmException; +import java.util.Base64; +import java.util.Map; + + +/** + * @version 1.0 + * @Author whai文海 + * @Date 2024/11/22 20:32 + * @注释 + */ +@Configuration +public class UDPConfig { + + static final String encoderNumber = "123"; + + /* + + org.springframework.boot + spring-boot-starter-integration + + + org.springframework.integration + spring-integration-ip + + +*/ + + /** + * 接收 + * + * @param udpClient + * @return + */ + @Bean + public IntegrationFlow processUniCastUdpMessage(@Qualifier("UDPGet") MessageHandler udpClient) { + UnicastReceivingChannelAdapter channelAdapter = new UnicastReceivingChannelAdapter(9030); + channelAdapter.setReceiveBufferSize(4096); + channelAdapter.setLengthCheck(false); + return IntegrationFlow + .from(channelAdapter) + .handle(udpClient) +// .transform(this, "encoderTransformer") +// .channel("udpChannel") + .get(); + } + + class Encryptor { + +// public static String encrypt(Object msg, String key) { +// try { +// SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(), "AES"); +// Cipher cipher = Cipher.getInstance("AES"); +// cipher.init(Cipher.ENCRYPT_MODE, secretKey); +// byte[] encrypted = cipher.doFinal((byte[]) msg); +// return Base64.getEncoder().encodeToString(encrypted); +// } catch (Exception e) { +// throw new RuntimeException("Encryption failed", e); +// } +// } +// +// public static byte[] decrypt(byte[] encryptedMsg, String key) { +// try { +// SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(), "AES"); +// Cipher cipher = Cipher.getInstance("AES"); +// cipher.init(Cipher.DECRYPT_MODE, secretKey); +// return cipher.doFinal(Base64.getDecoder().decode(encryptedMsg)); +// } catch (Exception e) { +// throw new RuntimeException(e); +// } +// } + + private static final String ALGORITHM = "AES"; + private static byte[] SECRET_KEY = null; // 替换为你自己的密钥，密钥长度必须符合算法要求 + + public static byte[] encrypt(byte[] data) throws Exception { + Cipher cipher = Cipher.getInstance(ALGORITHM); + SecretKeySpec secretKey = new SecretKeySpec(SECRET_KEY, ALGORITHM); + cipher.init(Cipher.ENCRYPT_MODE, secretKey); + return cipher.doFinal(data); + } + + public static byte[] decrypt(byte[] encryptedData) throws Exception { + Cipher cipher = Cipher.getInstance(ALGORITHM); + SecretKeySpec secretKey = new SecretKeySpec(SECRET_KEY, ALGORITHM); + cipher.init(Cipher.DECRYPT_MODE, secretKey); + return cipher.doFinal(encryptedData); + } + + } + +// +// /** +// * 转换器 在消息通道（Channel）之间传递消息时进行数据格式的转换 +// * +// * @param payload +// * @param headers +// */ +// @Transformer(inputChannel = "channelAdapter", outputChannel = "udpChannel") +// public byte[] encoderTransformer(@Payload byte[] payload, @Headers Map headers) { +// return Encryptor.decrypt(payload, encoderNumber); +// } + +// @Bean("udpChannel") +// public MessageChannel udpChannel() { +// return new DirectChannel(); +// } + + +// /** +// * 一对一 +// * +// * @return +// */ +// @Bean +// @BridgeFrom("messageChannel2") +// public MessageChannel directChannel2() { +// return new DirectChannel(); +// } +// +// /** +// * 一对一 +// * +// * @return +// */ +// @Bean +// @BridgeFrom("messageChannel2") +// public MessageChannel directChannel() { +// return MessageChannels.direct().getObject(); +// } +// +// /** +// * 发布订阅 一对多 +// */ +// @Bean +// public MessageChannel messageChannel2() { +// return MessageChannels.publishSubscribe().getObject(); +// } + + + /** + * 发送 + * + * @return + */ + @Bean + public UnicastSendingMessageHandler sending() { + return new UnicastSendingMessageHandler("localhost", 9030, false); + } + +} + + +@Component("UDPGet") +@Slf4j +class UdpGet implements MessageHandler { + @Override + public void handleMessage(Message message) throws MessagingException { + byte[] frame = (byte[]) message.getPayload(); + StringBuilder result = new StringBuilder("16进制表示："); + for (byte aByte : frame) { + result.append(String.format("%02x ", aByte)); + } + log.info(result.toString()); + ProcessStrategy processStrategy = ProcessTarget.getProcessStrategy(frame); + processStrategy.process(frame); + } +} diff --git a/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPSendUtils.java b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPSendUtils.java new file mode 100644 index 0000000..6e52a47 --- /dev/null +++ b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/UDP/UDPSendUtils.java @@ -0,0 +1,30 @@ +package cn.whaifree.springdemo.controller.TS.UDP; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.integration.ip.udp.UnicastSendingMessageHandler; +import org.springframework.integration.support.MessageBuilder; +import org.springframework.messaging.Message; +import org.springframework.stereotype.Component; + +/** + * @version 1.0 + * @Author whai文海 + * @Date 2024/11/22 20:24 + * @注释 + */ +@Component +public class UDPSendUtils { + + private final UnicastSendingMessageHandler sender; + + @Autowired + public UDPSendUtils(UnicastSendingMessageHandler sender) { + this.sender = sender; + } + + public void send(Object object) { + Message msg = MessageBuilder.withPayload(object).build(); + sender.handleMessage(msg); + } + +} diff --git a/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetDown.java b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetDown.java new file mode 100644 index 0000000..ab8abff --- /dev/null +++ b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetDown.java @@ -0,0 +1,27 @@ +package cn.whaifree.springdemo.controller.TS.common; + +import org.springframework.stereotype.Component; + +/** + * @version 1.0 + * @Author whai文海 + * @Date 2024/11/22 21:22 + * @注释 + */ +@Component +public class TargetDown implements ProcessStrategy { + + @Override + public void process(byte[] frame) { + + + // 封装成object + process(frame); + } + + @Override + public void process(Object o) { + System.out.println("TargetDown"); + + } +} diff --git a/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetImage.java b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetImage.java new file mode 100644 index 0000000..aa55bb5 --- /dev/null +++ b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetImage.java @@ -0,0 +1,78 @@ +package cn.whaifree.springdemo.controller.TS.common; + +import org.springframework.stereotype.Component; + +import javax.imageio.ImageIO; +import java.awt.image.BufferedImage; +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.util.HashMap; +import java.util.Map; +import java.util.TreeMap; +import java.util.concurrent.ConcurrentHashMap; + +/** + * @version 1.0 + * @Author whai文海 + * @Date 2024/11/22 21:22 + * @注释 + */ +@Component +public class TargetImage implements ProcessStrategy { + + private final Map> imageMap = new ConcurrentHashMap<>(); + + /** + * @param frame + */ + @Override + public void process(byte[] frame) { + try { + + // 读取第5-8个字节作为id + int id = getIntByByteArray(frame, 5, 8); + int seq = getIntByByteArray(frame, 9, 12); + int sumGramSize = getIntByByteArray(frame, 13, 16); // 数据报数量 + if (!imageMap.containsKey(id)) { + imageMap.put(id, new TreeMap<>()); + } + TreeMap treeMap = imageMap.get(id); + treeMap.put(seq, frame); + if (treeMap.size() == sumGramSize) { + // 满了 + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); + for (int i = 1; i <= sumGramSize; i++) { + byte[] fragment = treeMap.get(i); + if (fragment != null) { + // 假设图像数据从第17字节开始 + outputStream.write(fragment, 17, fragment.length - 17); + } + } + outputStream.write(frame); + // 输入到Minio + // minioClient.putObject("test", "image.jpg", outputStream.toByteArray(), null); + } + } catch (IOException e) { + throw new RuntimeException(e); + } + // 封装成object + process(frame); + } + + public int getIntByByteArray(byte[] frame, int left, int right) { + int result = 0; + for (int i = left; i <= right; i++) { + result = result | (frame[i] & 0xFF) << (right - i) * 8; + } + return result; + } + + + @Override + public void process(Object o) { + System.out.println("图像"); + + } + +} diff --git a/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetStorage.java b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetStorage.java new file mode 100644 index 0000000..1164fae --- /dev/null +++ b/SpringDemo/src/main/java/cn/whaifree/springdemo/controller/TS/common/TargetStorage.java @@ -0,0 +1,26 @@ +package cn.whaifree.springdemo.controller.TS.common; + +import org.springframework.stereotype.Component; + +/** + * @version 1.0 + * @Author whai文海 + * @Date 2024/11/22 21:21 + * @注释 + */ +@Component +public class TargetStorage implements ProcessStrategy { + + @Override + public void process(byte[] frame) { + // 封装 + process(frame); + } + + @Override + public void process(Object o) { + System.out.println("TargetStorage"); + } +} + +