Add SetWriteDeadline to Underlay

openziti · Mar 3, 2023 · 22655f6 · 22655f6
1 parent a28dff8
commit 22655f6
Show file tree

Hide file tree

Showing 8 changed files with 29 additions and 0 deletions.
diff --git a/channel.go b/channel.go
@@ -168,6 +168,7 @@ type Underlay interface {
 	IsClosed() bool
 	Headers() map[int32][]byte
 	SetWriteTimeout(duration time.Duration) error
+	SetWriteDeadline(time time.Time) error
 	GetLocalAddr() net.Addr
 	GetRemoteAddr() net.Addr
 }

diff --git a/classic_impl.go b/classic_impl.go
@@ -49,6 +49,10 @@ func (impl *classicImpl) SetWriteTimeout(duration time.Duration) error {
 	return impl.peer.SetWriteDeadline(time.Now().Add(duration))
 }
 
+func (impl *classicImpl) SetWriteDeadline(deadline time.Time) error {
+	return impl.peer.SetWriteDeadline(deadline)
+}
+
 func (impl *classicImpl) rxHello() (*Message, error) {
 	msg, readF, marshallF, err := readHello(impl.peer)
 	impl.readF = readF

diff --git a/datagram/underlay.go b/datagram/underlay.go
@@ -112,3 +112,7 @@ func (self *Underlay) Headers() map[int32][]byte {
 func (self *Underlay) SetWriteTimeout(duration time.Duration) error {
 	return self.peer.SetWriteDeadline(time.Now().Add(duration))
 }
+
+func (self *Underlay) SetWriteDeadline(deadline time.Time) error {
+	return self.peer.SetWriteDeadline(deadline)
+}
diff --git a/existing_conn_impl.go b/existing_conn_impl.go
@@ -33,6 +33,10 @@ func (impl *existingConnImpl) SetWriteTimeout(duration time.Duration) error {
 	return impl.peer.SetWriteDeadline(time.Now().Add(duration))
 }
 
+func (impl *existingConnImpl) SetWriteDeadline(deadline time.Time) error {
+	return impl.peer.SetWriteDeadline(deadline)
+}
+
 func (impl *existingConnImpl) rxHello() (*Message, error) {
 	msg, readF, marshallF, err := readHello(impl.peer)
 	impl.readF = readF

diff --git a/memory/impl.go b/memory/impl.go
@@ -61,6 +61,10 @@ func (impl *memoryImpl) SetWriteTimeout(time.Duration) error {
 	panic("SetWriteTimeout not implemented")
 }
 
+func (self *memoryImpl) SetWriteDeadline(deadline time.Time) error {
+	panic("SetWriteDeadline not implemented")
+}
+
 func (impl *memoryImpl) Rx() (*channel.Message, error) {
 	if impl.closed {
 		return nil, errors.New("underlay closed")

diff --git a/reconnecting_impl.go b/reconnecting_impl.go
@@ -201,6 +201,10 @@ func (impl *reconnectingImpl) SetWriteTimeout(duration time.Duration) error {
 	return impl.peer.SetWriteDeadline(time.Now().Add(duration))
 }
 
+func (impl *reconnectingImpl) SetWriteDeadline(deadline time.Time) error {
+	return impl.peer.SetWriteDeadline(deadline)
+}
+
 type reconnectingImpl struct {
 	peer                transport.Conn
 	id                  *identity.TokenId

diff --git a/websockets/ws_underlay.go b/websockets/ws_underlay.go
@@ -115,3 +115,7 @@ func (self *Underlay) Headers() map[int32][]byte {
 func (self *Underlay) SetWriteTimeout(duration time.Duration) error {
 	return self.peer.SetWriteDeadline(time.Now().Add(duration))
 }
+
+func (self *Underlay) SetWriteDeadline(deadline time.Time) error {
+	return self.peer.SetWriteDeadline(deadline)
+}
diff --git a/ws_impl.go b/ws_impl.go
@@ -41,6 +41,10 @@ func (impl *wsImpl) SetWriteTimeout(duration time.Duration) error {
 	return impl.peer.SetWriteDeadline(time.Now().Add(duration))
 }
 
+func (self *wsImpl) SetWriteDeadline(deadline time.Time) error {
+	return self.peer.SetWriteDeadline(deadline)
+}
+
 func (impl *wsImpl) Rx() (*Message, error) {
 	if impl.closed {
 		return nil, errors.New("underlay closed")