Take a size. (#132)

openziti · Aug 24, 2021 · ffdc817 · ffdc817
1 parent 417dc81
commit ffdc817
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/txportal.go b/txportal.go
@@ -1,14 +1,15 @@
 package dilithium
 
 import (
+	"github.com/emirpasic/gods/trees/btree"
 	"sync"
 )
 
 // TxAlgorithm is an abstraction of an extensible flow-control implementation, which can be plugged into a TxPortal
 // instance.
 //
 type TxAlgorithm interface {
-	Ready()
+	Ready(int)
 	Tx(int)
 	Success(int)
 	DuplicateAck()
@@ -24,6 +25,7 @@ type TxAlgorithm interface {
 //
 type TxPortal struct {
 	lock      *sync.Mutex
+	tree      *btree.Tree
 	transport Transport
 	alg       TxAlgorithm
 	monitor   *TxMonitor

diff --git a/westworld.go b/westworld.go
@@ -11,7 +11,7 @@ func NewWestworldAlgorithm(pf *WestworldProfile, txPortal *TxPortal) TxAlgorithm
 	return &WestworldAlgorithm{pf, txPortal}
 }
 
-func (self *WestworldAlgorithm) Ready() {
+func (self *WestworldAlgorithm) Ready(size int) {
 }
 
 func (self *WestworldAlgorithm) Tx(size int) {