网站建站方案,在线教学网站开发,wordpress在线计算程序,网店设计流程应用环境#xff1a; B/S架构 需求描述#xff1a; 1、判断U盘接入 2、扫描U盘指定文件#xff0c;将满足条件的文件发送给服务器 解决思路#xff1a; 1、因为bs架构#xff0c;无法获取本机资源#xff0c;计划在U盘所在服务器部署websocket服务 2、websocket服务扫描u… 应用环境 B/S架构 需求描述 1、判断U盘接入 2、扫描U盘指定文件将满足条件的文件发送给服务器 解决思路 1、因为bs架构无法获取本机资源计划在U盘所在服务器部署websocket服务 2、websocket服务扫描u盘拿到指定文件使用session.getBasicRemote().sendBinary(data)分批发送二进制流到客户端 3、web端的收到二进制流后将分批数据存到数组 4、数据接收全部完毕后通过formData将数据提交到服务端进行保存 5、当时想到websocket直接将文件传给后端服务器但只想websocket服务只是作与web端的数据传输具体文件还是由web端与后端服务器进行交互。 定时任务检查U盘插入找到指定文件并将文件的二进制流发给客户端
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.util.concurrent.ConcurrentHashMap;/*** websocket的处理类。* 作用相当于HTTP请求* 中的controller*/
Component
Slf4j
ServerEndpoint(/websocket/{userId})
public class WebSocketServer {/**静态变量用来记录当前在线连接数。应该把它设计成线程安全的。*/private static int onlineCount 0;/**concurrent包的线程安全Set用来存放每个客户端对应的WebSocket对象。*/private static ConcurrentHashMapString,WebSocketServer webSocketMap new ConcurrentHashMap();/**与某个客户端的连接会话需要通过它来给客户端发送数据*/private Session session;/**接收userId*/private String userId ;public String getUserId(){return this.userId;}/*** 连接建立成* 功调用的方法*/OnOpenpublic void onOpen(Session session,PathParam(userId) String userId) {this.session session;this.userIduserId;if(webSocketMap.containsKey(userId)){webSocketMap.remove(userId);//加入set中webSocketMap.put(userId,this);}else{//加入set中webSocketMap.put(userId,this);//在线数加1addOnlineCount();}System.out.println(用户连接:userId,当前在线人数为: getOnlineCount());try {InetAddress ip InetAddress.getLocalHost();System.out.println(Current IP address : ip.getHostAddress());NetworkInterface network NetworkInterface.getByInetAddress(ip);byte[] mac network.getHardwareAddress();System.out.print(Current MAC address : );StringBuilder sb new StringBuilder();for (int i 0; i mac.length; i) {sb.append(String.format(%02X%s, mac[i], (i mac.length - 1) ? - : ));}System.out.println(sb);JSONObject object new JSONObject();object.put(ip,ip.getHostAddress());object.put(mac,sb);JSONObject obj new JSONObject();obj.put(msgType,ipAndMac);obj.put(result,object);sendInfo(JSON.toJSONString(obj),this.userId);} catch (UnknownHostException e) {e.printStackTrace();} catch (SocketException e){e.printStackTrace();}}/*** 连接关闭* 调用的方法*/OnClosepublic void onClose() {if(webSocketMap.containsKey(userId)){webSocketMap.remove(userId);//从set中删除subOnlineCount();}System.out.println(用户退出:userId,当前在线人数为: getOnlineCount());}/*** 收到客户端消* 息后调用的方法* param message* 客户端发送过来的消息**/OnMessagepublic void onMessage(String message, Session session) {System.out.println(用户消息:userId,报文:message);//可以群发消息//消息保存到数据库、redisif(message!null message.length()0){try {//解析发送的报文JSONObject jsonObject JSON.parseObject(message);//追加发送人(防止串改)jsonObject.put(fromUserId,this.userId);String toUserIdjsonObject.getString(toUserId);//传送给对应toUserId用户的websocketif(toUserId!null toUserId.length()0 webSocketMap.containsKey(toUserId)){webSocketMap.get(toUserId).sendMessage(message);}else{//否则不在这个服务器上发送到mysql或者redisSystem.out.println(请求的userId:toUserId不在该服务器上);}}catch (Exception e){e.printStackTrace();}}}/*** param session* param error*/OnErrorpublic void onError(Session session, Throwable error) {System.out.println(用户错误:this.userId,原因:error.getMessage());error.printStackTrace();}/*** 实现服务* 器主动推送*/public void sendMessage(String message) {try {this.session.getBasicRemote().sendText(message);} catch (IOException e) {e.printStackTrace();}}public void sendByteBuffer(ByteBuffer data) {System.out.print(data);try {session.getBasicRemote().sendBinary(data);} catch (IOException e) {e.printStackTrace();}}/***发送自定*义消息**/public static void sendInfo(String message, String userId) {System.out.println(发送消息到:userId报文:message);if(userId!null userId.length()0 webSocketMap.containsKey(userId)){webSocketMap.get(userId).sendMessage(message);}else{System.out.println(用户userId,不在线);}}public static void sendFile(String path, String userId) {if(userId!null userId.length()0 webSocketMap.containsKey(userId)){try {File file new File(path);FileInputStream fis new FileInputStream(file);byte[] buffer new byte[1024];int bytesRead;while ((bytesRead fis.read(buffer)) ! -1) {ByteBuffer byteBuffer ByteBuffer.wrap(buffer, 0, bytesRead);webSocketMap.get(userId).sendByteBuffer(byteBuffer);}fis.close();} catch (Exception e) {e.printStackTrace();}}else{System.out.println(用户userId,不在线);}}/***发送自定*义消息**/public void sendInfoToAll(String message) {System.out.println(发送报文:message);sendMessage(message);}/*** 获得此时的* 在线人数* return*/public static synchronized int getOnlineCount() {return onlineCount;}/*** 在线人* 数加1*/public static synchronized void addOnlineCount() {WebSocketServer.onlineCount;}/*** 在线人* 数减1*/public static synchronized void subOnlineCount() {WebSocketServer.onlineCount--;}}
Websocket服务代码
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.util.concurrent.ConcurrentHashMap;/*** websocket的处理类。* 作用相当于HTTP请求* 中的controller*/
Component
Slf4j
ServerEndpoint(/websocket/{userId})
public class WebSocketServer {/**静态变量用来记录当前在线连接数。应该把它设计成线程安全的。*/private static int onlineCount 0;/**concurrent包的线程安全Set用来存放每个客户端对应的WebSocket对象。*/private static ConcurrentHashMapString,WebSocketServer webSocketMap new ConcurrentHashMap();/**与某个客户端的连接会话需要通过它来给客户端发送数据*/private Session session;/**接收userId*/private String userId ;public String getUserId(){return this.userId;}/*** 连接建立成* 功调用的方法*/OnOpenpublic void onOpen(Session session,PathParam(userId) String userId) {this.session session;this.userIduserId;if(webSocketMap.containsKey(userId)){webSocketMap.remove(userId);//加入set中webSocketMap.put(userId,this);}else{//加入set中webSocketMap.put(userId,this);//在线数加1addOnlineCount();}System.out.println(用户连接:userId,当前在线人数为: getOnlineCount());try {InetAddress ip InetAddress.getLocalHost();System.out.println(Current IP address : ip.getHostAddress());NetworkInterface network NetworkInterface.getByInetAddress(ip);byte[] mac network.getHardwareAddress();System.out.print(Current MAC address : );StringBuilder sb new StringBuilder();for (int i 0; i mac.length; i) {sb.append(String.format(%02X%s, mac[i], (i mac.length - 1) ? - : ));}System.out.println(sb);JSONObject object new JSONObject();object.put(ip,ip.getHostAddress());object.put(mac,sb);JSONObject obj new JSONObject();obj.put(msgType,ipAndMac);obj.put(result,object);sendInfo(JSON.toJSONString(obj),this.userId);} catch (UnknownHostException e) {e.printStackTrace();} catch (SocketException e){e.printStackTrace();}}/*** 连接关闭* 调用的方法*/OnClosepublic void onClose() {if(webSocketMap.containsKey(userId)){webSocketMap.remove(userId);//从set中删除subOnlineCount();}System.out.println(用户退出:userId,当前在线人数为: getOnlineCount());}/*** 收到客户端消* 息后调用的方法* param message* 客户端发送过来的消息**/OnMessagepublic void onMessage(String message, Session session) {System.out.println(用户消息:userId,报文:message);//可以群发消息//消息保存到数据库、redisif(message!null message.length()0){try {//解析发送的报文JSONObject jsonObject JSON.parseObject(message);//追加发送人(防止串改)jsonObject.put(fromUserId,this.userId);String toUserIdjsonObject.getString(toUserId);//传送给对应toUserId用户的websocketif(toUserId!null toUserId.length()0 webSocketMap.containsKey(toUserId)){webSocketMap.get(toUserId).sendMessage(message);}else{//否则不在这个服务器上发送到mysql或者redisSystem.out.println(请求的userId:toUserId不在该服务器上);}}catch (Exception e){e.printStackTrace();}}}/*** param session* param error*/OnErrorpublic void onError(Session session, Throwable error) {System.out.println(用户错误:this.userId,原因:error.getMessage());error.printStackTrace();}/*** 实现服务* 器主动推送*/public void sendMessage(String message) {try {this.session.getBasicRemote().sendText(message);} catch (IOException e) {e.printStackTrace();}}public void sendByteBuffer(ByteBuffer data) {System.out.print(data);try {session.getBasicRemote().sendBinary(data);} catch (IOException e) {e.printStackTrace();}}/***发送自定*义消息**/public static void sendInfo(String message, String userId) {System.out.println(发送消息到:userId报文:message);if(userId!null userId.length()0 webSocketMap.containsKey(userId)){webSocketMap.get(userId).sendMessage(message);}else{System.out.println(用户userId,不在线);}}public static void sendFile(String path, String userId) {if(userId!null userId.length()0 webSocketMap.containsKey(userId)){try {File file new File(path);FileInputStream fis new FileInputStream(file);byte[] buffer new byte[1024];int bytesRead;while ((bytesRead fis.read(buffer)) ! -1) {ByteBuffer byteBuffer ByteBuffer.wrap(buffer, 0, bytesRead);webSocketMap.get(userId).sendByteBuffer(byteBuffer);}fis.close();} catch (Exception e) {e.printStackTrace();}}else{System.out.println(用户userId,不在线);}}/***发送自定*义消息**/public void sendInfoToAll(String message) {System.out.println(发送报文:message);sendMessage(message);}/*** 获得此时的* 在线人数* return*/public static synchronized int getOnlineCount() {return onlineCount;}/*** 在线人* 数加1*/public static synchronized void addOnlineCount() {WebSocketServer.onlineCount;}/*** 在线人* 数减1*/public static synchronized void subOnlineCount() {WebSocketServer.onlineCount--;}}
Vue前端代码
websocketOnmessage: function (e){let data e.datalet that this;console.log(rec, data);if (this.isBlob(data)) {console.log(recdata, data)this.recData.push(data);} else {let record JSON.parse(data)if (record.msgType ipAndMac) {if(!sessionStorage.getItem(ipAndMac)){let result record.result;loginLog(result).then(res{console.log(res,res)if(res.data.success){sessionStorage.setItem(ipAndMac,result)} })}} else if (record.msgType sendFileBegin) {//开始接收服务端发送过来的文件数据this.recData [];} else if (record.msgType sendFileEnd) {//文件数据的接收完毕合并数据并上传到业务服务器if (this.recData.length 0) return;this.$showMsgBox({caption:询问,msg: 检查到待导入文件 record.fileName ,是否导入,callback:(data) {if (data yes) {var formData new FormData()formData.append(file, new Blob(this.recData))formData.append(fileName, record.fileName); let url config.configData.api_url /business/biz/importAllByPath;utils.httpFile(url,formData).then((res) {if (res.data.success true) { that.$showImport({tableData:res.data.data})} else {this.$showToast({msg: res.data.message});return; }})}}})}}},
特别注意这个写法不要搞错formData.append(file, new Blob(this.recData))否则后端接受不到正确的格式数据。
服务端接收前端上传数据代码
RequestMapping(value /importAllByPath, method RequestMethod.POST)public Result? importAllByPath(HttpServletRequest request, HttpServletResponse response) {String fileName request.getParameter(fileName);MultipartHttpServletRequest multipartRequest (MultipartHttpServletRequest) request;MapString, MultipartFile fileMap multipartRequest.getFileMap();for (Map.EntryString, MultipartFile entity : fileMap.entrySet()) {MultipartFile file entity.getValue();// 获取上传文件对象try {String uuid StrUtils.getTimeNo(sims_data_);String path uploadpath File.separator temp File.separator uuid File.separator;File dir new File(path);if (!dir.exists()) {dir.mkdirs();}String sFile new File(path).getAbsolutePath() File.separator uuid .erc;try (FileOutputStream outputStream new FileOutputStream(sFile)) {byte[] bytes file.getBytes();outputStream.write(bytes);} catch (Exception e) {return Result.error(文件导入失败: e.getMessage());}return Result.ok(result);} catch (Exception e) {return Result.error(文件导入失败: e.getMessage());}}return Result.ok(导入失败);}
