快捷導(dǎo)航

使用Go實(shí)現(xiàn)一個(gè)百行聊天服務(wù)器的示例代碼

更新時(shí)間：2023年12月17日 09:17:40 作者：煙草的香味.

前段時(shí)間, redis作者整了個(gè)c語言版本的聊天服務(wù)器,代碼量攏共不過百行,于是, 心血來潮下, 我也整了個(gè)Go語言版本, 簡單來說就是實(shí)現(xiàn)了一個(gè)聊天室的功能,文中通過代碼示例給大家介紹的非常詳細(xì),需要的朋友可以參考下

前段時(shí)間, redis作者不是整了個(gè)c語言版本的聊天服務(wù)器嘛, 地址, 代碼量攏共不過百行.

于是, 心血來潮下, 我也整了個(gè)Go語言版本. 簡單來說就是實(shí)現(xiàn)了一個(gè)聊天室的功能. 將所有注釋空行都去掉, 剛好100行實(shí)現(xiàn). 廢話不多說, 先上代碼:

package main

import (
	"fmt"
	"log"
	"net"
	"strings"
	"sync"
)

type Server struct { // 服務(wù)端內(nèi)容
	clients map[string]*Client
	lock    sync.Mutex
}

func (s *Server) delClient(client *Client) { // 客戶端關(guān)閉
	s.lock.Lock()
	defer s.lock.Unlock()
	delete(s.clients, client.name)
}

func (s *Server) addClient(client *Client) { // 客戶端關(guān)閉
	s.lock.Lock()
	defer s.lock.Unlock()
	s.clients[client.name] = client
}

func (s *Server) sendMsgToOtherClient(msg string, client *Client) { // 將消息發(fā)送給其他所有客戶端
	s.lock.Lock()
	defer s.lock.Unlock()
	// 將消息轉(zhuǎn)發(fā)給其他客戶端
	for _, c := range s.clients {
		if c != client {
			c.msgCh <- "msg-> " + client.name + ": " + msg + "\n"
		}
	}
}

type Client struct { // 定義客戶端
	conn   net.Conn
	name   string      // 當(dāng)前客戶端的名稱
	msgCh  chan string // 發(fā)送消息的管道
	server *Server
}

func (c *Client) receive() { // 接收消息
	for msg := range c.msgCh {
		_, _ = c.conn.Write([]byte(msg))
	}
}

func (c *Client) close() {
	c.server.delClient(c)
	close(c.msgCh)
	_ = c.conn.Close()
}

func (c *Client) handle() { // 開始處理連接
	_, _ = c.conn.Write([]byte(fmt.Sprintf("hello %s!\n", c.name))) // 發(fā)送歡迎信息
	c.server.sendMsgToOtherClient("join", c)                        // 通知大家, 有人加入了聊天室
	defer c.close()
	for {
		buf := make([]byte, 2048)
		n, err := c.conn.Read(buf) // 接收客戶端發(fā)送的消息
		if err != nil {
			log.Printf("receive client data error: %s", err.Error())
			return
		}

		msg := strings.TrimSpace(string(buf[:n]))
		if len(msg) == 0 {
			continue
		}
		if msg == "quit" {
			c.server.sendMsgToOtherClient("quit", c) // 通知大家, 有人退出了聊天室
			return
		}
		c.server.sendMsgToOtherClient(msg, c)
	}
}

func main() {
	// 監(jiān)聽端口
	listener, err := net.Listen("tcp", ":8080")
	if err != nil {
		panic(err)
	}
	server := &Server{
		clients: make(map[string]*Client),
		lock:    sync.Mutex{},
	}

	nameIndex := 1
	for {
		conn, err := listener.Accept() // 建立連接
		if err != nil {
			log.Println(err)
			continue
		}

		client := &Client{
			conn:   conn,
			msgCh:  make(chan string, 100),
			name:   fmt.Sprintf("user%d", nameIndex),
			server: server,
		}
		nameIndex++
		server.addClient(client)
		go client.handle()
		go client.receive()
		log.Printf("new client: %s\n", conn.RemoteAddr())
	}
}

可以直接telnet作為客戶端連接, 實(shí)現(xiàn)的功能簡單來說就是一個(gè)大的聊天室, 用戶發(fā)的消息會同步發(fā)給所有用戶. (因代碼篇幅和復(fù)雜度原因, 就不對代碼做詳細(xì)說明了)

看下聊天室效果: