From 8fd144662721c10842db0ee56ef8b4f76370bee2 Mon Sep 17 00:00:00 2001
From: Shivaram Lingamneni <slingamn@cs.stanford.edu>
Date: Fri, 16 Mar 2018 18:16:04 -0400
Subject: [PATCH] tweak: clean up Socket.Close()

---
 irc/socket.go | 26 +++++++++++---------------
 1 file changed, 11 insertions(+), 15 deletions(-)

diff --git a/irc/socket.go b/irc/socket.go
index 9d3282ab..aa178530 100644
--- a/irc/socket.go
+++ b/irc/socket.go
@@ -52,18 +52,11 @@ func NewSocket(conn net.Conn, maxSendQBytes uint64) Socket {
 
 // Close stops a Socket from being able to send/receive any more data.
 func (socket *Socket) Close() {
-	alreadyClosed := func() bool {
-		socket.Lock()
-		defer socket.Unlock()
-		result := socket.closed
-		socket.closed = true
-		return result
-	}()
+	socket.Lock()
+	socket.closed = true
+	socket.Unlock()
 
-	if !alreadyClosed {
-		// force close loop to happen if it hasn't already
-		socket.Write("")
-	}
+	socket.wakeWriter()
 }
 
 // CertFP returns the fingerprint of the certificate provided by the client.
@@ -131,14 +124,17 @@ func (socket *Socket) Write(data string) (err error) {
 	}
 	socket.Unlock()
 
-	// notify the consumer that data is available
+	socket.wakeWriter()
+	return
+}
+
+// wakeWriter wakes up the goroutine that actually performs the write, without blocking
+func (socket *Socket) wakeWriter() {
+	// nonblocking send to the channel, no-op if it's full
 	select {
 	case socket.lineToSendExists <- true:
 	default:
-		// a notification is already in the queue, this is fine
 	}
-
-	return
 }
 
 // SetFinalData sets the final data to send when the SocketWriter closes.