create congestion RTTstats in Session

ref #106
2016-10-30 17:27:21 +07:00
parent bb24be8281
commit c8b7246159
3 changed files with 9 additions and 5 deletions
--- a/ackhandler/sent_packet_handler.go
+++ b/ackhandler/sent_packet_handler.go
@@ -47,9 +47,7 @@ type sentPacketHandler struct {
 }

 // NewSentPacketHandler creates a new sentPacketHandler
-func NewSentPacketHandler() SentPacketHandler {
-	rttStats := &congestion.RTTStats{}
-
+func NewSentPacketHandler(rttStats *congestion.RTTStats) SentPacketHandler {
 	congestion := congestion.NewCubicSender(
 		congestion.DefaultClock{},
 		rttStats,
--- a/ackhandler/sent_packet_handler_test.go
+++ b/ackhandler/sent_packet_handler_test.go
@@ -57,7 +57,8 @@ var _ = Describe("SentPacketHandler", func() {
 	)

 	BeforeEach(func() {
-		handler = NewSentPacketHandler().(*sentPacketHandler)
+		rttStats := &congestion.RTTStats{}
+		handler = NewSentPacketHandler(rttStats).(*sentPacketHandler)
 		streamFrame = frames.StreamFrame{
 			StreamID: 5,
 			Data:     []byte{0x13, 0x37},