deregistering for clients

cmd/client.go

@@ -69,8 +69,7 @@ func runClient(cmd *cobra.Command, args []string) {
 	run := true
 	for run {
 		var oFrame data.Frame
-		// always use new decoder - reusing may lead to errors
+		gd := gob.NewDecoder(con) // always use new decoder
-		gd := gob.NewDecoder(con)
 		err = gd.Decode(&oFrame)
 		if err != nil {
 			log.Println("RGS:", err)

cmd/server.go

@@ -42,9 +42,8 @@ var clientKeys []int = make([]int, 100)
 func handleConn(conn *net.TCPConn) {
 	var hs data.Handshake
 	log.Println("Got a connection!")
-	// handshake
+	dec := gob.NewDecoder(conn)
-	dec := gob.NewDecoder(conn) // Decoder
+	dec.Decode(&hs) // decode handshake
-	dec.Decode(&hs)
 	log.Println("Got client! ID:", hs.ID, "dimensions:", hs.W, hs.H)
 	clientKeys = append(clientKeys, hs.ID)
 	sort.Ints(clientKeys)
@@ -68,8 +67,23 @@ func serveClients() {
 			id, client := k, clients[k]
 			if id > 0 {
 				enc := gob.NewEncoder(client.con)
-				enc.Encode(oFrame)
+				err := enc.Encode(oFrame)
+				if err != nil {
+					// client disconnected
 
+					//log.Println("BEFORE remove: clients:", clients, " clientKeys:", clientKeys)
+
+					// delete client
+					delete(clients, client.id)
+
+					// delete client key
+					// ugly as fuck in go to remove from a slice
+					// it *should* work though
+					idInKeys := sort.SearchInts(clientKeys, client.id)
+					clientKeys = append(clientKeys[:idInKeys], clientKeys[idInKeys+1:]...)
+
+					//log.Println("AFTER remove: clients:", clients, " clientKeys:", clientKeys)
+				}
 				//log.Println("ID:", id, "Client:", client)
 			}
 		}