try to optimize selectgo

protocol/peer.go

@@ -41,17 +41,28 @@ func (p *Peer) keepRead(ctx context.Context) error {
 			time := time.After(p.readAccumtTimeLimit)
 			var buff Records
 			for len(buff) <= p.readBatchSize {
-				select {
-				case <-time:
-					break
-				case <-ctx.Done():
+				if len(time) > 0 || ctx.Err() != nil {
 					break
-				case recs, ok := <-reading:
-					// closed
+				}
+				if len(reading) > 0 {
+					recs, ok := <-reading
 					if !ok {
 						break
 					}
 					buff = append(buff, recs...)
+				} else {
+					select {
+					case <-time:
+						break
+					case <-ctx.Done():
+						break
+					case recs, ok := <-reading:
+						// closed
+						if !ok {
+							break
+						}
+						buff = append(buff, recs...)
+					}
 				}
 			}
 			if err := p.inout.Drain(ctx, buff); err != nil {