添加通用上传方法，添加websocket

DESKTOP-AO0VKC8\mahua
1 parent 86c4c18a
Showing 11 changed files with 257 additions and 103 deletions
pom.xml
src/main/java/com/huaheng/common/config/ServerConfig.java
src/main/java/com/huaheng/framework/config/CustomizationBean.java
src/main/java/com/huaheng/framework/config/WebSocketConfig.java
src/main/java/com/huaheng/framework/web/service/WebSocketServer.java
src/main/java/com/huaheng/pc/common/CommonController.java
src/main/java/com/huaheng/pc/config/transfer/controller/TransferController.java
src/main/java/com/huaheng/pc/config/transfer/domain/Transfer.java
src/main/java/com/huaheng/pc/config/transfer/mapper/TransferMapper.java
src/main/java/com/huaheng/pc/config/transfer/service/TransferService.java
src/main/java/com/huaheng/pc/config/transfer/service/TransferServiceImpl.java
@@ -65,6 +65,12 @@
             <artifactId>spring-boot-starter-undertow</artifactId>
         </dependency>
  
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-websocket</artifactId>
+			<version>2.1.3.RELEASE</version>
+		</dependency>
+
         <!-- SpringBoot 测试 -->
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
+package com.huaheng.common.config;
+
+import com.huaheng.common.utils.ServletUtils;
+import org.springframework.stereotype.Component;
+
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * 服务相关配置
+ */
+@Component
+public class ServerConfig {
+
+    /**
+     * 获取完整的请求路径，包括：域名，端口，上下文访问路径
+     *
+     * @return 服务地址
+     */
+    public String getUrl() {
+        HttpServletRequest request = ServletUtils.getRequest();
+        return getDomain(request);
+    }
+
+    public static String getDomain(HttpServletRequest request) {
+        StringBuffer url = request.getRequestURL();
+        String contextPath = request.getServletContext().getContextPath();
+        return url.delete(url.length() - request.getRequestURI().length(), url.length()).append(contextPath).toString();
+    }
+}
+package com.huaheng.framework.config;
+
+import io.undertow.server.DefaultByteBufferPool;
+import io.undertow.websockets.jsr.WebSocketDeploymentInfo;
+import org.springframework.boot.web.embedded.undertow.UndertowServletWebServerFactory;
+import org.springframework.boot.web.server.WebServerFactoryCustomizer;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author mahua
+ */
+@Component
+public class CustomizationBean implements WebServerFactoryCustomizer<UndertowServletWebServerFactory> {
+
+    @Override
+    public void customize(UndertowServletWebServerFactory factory) {
+        factory.addDeploymentInfoCustomizers(deploymentInfo -> {
+            WebSocketDeploymentInfo webSocketDeploymentInfo = new WebSocketDeploymentInfo();
+            webSocketDeploymentInfo.setBuffers(new DefaultByteBufferPool(false, 1024));
+            deploymentInfo.addServletContextAttribute("io.undertow.websockets.jsr.WebSocketDeploymentInfo", webSocketDeploymentInfo);
+        });
+    }
+}
+package com.huaheng.framework.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+
+/**
+ * webSocket配置
+ * @author mahua
+ */
+@Configuration
+public class WebSocketConfig {
+
+    @Bean
+    public ServerEndpointExporter serverEndpointExporter() {
+        return new ServerEndpointExporter();
+    }
+}
+package com.huaheng.framework.web.service;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import org.apache.commons.lang.StringUtils;
+import org.springframework.stereotype.Component;
+
+import javax.websocket.*;
+import javax.websocket.server.PathParam;
+import javax.websocket.server.ServerEndpoint;
+import java.io.IOException;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * WebSocket服务
+ * @author mahua
+ * @date 2020/5/17
+ */
+@ServerEndpoint("/imserver/{userId}")
+@Component
+public class WebSocketServer {
+
+    /**静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。*/
+    private static int onlineCount = 0;
+    /**concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。*/
+    private static ConcurrentHashMap<String,WebSocketServer> webSocketMap = new ConcurrentHashMap<>();
+    /**与某个客户端的连接会话，需要通过它来给客户端发送数据*/
+    private Session session;
+    /**接收userId*/
+    private String userId="";
+
+    /**
+     * 连接建立成功调用的方法*/
+    @OnOpen
+    public void onOpen(Session session,@PathParam("userId") String userId) {
+        this.session = session;
+        this.userId=userId;
+        if(webSocketMap.containsKey(userId)){
+            webSocketMap.remove(userId);
+            webSocketMap.put(userId,this);
+            //加入set中
+        }else{
+            webSocketMap.put(userId,this);
+            //加入set中
+            addOnlineCount();
+            //在线数加1
+        }
+
+        System.out.println("用户连接:"+userId+",当前在线人数为:" + getOnlineCount());
+        try {
+            sendMessage("连接成功");
+        } catch (IOException e) {
+            System.out.println("用户:"+userId+",网络异常!!!!!!");
+        }
+    }
+
+    /**
+     * 连接关闭调用的方法
+     */
+    @OnClose
+    public void onClose() {
+        if(webSocketMap.containsKey(userId)){
+            webSocketMap.remove(userId);
+            //从set中删除
+            subOnlineCount();
+        }
+        System.out.println("用户退出:"+userId+",当前在线人数为:" + getOnlineCount());
+    }
+
+    /**
+     * 收到客户端消息后调用的方法
+     *
+     * @param message 客户端发送过来的消息*/
+    @OnMessage
+    public void onMessage(String message, Session session) {
+        //可以群发消息
+        //消息保存到数据库、redis
+        if(StringUtils.isNotBlank(message)){
+            try {
+                //解析发送的报文
+                JSONObject jsonObject = JSON.parseObject(message);
+                //追加发送人(防止串改)
+                jsonObject.put("fromUserId",this.userId);
+                String toUserId=jsonObject.getString("toUserId");
+                //传送给对应toUserId用户的websocket
+                if(StringUtils.isNotBlank(toUserId)&&webSocketMap.containsKey(toUserId)){
+                    webSocketMap.get(toUserId).sendMessage(jsonObject.toJSONString());
+                }else{
+                    System.out.println("请求的userId:"+toUserId+"不在该服务器上");
+                    //否则不在这个服务器上，发送到mysql或者redis
+                }
+            }catch (Exception e){
+                e.printStackTrace();
+            }
+        }
+    }
+
+    /**
+     *
+     * @param session
+     * @param error
+     */
+    @OnError
+    public void onError(Session session, Throwable error) {
+        System.out.println("用户错误:"+this.userId+",原因:"+error.getMessage());
+        error.printStackTrace();
+    }
+
+    /**
+     * 实现服务器主动推送
+     */
+    public void sendMessage(String message) throws IOException {
+        this.session.getBasicRemote().sendText(message);
+    }
+
+
+    /**
+     * 发送自定义消息
+     * */
+    public static void sendInfo(String message,@PathParam("userId") String userId) throws IOException {
+        System.out.println("发送消息到:"+userId+"，报文:"+message);
+        if(StringUtils.isNotBlank(userId)&&webSocketMap.containsKey(userId)){
+            webSocketMap.get(userId).sendMessage(message);
+        }else{
+            System.out.println("发送消息到:"+userId+"，报文:"+message);
+        }
+    }
+
+    public static synchronized int getOnlineCount() {
+        return onlineCount;
+    }
+
+    public static synchronized void addOnlineCount() {
+        WebSocketServer.onlineCount++;
+    }
+
+    public static synchronized void subOnlineCount() {
+        WebSocketServer.onlineCount--;
+    }
+
+}
@@ -3,6 +3,9 @@ package com.huaheng.pc.common;
 import java.io.IOException;
 import java.io.UnsupportedEncodingException;
 import java.net.URLEncoder;
+import java.util.HashMap;
+import java.util.Map;
+import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
  
@@ -10,19 +13,22 @@ import com.huaheng.common.config.Global;
  
 import com.google.zxing.WriterException;
 import com.huaheng.common.config.Global;
+import com.huaheng.common.config.ServerConfig;
 import com.huaheng.common.utils.QRCodeGenerator;
+import com.huaheng.common.utils.file.FileUploadUtils;
+import com.huaheng.framework.web.domain.AjaxResult;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
 import org.springframework.util.ResourceUtils;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.*;
 import com.huaheng.common.utils.file.FileUtils;
+import org.springframework.web.multipart.MultipartFile;
  
 /**
  * 通用请求处理
@@ -34,6 +40,9 @@ public class CommonController
 {
     private static final Logger log = LoggerFactory.getLogger(CommonController.class);
  
+    @Resource
+    private ServerConfig serverConfig;
+
     @RequestMapping("common/download")
     public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request) {
  
@@ -100,4 +109,32 @@ public class CommonController
         headers.setContentType(MediaType.IMAGE_PNG);
         return new ResponseEntity<byte[]>(qrcode, headers, HttpStatus.CREATED);
     }
+
+    /**
+     * 通用上传请求
+     */
+    @PostMapping("/common/upload")
+    @ResponseBody
+    public AjaxResult uploadFile(MultipartFile file) throws Exception
+    {
+        try
+        {
+            // 上传文件路径
+            String filePath = "D:/Huaheng/uploadPath/";
+            // 上传并返回新文件名称
+            String fileName = FileUploadUtils.upload(filePath, file);
+            String url = serverConfig.getUrl() + "/" + fileName;
+            AjaxResult ajax = AjaxResult.success();
+            Map<String, Object> map = new HashMap<>();
+
+            map.put("fileName", fileName);
+            map.put("url", url);
+            ajax.setData(map);
+            return ajax;
+        }
+        catch (Exception e)
+        {
+            return AjaxResult.error(e.getMessage());
+        }
+    }
 }
-package com.huaheng.pc.config.transfer.controller;
-
-
-import org.springframework.stereotype.Controller;
-import org.springframework.web.bind.annotation.RequestMapping;
-
-
-/**
- * 移库配置
- * 配置闲时时间段，
- * 闲时移库热门物料到靠近出口的位置
- *
- *需要首先收集热门物料的信息。
- * 或者查询最近一根约的出入单，确定物料热度
- *
- *
- * */
-@Controller
-@RequestMapping("/config/transfer")
-public class TransferController {
-
-
-
-
-
-
-}
-package com.huaheng.pc.config.transfer.domain;
-
-
-import com.baomidou.mybatisplus.annotation.TableName;
-import lombok.Data;
-
-import java.io.Serializable;
-
-@Data
-//@TableName(value = "")
-public class Transfer implements Serializable {
-
-
-
-
-
-
-
-}
-package com.huaheng.pc.config.transfer.mapper;
-
-import com.baomidou.mybatisplus.core.mapper.BaseMapper;
-import com.huaheng.pc.config.transfer.domain.Transfer;
-
-public interface TransferMapper extends BaseMapper<Transfer> {
-
-
-
-
-
-
-
-}
-package com.huaheng.pc.config.transfer.service;
-
-import com.baomidou.mybatisplus.extension.service.IService;
-import com.huaheng.pc.config.transfer.domain.Transfer;
-
-public interface TransferService extends IService<Transfer> {
-
-
-
-}
-package com.huaheng.pc.config.transfer.service;
-
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.huaheng.pc.config.transfer.domain.Transfer;
-import com.huaheng.pc.config.transfer.mapper.TransferMapper;
-import org.springframework.stereotype.Service;
-
-
-/**
- * 移库配置
- * */
-@Service("TransferService")
-public class TransferServiceImpl extends ServiceImpl<TransferMapper, Transfer> implements TransferService {
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-}