[Golang] Simple chat room implementation

This is a creation in Article, where the information may have evolved or changed. Recently wrote a chrome websocket extension, server End with Java Netty Framework, later found Golang implementation websocket very simple, efficient, the following is an example, simple implementation of user login, broadcast, equivalent to chat room!

[Plain]View Plaincopyprint?

2. Package Main
6. Import (
8. "Code.google.com/p/go.net/websocket"
10. "Html/template"
12. "Log"
14. "Net/http"
16. "OS"
18. "Strings"
20. "Time"
22. )
26. Const (
28. LISTENADDR = "localhost:9527"//server address
30. )
34. VAR (
36. PWD, _ = os. GETWD ()
38. Roottemp = template. Must (template. Parsefiles (pwd + "/chat.html"))
40. JSON = WebSocket. JSON//Codec for JSON
42. Message = WebSocket. Message//codec for string, []byte
44. Activeclients = Make (map[clientconn]string)//Map containing clients
46. User = Make (map[string]string)
48. )
52. Initialize handlers and WebSocket handlers
54. Func init () {
56. User["AAA"] = "AAA"
58. user["BBB"] = "BBB"
60. user["test"] = "Test"
62. user["test2"] = "test2"
64. user["test3"] = "test3"
66. }
70. Client connection consists of the WebSocket and the client IP
72. Type Clientconn struct {
74. WebSocket *websocket. Conn
76. ClientIP string
78. }
82. WebSocket server to handle chat between clients
84. Func sockserver (ws *websocket. Conn) {
86. var err error
88. var clientmessage string
90. Use []byte if WebSocket binary type is blob or Arraybuffer
92. var clientmessage []byte
96. Cleanup on server side
98. Defer func () {
100. If Err = ws. Close (); Err! = Nil {
102. Log. Println ("Websocket could not be closed", err. Error ())
104. }
106. }()
110. Client: = ws. Request (). Remoteaddr
112. Log. PRINTLN ("Client connected:", client)
114. SOCKCLI: = clientconn{ws, client}
116. ACTIVECLIENTS[SOCKCLI] = ""
118. Log. Println ("Number of clients connected:", Len (activeclients))
122. For loop so the websocket stays open otherwise
124. It ' ll close after one receieve and Send
126. for {
128. If Err = message.receive (WS, &clientmessage); Err! = Nil {
130. If We cannot Read then the connection is closed
132. Log. Println ("Websocket disconnected Waiting", err. Error ())
134. Remove the WS client conn from our active clients
136. Delete (activeclients, SOCKCLI)
138. Log. Println ("Number of clients still connected:", Len (activeclients))
140. Return
142. }
146. var Msg_arr = strings. Split (clientmessage, "|")
148. If msg_arr[0] = = "Login" && len (msg_arr) = = 3 {
150. Name: = Msg_arr[1]
152. Pass: = Msg_arr[2]
156. If pass = = User[name] {
158. ACTIVECLIENTS[SOCKCLI] = Name
162. If Err = Message.send (WS, "login|" +name); Err! = Nil {
164. Log. Println ("Could not send Message to", client, err.) Error ())
166. }
168. } else {
170. Log. PRINTLN ("Login faild:", Clientmessage)
172. }
176. } else if msg_arr[0] = = "MSG" {
178. If ACTIVECLIENTS[SOCKCLI]! = "" {
180. Clientmessage = "Msg|" + time. Now (). Format ("2006-01-02 15:04:05") + "" + ACTIVECLIENTS[SOCKCLI] + "said:" + msg_arr[1]
182. For cs, na: = Range activeclients {
184. If na! = "" {
186. If Err = Message.send (Cs.websocket, clientmessage); Err! = Nil {
188. Log. Println ("Could not send Message to", Cs.clientip, err.) Error ())
190. }
192. }
194. }
196. }
198. }
200. }
202. }
206. Roothandler renders the template for the root page
208. Func Roothandler (w http. Responsewriter, req *http. Request) {
210. ERR: = Roottemp.execute (W, LISTENADDR)
212. If err! = Nil {
214. http. Error (W, err. Error (), HTTP. Statusinternalservererror)
216. }
218. }
222. Func Main () {
224. http. Handlefunc ("/", Roothandler)
226. http. Handle ("/socket", WebSocket. Handler (Sockserver))
228. ERR: = http. Listenandserve (listenaddr, nil)
230. If err! = Nil {
232. Panic ("Listenandserve:" + Err. Error ())
234. }
236. }