快捷導(dǎo)航

利用GO語(yǔ)言實(shí)現(xiàn)多人聊天室實(shí)例教程

更新時(shí)間：2018年03月25日 11:14:26 作者：shangguan_1234

聊天室的實(shí)現(xiàn)大家應(yīng)該都遇到過(guò)，這篇文章主要給大家介紹了關(guān)于利用GO語(yǔ)言實(shí)現(xiàn)多人聊天室的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起看看吧。

前言

運(yùn)用go里面的net包中的相關(guān)方法來(lái)實(shí)現(xiàn)一個(gè)基于tcp的簡(jiǎn)單多人聊天室，用一個(gè)服務(wù)器來(lái)管理，主要反饋客戶(hù)端是否連接成功并顯示客戶(hù)端輸入的內(nèi)容，并且發(fā)送給每一個(gè)在服務(wù)器上連接的客服端，下面話(huà)不多說(shuō)了，來(lái)一起看看詳細(xì)的介紹吧。

示例代碼

服務(wù)器代碼

// server 
package main 
import ( 
 "fmt" 
 "net" 
) 
 
var ConnMap map[string]*net.TCPConn 
func checkErr(err error) int { 
 if err != nil { 
 if err.Error() == "EOF" { 
  fmt.Println("用戶(hù)退出") 
  return 0 
 } 
 fmt.Println("發(fā)生錯(cuò)誤") 
 return -1 
 } 
 return 1 
} 
func say(tcpConn *net.TCPConn) { 
 for { 
 data := make([]byte, 256) 
 total, err := tcpConn.Read(data) 
 if err != nil { 
  fmt.Println(string(data[:total]), err) 
 } else { 
  fmt.Println(string(data[:total])) 
 } 
 flag := checkErr(err) 
 if flag == 0 { 
  break 
 } 
 for _, conn := range ConnMap { 
  if conn.RemoteAddr().String() == tcpConn.RemoteAddr().String() { 
  continue 
  } 
  conn.Write(data[:total]) 
 } 
 } 
} 
func main() { 
 //var conn net.TCPConn 
 //localAddr :=conn.LocalAddr().String() 
 //fmt.Println(localAddr) 
 //tcpAddr, _ := net.ResolveTCPAddr("tcp",localAddr) 
 tcpAddr, _ := net.ResolveTCPAddr("tcp", "192.168.128.216:8080") 
 tcpListen, _ := net.ListenTCP("tcp", tcpAddr) 
 ConnMap = make(map[string]*net.TCPConn) 
 for { 
 tcpConn, _ := tcpListen.AcceptTCP() 
 defer tcpConn.Close() 
 ConnMap[tcpConn.RemoteAddr().String()] = tcpConn 
 fmt.Println("連接客戶(hù)端信息：", tcpConn.RemoteAddr().String()) 
 go say(tcpConn) 
 } 
}

本來(lái)打算用系統(tǒng)的LocalAddr()函數(shù)來(lái)直接調(diào)用本地網(wǎng)絡(luò)地址，但是這里調(diào)用一直報(bào)格式的錯(cuò)誤，所以這里用本地地址直接輸在里面，端口是8080//192.168.247.128：8080，也可以改成和客戶(hù)端相似的代碼來(lái)進(jìn)行調(diào)用，但鑒于服務(wù)器唯一，所以就不做更改了

客戶(hù)端代碼

// client 
package main 
import ( 
 "fmt" 
 "net" 
 "os" 
) 

var ch chan int = make(chan int) 
var nickname string 
func reader(conn *net.TCPConn) { 
 buff := make([]byte, 256) 
 for { 
 j, err := conn.Read(buff) 
 if err != nil { 
  ch <- 1 
  break 
 } 
 fmt.Printf("%s\n", buff[0:j]) 
 } 
} 
func main() { 
 if len(os.Args) != 2 { 
 fmt.Fprintf(os.Stderr, "Usage:%s host:port", os.Args[0]) 
 os.Exit(1) 
 } 
 service := os.Args[1] 
 TcpAdd, _ := net.ResolveTCPAddr("tcp", service) 
 //TcpAdd, _ := net.ResolveTCPAddr("tcp", "localhost:8080") 
 conn, err := net.DialTCP("tcp", nil, TcpAdd) 
 if err != nil { 
 fmt.Println("服務(wù)沒(méi)打開(kāi)") 
 os.Exit(1) 
 } 
 defer conn.Close() 
 go reader(conn) 
 fmt.Println("請(qǐng)輸入昵稱(chēng)") 
 fmt.Scanln(&nickname) 
 fmt.Println("你的昵稱(chēng)是：", nickname) 
 //var str string 
 for { 
 var msg string 
 fmt.Scan(&msg) 
 fmt.Print("<" + nickname + ">" + "說(shuō):") 
 //for i, _ := range msg { 
 // fmt.Printf("%c", msg[i]) 
 //} 
 fmt.Println(msg) 
 b := []byte("<" + nickname + ">" + "說(shuō)：" + msg) 
 conn.Write(b) 
 select { 
 case <-ch: 
  fmt.Println("server發(fā)生錯(cuò)誤，請(qǐng)重新連接") 
  os.Exit(2) 
 default: 
 } 
 } 
}

客戶(hù)端主要就是向服務(wù)器發(fā)起連接，然后發(fā)送和接收數(shù)據(jù)，原計(jì)劃定義一個(gè)字符切片來(lái)存儲(chǔ)讀來(lái)的數(shù)據(jù)，但是出現(xiàn)了調(diào)用fmt.Println()來(lái)輸出是時(shí)候顯示ASCII碼的錯(cuò)誤，所以改成直接往一個(gè)空字符串里面讀寫(xiě)，然后再來(lái)調(diào)用fmt.Println() ，顯示出正確的字符串

運(yùn)行服務(wù)器

go build server.go

./server

或 go run server.go

運(yùn)行客戶(hù)端

go build client.go

./client 服務(wù)器地址：端口（上面例子是192.168.247.128：8080)