Golang 簡易聊天室#

geminixiang/simple_chat

這周末學習了與MongoDB，使用了 mgo、mongo-go-driver 這兩個套件去學習，但…上手還需要一陣子，太久沒使用強型別，很多地方都還需要適應。

所以，這次先帶來 Go into Web!-iThome 上所分享的小專案來進行學習，也能更快去熟悉語法及debug、build、log等開發技巧，話不多說，讓我們開始吧。

知識點們#

melody#

https://github.com/olahol/melody Minimalist websocket framework for Go

是一個2018年已經棄坑(喂~)的 websocket framework，是 based on gorilla/websocket 上封裝的一個 package，更容易使用，以外如果要找還有在維護、Star還夠多的messaging server 可以參考 centrifugal/centrifugo

回到正題，而因為它是 websocket 框架，網站的靜態文件還是得找個人來幫忙掛載，在 melody 的官方教學文件上，就是與 gin 一起配合，就讓我們來看看以下的 code~

1
package main
2

3
import (
4
  "github.com/gin-gonic/gin"
5
  "gopkg.in/olahol/melody.v1"
6
  "net/http"
7
)
8

9
func main() {
10
  // 實例 gin、melody
11
  r := gin.Default()
12
  m := melody.New()
13

14
  // 把index.html掛載到 /
15
  r.GET("/", func(c *gin.Context) {
16
    http.ServeFile(c.Writer, c.Request, "index.html")
17
  })
18

19
  // melody's websocket 則是掛載到 /ws
20
  r.GET("/ws", func(c *gin.Context) {
21
    // gin的 response 與 request 轉手交給 melody 去 handle
22
    m.HandleRequest(c.Writer, c.Request)
23
  })
24

25
  // 那麼 melody 透過 handlemessage，當 message 進來時該如何處理，這邊的寫法是，有訊息進來將會告知所有 sessions
26
  m.HandleMessage(func(s *melody.Session, msg []byte) {
27
    m.Broadcast(msg)
28
  })
29

30
  // melody handleconnect 使用者連線，這邊我們定義websocket連線網址長得像
31
  // ws://localhost:5000/ws?id=Guest755
32
  m.HandleConnect(func(session *melody.Session) {
33
    // 取得url上的param id
34
    // := 冒號賦值，不用預先定義直接賦值
35
    id := session.Request.URL.Query().Get("id")
36
    // 廣播新的Message，這邊有個 GetByteMessage 會做 JSON encoding
37
    m.Broadcast(NewMessage("other", id, "加入聊天室").GetByteMessage())
38
  })
39

40
  // melody handleconnect 使用者離開後，傳訊息通知其他使用者
41
  m.HandleClose(func(session *melody.Session, i int, s string) error {
42
    id := session.Request.URL.Query().Get("id")
43
    m.Broadcast(NewMessage("other", id, "離開聊天室").GetByteMessage())
44
    return nil
45
  })
46

47
  // gin server run on port 5000
48
  r.Run(":5000")
49
}

Struct 結構資料#

非常像C++的struct，我目前使用起來可以說是一模一樣，可以定義一結構化的資料，使用已經定義好的Struct方式如下: Message{Event: "ABC", Name: "ABC", Content: "ABC", TimeStamp: time.Now()}

看起來有點繁瑣吧，這邊提供一個func NewMessage，將一些param mapping到定義好的struct，可以更方便的使用。

1
type Message struct {
2
  Event   string `json:"event"`
3
  Name    string `json:"name"`
4
  Content string `json:"content"`
5
  TimeStamp    time.Time `json:"timestamp"`
6
}
7

8
func NewMessage(event, name, content string) *Message {
9
  return &Message{
10
    Event:   event,
11
    Name:    name,
12
    Content: content,
13
    TimeStamp: time.Now(),
14
  }
15
}
16

17
func (m *Message) GetByteMessage() []byte {
18
  result, _ := json.Marshal(m)
19
  return result
20
}