實現邏輯
1、Golang 版本 1.3
2、實現原理：
1、主進程建立TCP監聽服務，并且初始化一個變量 talkChan := make(map[int]chan string)
2、當主進程ACCEPT連接請求后，利用go 啟動一個協程A去維持和客戶端的連接，把taokChan帶入到協程里
3、和客戶端建立連接的協程A，發送消息給客戶端，使其發送自己的用戶信息。
4、協程A在收到客戶端發送的用戶信息后，建立一個此用戶對應的管道 talkChan[uid] = make(chan string)
5、協程A再啟動一個協程A1去專門用來讀取客戶端發送的消息，并且用來判斷是發送給誰的消息，然后把消息放到對應的chan里。
6、協程A再啟動一個協程A2用來讀取此用戶對應的管道，如果里面有信息，則取出來發送到客戶端。
實現代碼
服務端測試代碼：server.go
package main
import (
"fmt"
"log"
"net"
"strconv"
)
func handleConnection(conn net.Conn, talkChan map[int]chan string) {
//fmt.Printf("%p/n", talkChan) //用以檢查是否是傳過來的指針
/*
定義當前用戶的uid
*/
var curUid int
var err error
/*
定義關閉通道
*/
var closed = make(chan bool)
defer func() {
fmt.Println("defer do : conn closed")
conn.Close()
fmt.Printf("delete userid [%v] from talkChan", curUid)
delete(talkChan, curUid)
}()
/**
* 提示用戶設置自己的uid， 如果沒設置，則不朝下執行
*/
for {
//提示客戶端設置用戶id
_, err = conn.Write([]byte("請設置用戶uid"))
if err != nil {
return
}
data := make([]byte, 1024)
c, err := conn.Read(data)
if err != nil {
//closed <- true //這樣會阻塞 | 后面取closed的for循環，沒有執行到。
return
}
sUid := string(data[0:c])
//轉成int類型
uid, _ := strconv.Atoi(sUid)
if uid < 1 {
continue
}
curUid = uid
talkChan[uid] = make(chan string)
//fmt.Println(conn, "have set uid ", uid, "can talk")
_, err = conn.Write([]byte("have set uid "+sUid+" can talk"))
if err != nil {
return
}
break
}
fmt.Println("err 3")
//當前所有的連接
fmt.Println(talkChan)
//讀取客戶端傳過來的數據
go func() {
for {
//不停的讀客戶端傳過來的數據
data := make([]byte, 1024)
c, err := conn.Read(data)
if err != nil {
fmt.Println("have no client write", err)
closed <- true //這里可以使用 | 因為是用用的go 新開的線程去處理的。 | 即便chan阻塞，后面的也會執行去讀 closed 這個chan
}
clientString := string(data[0:c])
//將客戶端過來的數據，寫到相應的chan里
if curUid == 3 {
talkChan[4] <- clientString
} else {
talkChan[3] <- clientString
}
}
}()
/*
從chan 里讀出給這個客戶端的數據 然后寫到該客戶端里
*/
go func() {
for {
talkString := <-talkChan[curUid]
_, err = conn.Write([]byte(talkString))
if err != nil {
closed <- true
}
}
}()
/*
檢查是否已經關閉連接 如果關閉則推出該線程 去執行defer語句
*/
for {
if <-closed {
return
}
}
}
func main() {
/**
建立監聽鏈接
*/
ln, err := net.Listen("tcp", "127.0.0.1:6010")
if err != nil {
panic(err)
}
//創建一個管道
//talkChan := map[f]
talkChan := make(map[int]chan string)
fmt.Printf("%p/n", talkChan)
/*
監聽是否有客戶端過來的連接請求
*/
for {
fmt.Println("wait connect...")
conn, err := ln.Accept()
if err != nil {
log.Fatal("get client connection error: ", err)
}
go handleConnection(conn, talkChan)
}
}
客戶端測試代碼：client.go
package main
import (
"fmt"
"math/rand"
"net"
)
func main() {
conn, err := net.Dial("tcp", "127.0.0.1:6010")
if err != nil {
panic(err)
}
fmt.Fprintf(conn, "hello server/n")
defer conn.Close()
go writeFromServer(conn)
for {
var talkContent string
fmt.Scanln(&talkContent)
if len(talkContent) > 0 {
_, err = conn.Write([]byte(talkContent))
if err != nil {
fmt.Println("write to server error")
return
}
}
}
}
func connect() {
conn, err := net.Dial("tcp", "127.0.0.1:6010")
if err != nil {
panic(err)
}
fmt.Fprintf(conn, "hello server/n")
defer conn.Close()
go writeFromServer(conn)
for {
var talkContent string
fmt.Scanln(&talkContent)
if len(talkContent) > 0 {
_, err = conn.Write([]byte(talkContent))
if err != nil {
fmt.Println("write to server error")
return
}
}
}
}
func writeFromServer(conn net.Conn) {
defer conn.Close()
for {
data := make([]byte, 1024)
c, err := conn.Read(data)
if err != nil {
fmt.Println("rand", rand.Intn(10), "have no server write", err)
return
}
fmt.Println(string(data[0:c]) + "/n ")
}
}
