update log
This commit is contained in:
13
TCPServer.go
13
TCPServer.go
@ -31,14 +31,13 @@ type TCPServer struct {
|
||||
Config *Config
|
||||
}
|
||||
|
||||
func NewTCPServer(address string, port string, log *log.Logger) *TCPServer {
|
||||
func NewTCPServer(address string, port string) *TCPServer {
|
||||
return &TCPServer{
|
||||
Address: address,
|
||||
Port: port,
|
||||
Clients: make(map[net.Conn]bool),
|
||||
MsgChan: make(chan string, 100),
|
||||
CmdChan: make(chan string),
|
||||
Log: log,
|
||||
Mutex: new(sync.Mutex),
|
||||
}
|
||||
}
|
||||
@ -47,12 +46,12 @@ func (s *TCPServer) StartServer() {
|
||||
s.LogWriter = bufio.NewWriter(os.Stdout)
|
||||
s.Listener, err = net.Listen("tcp", Cfg.TelnetServer.Host+":"+Cfg.TelnetServer.Port)
|
||||
if err != nil {
|
||||
s.Log.Info("Could not create telnet server")
|
||||
Log.Info("Could not create telnet server")
|
||||
}
|
||||
|
||||
defer s.Listener.Close()
|
||||
|
||||
s.Log.Infof("Telnet server listening on %s:%s", Cfg.TelnetServer.Host, Cfg.TelnetServer.Port)
|
||||
Log.Infof("Telnet server listening on %s:%s", Cfg.TelnetServer.Host, Cfg.TelnetServer.Port)
|
||||
|
||||
go func() {
|
||||
for message := range s.MsgChan {
|
||||
@ -62,9 +61,9 @@ func (s *TCPServer) StartServer() {
|
||||
|
||||
for {
|
||||
s.Conn, err = s.Listener.Accept()
|
||||
s.Log.Info("Client connected", s.Conn.RemoteAddr().String())
|
||||
Log.Info("Client connected", s.Conn.RemoteAddr().String())
|
||||
if err != nil {
|
||||
s.Log.Error("Could not accept connections to telnet server")
|
||||
Log.Error("Could not accept connections to telnet server")
|
||||
continue
|
||||
}
|
||||
s.Mutex.Lock()
|
||||
@ -99,7 +98,7 @@ func (s *TCPServer) handleConnection() {
|
||||
delete(s.Clients, s.Conn)
|
||||
s.Mutex.Unlock()
|
||||
s.Conn.Close()
|
||||
s.Log.Infof("client %s disconnected", s.Conn.RemoteAddr().String())
|
||||
Log.Infof("client %s disconnected", s.Conn.RemoteAddr().String())
|
||||
}
|
||||
|
||||
if strings.Contains(message, "DX") || strings.Contains(message, "SH/DX") || strings.Contains(message, "set") || strings.Contains(message, "SET") {
|
||||
|
Reference in New Issue
Block a user