300行代碼實現go語言即時通訊聊天室
學瞭2年Java,因為工作原因需要轉Golang,3天時間學習瞭下go的基本語法,做這樣一個聊天室小項目來鞏固串聯一下語法。
實現的功能:公聊,私聊,修改用戶名
隻用到瞭四個類:
- main.go:用來啟動服務器
- server.go:服務器相關代碼
- client.go:客戶端相關代碼,用戶可以直接操作的可視化界面
- user.go:用戶類,用來封裝用戶的業務邏輯
架構圖
完整代碼
server.go
package main import ( "fmt" "io" "net" "sync" "time" ) type Server struct { Ip string Port int //在線用戶列表 OnlineMap map[string]*User mapLock sync.RWMutex //消息廣播的Channel Message chan string } func NewServer(ip string, port int) *Server { server := &Server{ Ip: ip, Port: port, OnlineMap: make(map[string]*User), Message: make(chan string), } return server } func (s *Server) Handler(conn net.Conn) { //業務邏輯 //fmt.Println("鏈接建立成功") user := NewUser(conn, s) user.Online() //監聽用戶是否活躍 isLive := make(chan bool) go func() { buf := make([]byte, 4096) for { n, error := conn.Read(buf) if n == 0 { user.Offline() return } if error != nil && error != io.EOF { fmt.Println("read error") } msg := string(buf[:n-1]) user.DoMessage(msg) //表示用戶活躍 isLive <- true } }() for { select { case <-isLive: //當前用戶活躍,不做任何時,激活select,重置定時器 case <-time.After(time.Second * 300): //超時,將user強制關閉 user.SendMsg("你被踢瞭") close(user.C) conn.Close() return } } } func (s *Server) ListenMessager() { for { msg := <-s.Message s.mapLock.Lock() for _, user := range s.OnlineMap { user.C <- msg } s.mapLock.Unlock() } } func (s *Server) BroadCast(user *User, msg string) { sendMsg := "[" + user.Addr + "]" + user.Name + ":" + msg s.Message <- sendMsg } func (s *Server) Start() { listener, error := net.Listen("tcp", fmt.Sprintf("%s:%d", s.Ip, s.Port)) if error != nil { fmt.Println("listener error...") return } defer listener.Close() go s.ListenMessager() for { conn, error := listener.Accept() if error != nil { fmt.Println("accept error...") continue } go s.Handler(conn) } }
client.go
package main import ( "flag" "fmt" "io" "net" "os" ) type Client struct { ServerIp string ServerPort int Name string conn net.Conn flag int } func NewClient(serverIp string, serverPort int) *Client { client := &Client{ ServerIp: serverIp, ServerPort: serverPort, flag: 9999, } conn, error := net.Dial("tcp", fmt.Sprintf("%s:%d", serverIp, serverPort)) if error != nil { fmt.Println("net dial error...") return nil } client.conn = conn return client } func (c *Client) menu() bool { var flag int fmt.Println("1.公聊模式") fmt.Println("2.私聊模式") fmt.Println("3.修改用戶名") fmt.Println("0.退出") fmt.Scanln(&flag) if flag >= 0 && flag <= 3 { c.flag = flag return true } else { fmt.Println(">>>>請輸入合法數字<<<<") return false } } //修改用戶名 func (c *Client) UpdateName() bool { fmt.Println(">>>>請輸入用戶名") fmt.Scanln(&c.Name) sendMsg := "rename|" + c.Name + "\n" _, error := c.conn.Write([]byte(sendMsg)) if error != nil { fmt.Println("conn.write error...") return false } return true } //公聊 func (c *Client) PublicChat() { var chatMsg string fmt.Println(">>>>請輸入聊天內容,輸入exit退出") fmt.Scanln(&chatMsg) for chatMsg != "exit" { if len(chatMsg) != 0 { msg := chatMsg + "\n" _, error := c.conn.Write([]byte(msg)) if error != nil { fmt.Println("conn.Write error....") break } } chatMsg = "" fmt.Println(">>>>請輸入聊天內容,輸入exit退出") fmt.Scanln(&chatMsg) } } //私聊 func (c *Client) PrivateChat() { var remoteUser string var chatMsg string c.SelectUsers() fmt.Println(">>>>請輸入聊天對象的用戶名,輸入exit退出") fmt.Scanln(&remoteUser) for remoteUser != "exit" { fmt.Println(">>>>請輸入聊天內容,輸入exit退出") fmt.Scanln(&chatMsg) for chatMsg != "exit" { if len(chatMsg) != 0 { msg := "to|" + remoteUser + "|" + chatMsg + "\n\n" _, error := c.conn.Write([]byte(msg)) if error != nil { fmt.Println("conn.Write error....") break } } chatMsg = "" fmt.Println(">>>>請輸入聊天內容,輸入exit退出") fmt.Scanln(&chatMsg) } c.SelectUsers() remoteUser = "" fmt.Println(">>>>請輸入聊天對象的用戶名,輸入exit退出") fmt.Scanln(&remoteUser) } } //查詢在線用戶 func (c *Client) SelectUsers() { sendMsg := "who\n" _, error := c.conn.Write([]byte(sendMsg)) if error != nil { fmt.Println("conn.Write error....") return } } //處理server返回的消息 func (c *Client) DealResponse() { io.Copy(os.Stdout, c.conn) } func (c *Client) Run() { for c.flag != 0 { for c.menu() != true { } switch c.flag { case 1: //公聊 c.PublicChat() case 2: //私聊 c.PrivateChat() case 3: //修改用戶名 c.UpdateName() } } } var serverIp string var serverPort int func init() { flag.StringVar(&serverIp, "ip", "127.0.0.1", "設置服務器IP地址(默認為127.0.0.1)") flag.IntVar(&serverPort, "port", 8888, "設置服務器端口(默認為8888)") } func main() { flag.Parse() client := NewClient(serverIp, serverPort) if client == nil { fmt.Println(">>>>鏈接服務器失敗") return } go client.DealResponse() fmt.Println(">>>>鏈接服務器成功") client.Run() }
user.go
package main import ( "net" "strings" ) type User struct { Name string Addr string C chan string conn net.Conn server *Server } func NewUser(conn net.Conn, server *Server) *User { userAddr := conn.RemoteAddr().String() user := &User{ Name: userAddr, Addr: userAddr, C: make(chan string), conn: conn, server: server, } go user.ListenMessage() return user } //用戶上線 func (u *User) Online() { u.server.mapLock.Lock() u.server.OnlineMap[u.Name] = u u.server.mapLock.Unlock() u.server.BroadCast(u, "上線") } //用戶下線 func (u *User) Offline() { u.server.mapLock.Lock() delete(u.server.OnlineMap, u.Name) u.server.mapLock.Unlock() u.server.BroadCast(u, "下線") } //給當前user的客戶端發送消息 func (u *User) SendMsg(msg string) { u.conn.Write([]byte(msg)) } //處理消息 func (u *User) DoMessage(msg string) { if msg == "who" { //查詢當前在線用戶 u.server.mapLock.Lock() for _, user := range u.server.OnlineMap { onlineMsg := "[" + user.Addr + "]" + user.Name + ":在線...\n" u.SendMsg(onlineMsg) } u.server.mapLock.Unlock() } else if len(msg) > 7 && msg[:7] == "rename|" { //修改用戶名 rename|xxx newName := strings.Split(msg, "|")[1] //判斷名字是否已經存在 _, ok := u.server.OnlineMap[newName] if ok { u.SendMsg("用戶名已存在\n") } else { u.server.mapLock.Lock() delete(u.server.OnlineMap, u.Name) u.server.OnlineMap[newName] = u u.server.mapLock.Unlock() u.Name = newName u.SendMsg("用戶名成功修改為:" + newName + "\n") } } else if len(msg) > 4 && msg[:3] == "to|" { //私聊 to|zhangsan|你好 //獲取對方用戶名 remoteName := strings.Split(msg, "|")[1] if remoteName == "" { u.SendMsg("用戶名格式不對\n") return } //獲取對方user remoteUser, ok := u.server.OnlineMap[remoteName] if !ok { u.SendMsg("用戶不存在\n") return } //獲取消息 msg := strings.Split(msg, "|")[2] if msg == "" { u.SendMsg("無消息內容,重新發送\n") } //發送消息 remoteUser.SendMsg(u.Name + "對您說:" + msg) } else { u.server.BroadCast(u, msg) } } func (u *User) ListenMessage() { for { msg := <-u.C u.conn.Write([]byte(msg + "\n")) } }
main.go
package main func main() { server := NewServer("127.0.0.1", 8888) server.Start() }
到此這篇關於300行代碼實現go語言即時通訊聊天室的文章就介紹到這瞭,更多相關go語言即時通訊聊天室內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!