Add stream pool configuration for dynamic stream management

Author: mohanson

protocol/czar/engine.go

@@ -58,10 +58,12 @@ var Conf = struct {
 	// small, the transmission efficiency will be reduced, and if it is too large, the concurrency capability of mux
 	// will be reduced.
 	PacketSize int
+	StreamPool int
 }{
 	IdleProbeDuration: time.Second * 32,
 	IdleReplyDuration: time.Second * 48,
 	PacketSize:        2048,
+	StreamPool:        256,
 }
 
 // Server implemented the czar protocol.

protocol/czar/mux.go

@@ -296,15 +296,15 @@ func NewMux(conn io.ReadWriteCloser) *Mux {
 		idp: NewSip(),
 		pri: priority.NewPriority(2),
 		rer: NewErr(),
-		usb: make([]*Stream, 256),
+		usb: make([]*Stream, Conf.StreamPool),
 	}
 	return mux
 }
 
 // NewMuxServer returns a new MuxServer.
 func NewMuxServer(conn io.ReadWriteCloser) *Mux {
 	mux := NewMux(conn)
-	for i := range 256 {
+	for i := range Conf.StreamPool {
 		mux.usb[i] = NewWither(uint8(i), mux)
 	}
 	go mux.Recv()

protocol/czar/sip.go

@@ -20,11 +20,11 @@ func (s *Sip) Get() (uint8, error) {
 	s.m.Lock()
 	defer s.m.Unlock()
 	n := big.NewInt(0).Not(s.i)
-	m := n.TrailingZeroBits()
-	if m == 256 {
+	m := int(n.TrailingZeroBits())
+	if m == Conf.StreamPool {
 		return 0, errors.New("daze: out of stream")
 	}
-	s.i.SetBit(s.i, int(m), 1)
+	s.i.SetBit(s.i, m, 1)
 	return uint8(m), nil
 }
 

protocol/czar/sip_test.go

@@ -8,7 +8,7 @@ import (
 
 func TestProtocolCzarSip(t *testing.T) {
 	sid := NewSip()
-	for i := range 256 {
+	for i := range Conf.StreamPool {
 		doa.Doa(doa.Try(sid.Get()) == uint8(i))
 	}
 	doa.Doa(doa.Err(sid.Get()) != nil)