新增wss与https功能

network/httpserver.go

@@ -15,6 +15,10 @@ type HttpServer struct {
 	writetimeout time.Duration
 
 	httpserver *http.Server
+	certfile   string
+	keyfile    string
+
+	ishttps bool
 }
 
 func (slf *HttpServer) Init(port uint16, handler http.Handler, readtimeout time.Duration, writetimeout time.Duration) {
@@ -34,7 +38,6 @@ func (slf *HttpServer) Start() {
 
 func (slf *HttpServer) startListen() error {
 	listenPort := fmt.Sprintf(":%d", slf.port)
-
 	slf.httpserver = &http.Server{
 		Addr:           listenPort,
 		Handler:        slf.handler,
@@ -43,11 +46,24 @@ func (slf *HttpServer) startListen() error {
 		MaxHeaderBytes: 1 << 20,
 	}
 
-	err := slf.httpserver.ListenAndServe()
+	var err error
+	if slf.ishttps == true {
+		err = slf.httpserver.ListenAndServeTLS(slf.certfile, slf.keyfile)
+	} else {
+		err = slf.httpserver.ListenAndServe()
+	}
+
 	if err != nil {
-		fmt.Printf("http.ListenAndServe(%d, nil) error\n", slf.port)
+		fmt.Printf("http.ListenAndServe(%d, %v) error\n", slf.port, err)
 		os.Exit(1)
 	}
 
 	return nil
 }
+
+func (slf *HttpServer) SetHttps(certfile string, keyfile string) bool {
+	slf.certfile = certfile
+	slf.keyfile = keyfile
+	slf.ishttps = true
+	return true
+}

network/websocketserver.go

@@ -58,13 +58,16 @@ type WebsocketServer struct {
 
 	httpserver *http.Server
 	reciver    map[string]Reciver
+
+	certfile string
+	keyfile  string
+	iswss    bool
 }
 
 func (slf *WebsocketServer) Init(port uint16) {
 
 	slf.port = port
 	slf.mapClient = make(map[uint64]*WSClient)
-
 }
 
 func (slf *WebsocketServer) CreateClient(conn *websocket.Conn) *WSClient {
@@ -105,7 +108,13 @@ func (slf *WebsocketServer) startListen() {
 		MaxHeaderBytes: 1 << 20,
 	}
 
-	err := slf.httpserver.ListenAndServe()
+	var err error
+	if slf.iswss == true {
+		err = slf.httpserver.ListenAndServeTLS(slf.certfile, slf.keyfile)
+	} else {
+		err = slf.httpserver.ListenAndServe()
+	}
+
 	if err != nil {
 		fmt.Printf("http.ListenAndServe(%d, nil) error\n", slf.port)
 		os.Exit(1)
@@ -189,3 +198,10 @@ func (slf *WebsocketServer) initRouterHandler() http.Handler {
 	cors := cors.AllowAll()
 	return cors.Handler(r)
 }
+
+func (slf *WebsocketServer) SetWSS(certfile string, keyfile string) bool {
+	slf.certfile = certfile
+	slf.keyfile = keyfile
+	slf.iswss = true
+	return true
+}