From f2a0775208ee49b4a24c605e32e5f6eed68fb159 Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Wed, 1 Apr 2020 12:21:31 +0200
Subject: [PATCH] Set network.conn in user goroutine

One step closed to removing that lock.
---
 user.go | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/user.go b/user.go
index f6f1cdd..9e05ac3 100644
--- a/user.go
+++ b/user.go
@@ -77,20 +77,12 @@ func (net *network) run() {
 			continue
 		}
 
-		net.lock.Lock()
-		net.conn = uc
-		net.lock.Unlock()
-
 		net.user.events <- eventUpstreamConnected{uc}
 		if err := uc.readMessages(net.user.events); err != nil {
 			uc.logger.Printf("failed to handle messages: %v", err)
 		}
 		uc.Close()
 		net.user.events <- eventUpstreamDisconnected{uc}
-
-		net.lock.Lock()
-		net.conn = nil
-		net.lock.Unlock()
 	}
 }
 
@@ -176,12 +168,23 @@ func (u *user) run() {
 		switch e := e.(type) {
 		case eventUpstreamConnected:
 			uc := e.uc
+
+			uc.network.lock.Lock()
+			uc.network.conn = uc
+			uc.network.lock.Unlock()
+
 			uc.updateAway()
 		case eventUpstreamDisconnected:
 			uc := e.uc
+
+			uc.network.lock.Lock()
+			uc.network.conn = nil
+			uc.network.lock.Unlock()
+
 			for _, log := range uc.logs {
 				log.file.Close()
 			}
+
 			uc.endPendingLISTs(true)
 		case eventUpstreamMessage:
 			msg, uc := e.msg, e.uc