64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
|
package server_webSocket
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"net/http"
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
"common/clientMgr"
|
|||
|
|
"github.com/gorilla/websocket"
|
|||
|
|
"goutil/debugUtil"
|
|||
|
|
"goutil/logUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var upgrader = websocket.Upgrader{
|
|||
|
|
ReadBufferSize: 1024,
|
|||
|
|
WriteBufferSize: 1024,
|
|||
|
|
CheckOrigin: func(r *http.Request) bool {
|
|||
|
|
return true
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func handleConn(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|||
|
|
|
|||
|
|
//是否debug模式
|
|||
|
|
if debugUtil.IsDebug() {
|
|||
|
|
logUtil.DebugLog(fmt.Sprintf("收到客户的websockte连接,ip%v", conn.RemoteAddr()))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
logUtil.ErrorLog("websocket.handleConn获取连接出错，err:%v", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建客户端对象
|
|||
|
|
clientObj := newClient(conn)
|
|||
|
|
clientObj.start()
|
|||
|
|
clientMgr.RegisterClient(clientObj)
|
|||
|
|
|
|||
|
|
debugUtil.Printf("收到连接请求:remoteAdd:%s\n", conn.RemoteAddr())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Start 启动服务器
|
|||
|
|
func Start(wg *sync.WaitGroup, address string, isUseTSL bool) {
|
|||
|
|
defer wg.Done()
|
|||
|
|
|
|||
|
|
msg := fmt.Sprintf("server_websocket begins to listen on:%s...", address)
|
|||
|
|
fmt.Println(msg)
|
|||
|
|
logUtil.InfoLog(msg)
|
|||
|
|
|
|||
|
|
http.HandleFunc("/", handleConn)
|
|||
|
|
if isUseTSL {
|
|||
|
|
err := http.ListenAndServeTLS(address, "tlsFile/7qule.com.pem", "tlsFile/7qule.com.key", nil)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Sprintf("server_websocket.ListenAndServeTLS, err:%v", err))
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
err := http.ListenAndServe(address, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Sprintf("server_websocket.ListenAndServe, err:%v", err))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|