update log

This commit is contained in:
2024-10-31 11:14:35 +07:00
parent 91fc5836eb
commit f571d8ea18
8 changed files with 40 additions and 45 deletions

View File

@ -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") {