Imp: add SetTaskPool for session interface

getty.go

@@ -154,6 +154,7 @@ type Session interface {
 	SetRQLen(int)
 	SetWQLen(int)
 	SetWaitTime(time.Duration)
+	SetTaskPool(*TaskPool)
 
 	GetAttribute(interface{}) interface{}
 	SetAttribute(interface{}, interface{})

session.go

@@ -647,8 +647,6 @@ func (s *session) handleTCPPackage() error {
 					break
 				}
 				log.Errorf("%s, [session.conn.read] = error:%+v", s.sessionToken(), err)
-				// for (Codec)OnErr
-				// s.errFlag = true
 				exit = true
 			}
 			break
@@ -787,7 +785,6 @@ func (s *session) handleWSPackage() error {
 		if err != nil {
 			log.Warnf("%s, [session.handleWSPackage] = error{%+s}",
 				s.sessionToken(), err)
-			// s.errFlag = true
 			return perrors.WithStack(err)
 		}
 		s.UpdateActive()

task_pool.go

@@ -29,12 +29,17 @@ type TaskPool struct {
 }
 
 // build a task pool
-func newTaskPool(opts TaskPoolOptions) *TaskPool {
-	opts.validate()
+func NewTaskPool(opts ...TaskPoolOption) *TaskPool {
+	var tOpts TaskPoolOptions
+	for _, opt := range opts {
+		opt(&tOpts)
+	}
+
+	tOpts.validate()
 
 	p := &TaskPool{
-		TaskPoolOptions: opts,
-		qArray:          make([]chan task, opts.tQNumber),
+		TaskPoolOptions: tOpts,
+		qArray:          make([]chan task, tOpts.tQNumber),
 		done:            make(chan struct{}),
 	}
 
@@ -107,7 +112,7 @@ func (p *TaskPool) stop() {
 }
 
 // check whether the session has been closed.
-func (p *TaskPool) isClosed() bool {
+func (p *TaskPool) IsClosed() bool {
 	select {
 	case <-p.done:
 		return true
@@ -117,7 +122,7 @@ func (p *TaskPool) isClosed() bool {
 	}
 }
 
-func (p *TaskPool) close() {
+func (p *TaskPool) Close() {
 	p.stop()
 	p.wg.Wait()
 	for i := range p.qArray {