Java實現websocket與微信小程序鏈接

微信的WebSocket接口和HTML5的WebSocket基本同樣，是HTTP協議升級來的，作爲一個新的Socket在B/S上使用，它實現了瀏覽器與服務器全雙工通訊。

 

在WebSocket出來以前，實現即時通信一般使用Ajax來實現，而Ajax是經過輪詢的方式進行實時數據的獲取，輪詢就是在指定的時間間隔內，進行HTTP 請求來獲取數據，而這種方式會產生一些弊端，一方面產生過多的HTTP請求，佔用帶寬，增大服務器的相應，浪費資源，另外一方面，由於不是每一次請求都會有數據變化（就像聊天室），因此就會形成請求的利用率低。
 

而WebSocket正好可以解決上面的弊端，WebSocket是客戶端與服務器以前專門創建一條通道，請求也只請求一次，並且能夠從同道中實時的獲取服務器的數據，因此當應用到實時的應用上時，WebSocket是一個很不錯的選擇。

 

WebSocket的連接不是以http或https開頭的，而是以ws和wss開頭的，這裏須要注意一下。

 

import com.ds.tech.service.SocketChatService;
import com.ds.tech.utility.log4j.LogWriter;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.socket.server.standard.SpringConfigurator;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;

//websocket鏈接URL地址和可被調用配置
@ServerEndpoint(value="/wx/{modular}/{interfaces}",configurator = SpringConfigurator.class)
public class WebSocketChatCotroller {

    private static Logger logger = Logger.getRootLogger();

    //須要session來對用戶發送數據, 獲取鏈接特徵userId
    private Session session;
    private String api;

    @Autowired
    private SocketChatService service;

    /**
     * @Title: onOpen
     * @Description: websocekt鏈接創建時的操做
     * @param @param userId 用戶id
     * @param @param session websocket鏈接的session屬性
     * @param @throws IOException
     */
    @OnOpen
    public void onOpen(@PathParam("modular") String modular,@PathParam("interfaces") String interfaces, Session session) throws IOException{
        this.session = session;
        this.api = "wx/"+modular+"/"+interfaces;
    }

    /**
     * @Title: onClose
     * @Description: 鏈接關閉的操做
     */
    @OnClose
    public void onClose(){
        logger.info(api+"連接關閉");
        this.session = null;
    }

    /**
     * @Title: onMessage
     * @Description: 收到消息後的操做
     * @param @param message 收到的消息
     * @param @param session 該鏈接的session屬性
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        logger.info("接口："+api+",入參："+message);
        if(session ==null){
            logger.info("session null");
        }
        String apiResultString = null;
        try {
            apiResultString = service.getApi(api,message);
        }catch (Exception e){
            LogWriter.writeErrorLog("請求失敗，緣由：", e);
        }
        //向客戶端發送消息發送
        sendMessageToUser(this.api,apiResultString);
    }

    /**
     * @Title: onError
     * @Description: 鏈接發生錯誤時候的操做
     * @param @param session 該鏈接的session
     * @param @param error 發生的錯誤
     */
    @OnError
    public void onError(Session session, Throwable error){
        logger.info("接口："+api+"鏈接時發送錯誤：");
        error.printStackTrace();
    }

    /**
     * @Title: sendMessageToUser
     * @Description: 發送消息給用戶下的全部終端
     * @param @param userId 用戶id
     * @param @param message 發送的消息
     * @param @return 發送成功返回true，反則返回false
     */
    public Boolean sendMessageToUser(String api,String message){
            logger.info(api+"返回參數："+message);
            try {
                this.session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
                logger.info(api+"返回參數失敗："+e);
                return false;
            }
            return true;
    }

}

 

import com.alibaba.fastjson.JSONObject;
import com.ds.tech.service.base.BaseService;
import com.ds.tech.utility.common.ConfigUtil;
import com.ds.tech.utility.http.HttpRequestUtil;
import com.ds.tech.utility.model.InputObject;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.net.URLEncoder;

@Service
@Scope("prototype")
public class SocketChatService extends BaseService {
    public String getApi(String url,String message) throws IOException {
        InputObject inputObject = JSONObject.parseObject(message,InputObject.class);
        //請求參數
        String data =  inputObject.getData();
        apiParamMap.put("partner", inputObject.getPartner());
        apiParamMap.put("data", URLEncoder.encode(data, "utf-8"));
        apiParamMap.put("sign", inputObject.getSign());
        logger.info("小程序請求API系統-"+url+":"+apiParamMap);
        // 請求api
        return apiResultString = HttpRequestUtil.get(ConfigUtil.getSettings("api_host")+ url+ ".do", apiParamMap);

    }

}