jms 通用方法

src/main/java/com/scbfkj/uni/process/JMS.java

@@ -0,0 +1,181 @@
+package com.scbfkj.uni.process;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.scbfkj.uni.system.Config;
+import jakarta.jms.ConnectionFactory;
+import jakarta.jms.JMSException;
+import jakarta.jms.Message;
+import jakarta.jms.TextMessage;
+import org.springframework.jms.connection.CachingConnectionFactory;
+import org.springframework.jms.core.JmsTemplate;
+
+import java.io.File;
+import java.io.IOException;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
+
+public class JMS {
+
+
+    private JmsTemplate jmsTemplate;
+
+    private final ObjectMapper mapper = new ObjectMapper();
+
+    public List<String> receptionMessage(String path, String className,
+                                         String queueName, Long receiveTimeout, Long pollSize, Long retry, Object... args) throws JMSException, MalformedURLException, ClassNotFoundException, InvocationTargetException, NoSuchMethodException, InstantiationException, IllegalAccessException {
+        JmsTemplate template = getJmsTemplate(path, className, args);
+
+
+        jmsTemplate.setReceiveTimeout(receiveTimeout);
+        long maxSize = 100;
+        if (pollSize > 0) {
+            maxSize = pollSize;
+        }
+        int maxRetry = 0;
+        List<String> result = new ArrayList<>();
+
+
+        while (true) {
+            try {
+                Message message = template.receive(queueName);
+                if (message == null) {
+                    break;
+                } else if (message instanceof TextMessage msg) {
+                    String text = msg.getText();
+                    result.add(text);
+                    if (result.size() >= maxSize) {
+                        break;
+                    }
+                }
+            } catch (Exception e) {
+                maxRetry++;
+                if (maxRetry > retry) {
+                    if (Config.isDebug()) {
+                        e.printStackTrace();
+                    }
+                    if (jmsTemplate != null) {
+                        jmsTemplate = null;
+                    }
+
+                    return result;
+                }
+            }
+        }
+        return result;
+
+
+    }
+
+
+    public void sendMessage(String path, String className,
+                            String queueName,
+                            Object data, String... args
+    ) throws JMSException, MalformedURLException, ClassNotFoundException, InvocationTargetException, NoSuchMethodException, InstantiationException, IllegalAccessException {
+
+        JmsTemplate template = getJmsTemplate(path, className, args);
+
+        try {
+            if (data instanceof Object[] datas) {
+                for (Object it : datas) {
+
+                    execSend(template, it, queueName);
+                }
+            } else if (data instanceof Iterable<?> datas) {
+
+                for (Object it : datas) {
+
+                    execSend(template, it, queueName);
+                }
+            } else {
+                execSend(template, data, queueName);
+            }
+
+        } catch (Exception e) {
+            if (Config.isDebug()) {
+                e.printStackTrace();
+            }
+            if (jmsTemplate != null) {
+                jmsTemplate = null;
+            }
+
+        }
+    }
+
+
+    private void execSend(
+            JmsTemplate template,
+            Object data,
+            String queueName) throws IOException {
+        String message = mapper.writeValueAsString(data);
+
+        template.send(queueName, session -> session.createTextMessage(message));
+    }
+
+
+    private jakarta.jms.ConnectionFactory createConnectionFactory(String path, String className, Object... args) throws MalformedURLException, NoSuchMethodException, InvocationTargetException, InstantiationException, IllegalAccessException, ClassNotFoundException {
+        ClassLoader classLoader;
+        if (Objects.nonNull(path) && !path.trim().isEmpty()) {
+            String userPath = System.getProperty("user.dir");
+            File file = new File(userPath.concat(File.separator).concat("plugins").concat(File.separator).concat(path));
+            if (!file.exists()) {
+                throw new RuntimeException("外部文件加载不存在：".concat(userPath).concat(File.separator).concat("plugins").concat(File.separator).concat(path));
+            }
+            URL url = file.toURI().toURL();
+            classLoader = new URLClassLoader(new URL[]{url}, Thread.currentThread().getContextClassLoader());
+        } else {
+            classLoader = Thread.currentThread().getContextClassLoader();
+        }
+
+        try {
+
+            Class<?> classExample = classLoader.loadClass(className); //获取类实例
+            Class<?>[] array = Arrays.stream(args).map(Object::getClass).toArray(Class[]::new);
+            Constructor<?> constructor;
+            constructor = classExample.getConstructor(array);
+
+            Object o = constructor.newInstance(args);
+            return (ConnectionFactory) o;
+        } catch (Exception exception) {
+            exception.printStackTrace();
+            throw new RuntimeException(exception);
+        } finally {
+            try {
+                if (classLoader instanceof URLClassLoader cl) {
+                    cl.close();
+                }
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
+
+    private CachingConnectionFactory createCachingConnectionFactory(jakarta.jms.ConnectionFactory factory) {
+        CachingConnectionFactory cachingConnectionFactory = new CachingConnectionFactory();
+        cachingConnectionFactory.setTargetConnectionFactory(factory);
+        return cachingConnectionFactory;
+    }
+
+
+    private JmsTemplate getJmsTemplate(String path, String className, Object... args) throws MalformedURLException, ClassNotFoundException, InvocationTargetException, NoSuchMethodException, InstantiationException, IllegalAccessException {
+        if (jmsTemplate == null) {
+            jakarta.jms.ConnectionFactory connectionFactory = createConnectionFactory(path, className, args);
+            CachingConnectionFactory cachingConnectionFactory = createCachingConnectionFactory(connectionFactory);
+            jmsTemplate = createJmsTemplate(cachingConnectionFactory);
+        }
+        return jmsTemplate;
+    }
+
+    private JmsTemplate createJmsTemplate(CachingConnectionFactory factory) {
+        JmsTemplate jmsTemplate = new JmsTemplate();
+        jmsTemplate.setConnectionFactory(factory);
+        return jmsTemplate;
+    }
+}