Add idle disconnect logic to czar

Author: mohanson

protocol/czar/engine.go

@@ -46,12 +46,15 @@ import (
 var Conf = struct {
 	// The newly created stream has a higher write priority.
 	FastWriteDuration time.Duration
+	// If no new data is received for a period of time, the mux is disconnected.
+	IdleAfterDuration time.Duration
 	// Packet size. Since the size of the packet header is 4, this value must be greater than 4. If the value is too
 	// small, the transmission efficiency will be reduced, and if it is too large, the concurrency capability of mux
 	// will be reduced.
 	PacketSize int
 }{
 	FastWriteDuration: time.Second * 8,
+	IdleAfterDuration: time.Minute * 8,
 	PacketSize:        2048,
 }
 

protocol/czar/mux.go

@@ -201,12 +201,16 @@ func (m *Mux) Recv() {
 		buf = make([]byte, 4)
 		cmd uint8
 		err error
+		fin = time.AfterFunc(Conf.IdleAfterDuration, func() {
+			m.Close()
+		})
 		idx uint8
 		msg []byte
 		old *Stream
 		stm *Stream
 	)
 	for {
+		fin.Reset(Conf.IdleAfterDuration)
 		_, err = io.ReadFull(m.con, buf[:4])
 		if err != nil {
 			m.rer.Put(err)
@@ -253,6 +257,7 @@ func (m *Mux) Recv() {
 		}
 	}
 	close(m.ach)
+	fin.Stop()
 }
 
 // NewMux returns a new Mux.