Fix command client connect

This commit is contained in:
世界 2023-03-02 16:40:28 +08:00
parent 8151bcfd6b
commit 27aba99e6c
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
3 changed files with 2 additions and 2 deletions

View file

@ -53,12 +53,14 @@ func (c *CommandClient) Connect() error {
}
switch c.options.Command {
case CommandLog:
c.handler.Connected()
go c.handleLogConn(conn)
case CommandStatus:
err = binary.Write(conn, binary.BigEndian, c.options.StatusInterval)
if err != nil {
return E.Cause(err, "write interval")
}
c.handler.Connected()
go c.handleStatusConn(conn)
}
return nil

View file

@ -78,7 +78,6 @@ func (s *CommandServer) handleLogConn(conn net.Conn) error {
}
func (c *CommandClient) handleLogConn(conn net.Conn) {
c.handler.Connected()
for {
message, err := readLog(conn)
if err != nil {

View file

@ -48,7 +48,6 @@ func (s *CommandServer) handleStatusConn(conn net.Conn) error {
}
func (c *CommandClient) handleStatusConn(conn net.Conn) {
c.handler.Connected()
for {
var message StatusMessage
err := binary.Read(conn, binary.BigEndian, &message)