Message.Close(): Allow the return path to go to a Link as well

1 files changed, 6 insertions, 1 deletions

diff --git a/msg.go b/msg.go
index 305b222..d0f0410 100644
--- a/msg.go
+++ b/msg.go
@@ -34,7 +34,12 @@ func (m *Message) Reply(t Tuple) {
 
 func (m *Message) Close() {
 	if m.ret != nil {
-		n := m.ret.obj.(*Session).node
+		var n *Node
+		if s, ok := m.ret.obj.(*Session); ok {
+			n = s.node
+		} else {
+			n = m.ret.obj.(*Link).node
+		}
 		n.lock.Lock()
 		m.ret.unref()
 		n.lock.Unlock()