refactor(商城): 代码优化

waynboot-admin-api/pom.xml

@@ -17,7 +17,10 @@
     </properties>
 
     <dependencies>
-
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-websocket</artifactId>
+        </dependency>
         <!--阿里数据库连接池 -->
         <dependency>
             <groupId>com.alibaba</groupId>

waynboot-admin-api/src/main/java/com/wayn/admin/api/controller/websocket/WebSocketServer.java

@@ -0,0 +1,124 @@
+package com.wayn.admin.api.controller.websocket;
+
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import javax.websocket.*;
+import javax.websocket.server.ServerEndpoint;
+import java.io.IOException;
+import java.util.concurrent.CopyOnWriteArraySet;
+import java.util.concurrent.atomic.AtomicInteger;
+
+@ServerEndpoint(value = "/ws/asset")
+@Component
+@Slf4j
+public class WebSocketServer {
+
+    private static final AtomicInteger OnlineCount = new AtomicInteger(0);
+    // concurrent包的线程安全Set，用来存放每个客户端对应的Session对象。
+    private static final CopyOnWriteArraySet<Session> SessionSet = new CopyOnWriteArraySet<>();
+
+    /**
+     * 发送消息，实践表明，每次浏览器刷新，session会发生变化。
+     *
+     * @param session
+     * @param message
+     */
+    public static void SendMessage(Session session, String message) {
+        try {
+//            session.getBasicRemote().sendText(String.format("%s (From Server，Session ID=%s)",message,session.getId()));
+            session.getBasicRemote().sendText(message);
+        } catch (IOException e) {
+            log.error("发送消息出错：{}", e.getMessage());
+            e.printStackTrace();
+        }
+    }
+
+    /**
+     * 群发消息
+     *
+     * @param message
+     * @throws IOException
+     */
+    public static void BroadCastInfo(String message) throws IOException {
+        for (Session session : SessionSet) {
+            if (session.isOpen()) {
+                SendMessage(session, message);
+            }
+        }
+    }
+
+    /**
+     * 指定Session发送消息
+     *
+     * @param sessionId
+     * @param message
+     * @throws IOException
+     */
+    public static void SendMessage(String message, String sessionId) throws IOException {
+        Session session = null;
+        for (Session s : SessionSet) {
+            if (s.getId().equals(sessionId)) {
+                session = s;
+                break;
+            }
+        }
+        if (session != null) {
+            SendMessage(session, message);
+        } else {
+            log.warn("没有找到你指定ID的会话：{}", sessionId);
+        }
+    }
+
+    @PostConstruct
+    public void init() {
+        log.info("websocket init");
+    }
+
+    /**
+     * 连接建立成功调用的方法
+     */
+    @OnOpen
+    public void onOpen(Session session) {
+        SessionSet.add(session);
+        int cnt = OnlineCount.incrementAndGet(); // 在线数加1
+        log.info("有连接加入，当前连接数为：{}", cnt);
+        SendMessage(session, "连接成功");
+    }
+
+    /**
+     * 连接关闭调用的方法
+     */
+    @OnClose
+    public void onClose(Session session) {
+        SessionSet.remove(session);
+        int cnt = OnlineCount.decrementAndGet();
+        log.info("有连接关闭，当前连接数为：{}", cnt);
+    }
+
+    /**
+     * 收到客户端消息后调用的方法
+     *
+     * @param message 客户端发送过来的消息
+     */
+    @OnMessage
+    public void onMessage(String message, Session session) {
+        log.info("来自客户端的消息：{}", message);
+        SendMessage(session, "收到消息，消息内容：" + message);
+
+    }
+
+    /**
+     * 出现错误
+     *
+     * @param session
+     * @param error
+     */
+    @OnError
+    public void onError(Session session, Throwable error) {
+        log.error("发生错误：{}，Session ID： {}", error.getMessage(), session.getId());
+        error.printStackTrace();
+    }
+}

waynboot-admin-api/src/main/java/com/wayn/admin/framework/config/SecurityConfig.java

@@ -75,6 +75,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter {
                 .antMatchers("/druid/**").anonymous()
                 .antMatchers("/elastic/**").anonymous()
                 .antMatchers("/message/**").anonymous()
+                .antMatchers("/ws/**").anonymous()
                 // 除上面外的所有请求全部需要鉴权认证
                 .anyRequest().authenticated().and()
                 .headers().frameOptions().disable();

waynboot-admin-api/src/main/java/com/wayn/admin/framework/config/WebSocketConfig.java

@@ -0,0 +1,18 @@
+package com.wayn.admin.framework.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+
+/**
+ * websocket配置
+ */
+@Configuration
+public class WebSocketConfig {
+
+    @Bean
+    public ServerEndpointExporter serverEndpointExporter() {
+        return new ServerEndpointExporter();
+    }
+
+}

waynboot-common/src/main/java/com/wayn/common/util/mail/MailUtil.java

@@ -39,16 +39,17 @@ public class MailUtil {
                 properties.put("mail.smtp.ssl.socketFactory", sf);
                 // 设置系统属性
                 properties.setProperty("mail.smtp.host", host);
+                properties.setProperty("mail.transport.protocol", "smtp");
                 properties.setProperty("mail.smtp.port", emailConfig.getPort());
                 properties.put("mail.smtp.auth", "true");
-                //获取发送邮件会话、获取第三方登录授权码
+                // 获取发送邮件会话、获取第三方登录授权码
                 Session session = Session.getDefaultInstance(properties, new Authenticator() {
                     @Override
                     protected PasswordAuthentication getPasswordAuthentication() {
                         return new PasswordAuthentication(from, emailConfig.getPass());
                     }
                 });
-                // session.setDebug(true);
+                session.setDebug(true);
                 // 创建默认的 MimeMessage 对象
                 MimeMessage message = new MimeMessage(session);
                 MimeMessageHelper helper = new MimeMessageHelper(message, true, "UTF-8");