websocket bugfix

zhima1234 · zhima1234 · commit 40031e08a75b · 2017-08-22T19:54:04.000+08:00
diff --git a/src/logic/book.go b/src/logic/book.go
@@ -251,7 +251,11 @@ func (this *book) BroadcastAllUsersMessage(message *Message) {
 
 	this.rwMutex.RLock()
 	defer this.rwMutex.RUnlock()
-	for _, userData := range this.users {
+	for uid, userData := range this.users {
+		if userData.Len() == 0 {
+			delete(this.users, uid)
+			delete(this.uids, uid)
+		}
 		userData.SendMessage(message)
 	}
 }
@@ -266,6 +270,11 @@ func (this *book) BroadcastToOthersMessage(message *Message, myself int) {
 		if uid == myself {
 			continue
 		}
+
+		if userData.Len() == 0 {
+			delete(this.users, uid)
+			delete(this.uids, uid)
+		}
 		userData.SendMessage(message)
 	}
 }
diff --git a/template/wiki/content.html b/template/wiki/content.html
@@ -86,7 +86,7 @@ <h5>本页贡献者:</h5>
 <script type="text/javascript" src="/static/js/libs/emojis.js"></script>
 <script type="text/javascript" src="/static/js/libs/plupload.full.min.js"></script>
 <script type="text/javascript" src="/static/js/wiki.js?v=0.1"></script>
-<script type="text/javascript" src="/static/js/comment.js?v=1.2"></script>
+<!-- <script type="text/javascript" src="/static/js/comment.js?v=1.2"></script> -->
 <script type="text/javascript" src="/static/js/puploader.js?v=1.1"></script>
 <script type="text/javascript">
 // 需要加载的侧边栏

Original file line number	Diff line number	Diff line change
`@@ -251,7 +251,11 @@ func (this book) BroadcastAllUsersMessage(message Message) {`
`251`	`251`
`252`	`252`	`this.rwMutex.RLock()`
`253`	`253`	`defer this.rwMutex.RUnlock()`
`254`		`- for _, userData := range this.users {`
	`254`	`+ for uid, userData := range this.users {`
	`255`	`+ if userData.Len() == 0 {`
	`256`	`+ delete(this.users, uid)`
	`257`	`+ delete(this.uids, uid)`
	`258`	`+ }`
`255`	`259`	`userData.SendMessage(message)`
`256`	`260`	`}`
`257`	`261`	`}`
`@@ -266,6 +270,11 @@ func (this book) BroadcastToOthersMessage(message Message, myself int) {`
`266`	`270`	`if uid == myself {`
`267`	`271`	`continue`
`268`	`272`	`}`
	`273`	`+`
	`274`	`+ if userData.Len() == 0 {`
	`275`	`+ delete(this.users, uid)`
	`276`	`+ delete(this.uids, uid)`
	`277`	`+ }`
`269`	`278`	`userData.SendMessage(message)`
`270`	`279`	`}`
`271`	`280`	`}`