nnet/net/server.go

package net

import (
	"errors"
	"fmt"
	"github.com/gorilla/websocket"
	"github.com/panjf2000/ants/v2"
	"net"
	"net/http"
)

// Server TCP-Server
type Server struct {
	// Name 服务端名称，默认为 n-net
	Name string
	// protocol 协议名
	// "tcp", "tcp4", "tcp6", "unix" or "unixpacket"
	// 若只想开启IPv4, 使用tcp4即可
	protocol string
	// address 服务地址
	// 地址可直接使用hostname,但强烈不建议这样做,可能会同时监听多个本地IP
	// 如果端口号不填或端口号为0，例如："127.0.0.1:" 或 ":0"，服务端将选择随机可用端口
	address string
	// 一些其它的东西 .etc..
	handler    *Handler
	sessionMgr *SessionMgr
	pool       *ants.Pool
}

func newServer(opts ...Option) *Server {
	s := &Server{
		Name:       "",
		protocol:   "",
		address:    "",
		handler:    nil,
		sessionMgr: nil,
	}

	for _, opt := range opts {
		opt(s)
	}

	s.pool, _ = ants.NewPool(0)

	return s
}

func (s *Server) listenAndServe() {
	listener, err := net.Listen(s.protocol, s.address)
	if err != nil {
		panic(err)
	}

	// 监听成功，服务已启动
	// TODO log
	defer listener.Close()

	go func() {
		for {
			conn, err := listener.Accept()
			if err != nil {
				if errors.Is(err, net.ErrClosed) {
					// 服务端网络错误
					// TODO print log
					return
				}
				// TODO log
				continue
			}

			// TODO 最大连接限制
			//if s.ConnMgr.Len() >= utils.GlobalObject.MaxConn {
			//	conn.Close()
			//	continue
			//}

			r := newRequest(s, conn)

			s.pool.Submit(func() {
				s.handler.handle(r)
			})
		}
	}()
}

func (s *Server) listenAndServeWS() {
	upgrade := websocket.Upgrader{
		ReadBufferSize:    1024,
		WriteBufferSize:   1024,
		CheckOrigin:       nil,
		EnableCompression: false,
	}
	http.HandleFunc(fmt.Sprintf("/%s/", "path"), func(w http.ResponseWriter, r *http.Request) {
		conn, err := upgrade.Upgrade(w, r, nil)
		if err != nil {
			// TODO upgrade failure, uri=r.requestURI err=err.Error()
			return
		}
		// TODO s.handler.handleWS(conn)
	})
	if err := http.ListenAndServe(s.address, nil); err != nil {
		panic(err)
	}
}

func (s *Server) listenAndServeWSTLS() {
	upgrade := websocket.Upgrader{
		ReadBufferSize:    1024,
		WriteBufferSize:   1024,
		CheckOrigin:       nil,
		EnableCompression: false,
	}
	http.HandleFunc(fmt.Sprintf("/%s/", "path"), func(w http.ResponseWriter, r *http.Request) {
		conn, err := upgrade.Upgrade(w, r, nil)
		if err != nil {
			// TODO upgrade failure, uri=r.requestURI err=err.Error()
			return
		}
		// TODO s.handler.handleWS(conn)
	})
	if err := http.ListenAndServeTLS(s.address, "", "", nil); err != nil {
		panic(err)
	}
}