Write newlines to trigger read error out on a closed connection

This commit is contained in:
Kelvin Ly 2023-05-17 07:22:44 -04:00
parent 738c52984e
commit 811ed951e1
1 changed files with 7 additions and 1 deletions

View File

@ -250,13 +250,19 @@ func InitTcpServer(db *sql.DB, state *ShroomState) {
left = buf
}
_, err := conn.Write([]byte{})
_, err := conn.Write([]byte("\n"))
if err != nil {
log.Println("tcp write error: ", err)
_ = conn.Close()
log.Println("disconnected from client")
break
}
err = conn.SetReadDeadline(time.Now().Add(time.Second))
if err != nil {
log.Println("unable to set read deadline ", err)
_ = conn.Close()
break
}
num_read, err := conn.Read(left)
left = left[num_read:]