67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
logger "git.pyer.club/kingecg/gologger"
|
||
|
|
"github.com/soheilhy/cmux"
|
||
|
|
)
|
||
|
|
|
||
|
|
var ServerManager map[string]*ServerListener = make(map[string]*ServerListener)
|
||
|
|
|
||
|
|
type ServerListener struct {
|
||
|
|
listener cmux.CMux
|
||
|
|
servers map[string]*http.Server
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerListener) AddServer(name string, server *http.Server) {
|
||
|
|
s.servers[name] = server
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerListener) StartServer(name string) error {
|
||
|
|
server, ok := s.servers[name]
|
||
|
|
if !ok {
|
||
|
|
panic("No named server")
|
||
|
|
}
|
||
|
|
|
||
|
|
l := s.listener.Match(cmux.HTTP1HeaderField("Host", name))
|
||
|
|
return server.Serve(l)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerListener) Start() {
|
||
|
|
for name, _ := range s.servers {
|
||
|
|
err := s.StartServer(name)
|
||
|
|
if err != nil {
|
||
|
|
l := logger.GetLogger("ServerListener")
|
||
|
|
l.Error("Start server error:", name, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerListener) ShutDown() {
|
||
|
|
// may need add wait group
|
||
|
|
for _, server := range s.servers {
|
||
|
|
server.Shutdown(context.Background())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerListener) Serve() {
|
||
|
|
go s.listener.Serve()
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewServerListener(port int) *ServerListener {
|
||
|
|
listenStr := fmt.Sprintf(":%d", port)
|
||
|
|
l, err := net.Listen("tcp", listenStr)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
l := logger.GetLogger("ServerListener")
|
||
|
|
l.Error("Listen error:", err)
|
||
|
|
}
|
||
|
|
muxer := cmux.New(l)
|
||
|
|
s := &ServerListener{listener: muxer, servers: make(map[string]*http.Server)}
|
||
|
|
return s
|
||
|
|
}
|