狼山资源网

Go语言多人聊天室项目实战
脚本专栏 2024/4/29 佚名
3 2

本文为大家分享了Go语言多人聊天室项目实战，供大家参考，具体内容如下

功能需求

  • 实现单撩
  • 实现群撩
  • 实现用户上线的全网通知
  • 实现用户昵称
  • 实现聊天日志的存储和查看

服务端实现

type Client struct {
 conn net.Conn
 name string
 addr string
}

var (
 //客户端信息,用昵称为键
 //clientsMap = make(map[string]net.Conn)
 clientsMap = make(map[string]Client)
)

func SHandleError(err error, why string) {
 if err != nil {
 fmt.Println(why, err)
 os.Exit(1)
 }
}

func main() {

 //建立服务端监听
 listener, e := net.Listen("tcp", "127.0.0.1:8888")
 SHandleError(e, "net.Listen")
 defer func() {
 for _, client := range clientsMap {
 client.conn.Write([]byte("all:服务器进入维护状态，大家都洗洗睡吧！"))
 }
 listener.Close()
 }()

 for {
 //循环接入所有女朋友
 conn, e := listener.Accept()
 SHandleError(e, "listener.Accept")
 clientAddr := conn.RemoteAddr()

 //TODO:接收并保存昵称
 buffer := make([]byte, 1024)
 var clientName string
 for {
 n, err := conn.Read(buffer)
 SHandleError(err, "conn.Read(buffer)")
 if n > 0 {
 clientName = string(buffer[:n])
 break
 }
 }
 fmt.Println(clientName + "上线了")

 //TODO:将每一个女朋友丢入map
 client := Client{conn, clientName, clientAddr.String()}
 clientsMap[clientName] = client

 //TODO:给已经在线的用户发送上线通知——使用昵称
 for _, client := range clientsMap {
 client.conn.Write([]byte(clientName + "上线了"))
 }

 //在单独的协程中与每一个具体的女朋友聊天
 go ioWithClient(client)
 }

 //设置优雅退出逻辑

}

//与一个Client做IO
func ioWithClient(client Client) {
 //clientAddr := conn.RemoteAddr().String()
 buffer := make([]byte, 1024)

 for {
 n, err := client.conn.Read(buffer)
 if err != io.EOF {
 SHandleError(err, "conn.Read")
 }

 if n > 0 {
 msg := string(buffer[:n])
 fmt.Printf("%s:%s\n", client.name, msg)

 //将客户端说的每一句话记录在【以他的名字命名的文件里】
 writeMsgToLog(msg, client)

 strs := strings.Split(msg, "#")
 if len(strs) > 1 {
 //all#hello
 //zqd#hello

 //要发送的目标昵称
 targetName := strs[0]
 targetMsg := strs[1]

 //TODO:使用昵称定位目标客户端的Conn
 if targetName == "all" {
  //群发消息
  for _, c := range clientsMap {
  c.conn.Write([]byte(client.name + ":" + targetMsg))
  }
 } else {
  //点对点消息
  for key, c := range clientsMap {
  if key == targetName {
  c.conn.Write([]byte(client.name + ":" + targetMsg))

  //在点对点消息的目标端也记录日志
  go writeMsgToLog(client.name + ":" + targetMsg,c)
  break
  }
  }
 }

 } else {

 //客户端主动下线
 if msg == "exit" {
  //将当前客户端从在线用户中除名
  //向其他用户发送下线通知
  for name, c := range clientsMap {
  if c == client {
  delete(clientsMap, name)
  } else {
  c.conn.Write([]byte(name + "下线了"))
  }
  }
 }else if strings.Index(msg,"log@")==0 {
  //log@all
  //log@张全蛋
  filterName := strings.Split(msg, "@")[1]
  //向客户端发送它的聊天日志
  go sendLog2Client(client,filterName)
 } else {
  client.conn.Write([]byte("已阅：" + msg))
 }

 }

 }
 }

}

//向客户端发送它的聊天日志
func sendLog2Client(client Client,filterName string) {
 //读取聊天日志
 logBytes, e := ioutil.ReadFile("D:/BJBlockChain1801/demos/W4/day1/01ChatRoomII/logs/" + client.name + ".log")
 SHandleError(e,"ioutil.ReadFile")

 if filterName != "all"{
 //查找与某个人的聊天记录
 //从内容中筛选出带有【filterName#或filterName:】的行，拼接起来
 logStr := string(logBytes)
 targetStr := ""
 lineSlice := strings.Split(logStr, "\n")
 for _,lineStr := range lineSlice{
 if len(lineStr)>20{
 contentStr := lineStr[20:]
 if strings.Index(contentStr,filterName+"#")==0 || strings.Index(contentStr,filterName+":")==0{
  targetStr += lineStr+"\n"
 }
 }
 }
 client.conn.Write([]byte(targetStr))
 }else{
 //查询所有的聊天记录
 //向客户端发送
 client.conn.Write(logBytes)
 }

}

//将客户端说的一句话记录在【以他的名字命名的文件里】
func writeMsgToLog(msg string, client Client) {
 //打开文件
 file, e := os.OpenFile(
 "D:/BJBlockChain1801/demos/W4/day1/01ChatRoomII/logs/"+client.name+".log",
 os.O_CREATE|os.O_WRONLY|os.O_APPEND,
 0644)
 SHandleError(e, "os.OpenFile")
 defer file.Close()

 //追加这句话
 logMsg := fmt.Sprintln(time.Now().Format("2006-01-02 15:04:05"), msg)
 file.Write([]byte(logMsg))
}

客户端实现

import (
 "net"
 "fmt"
 "os"
 "bufio"
 "io"
 "flag"
)

var (
 chanQuit = make(chan bool, 0)
 conn  net.Conn
)

func CHandleError(err error, why string) {
 if err != nil {
 fmt.Println(why, err)

 os.Exit(1)
 }
}

func main() {

 //TODO:在命令行参数中携带昵称
 nameInfo := [3]interface{}{"name", "无名氏", "昵称"}
 retValuesMap := GetCmdlineArgs(nameInfo)
 name := retValuesMap["name"].(string)

 //拨号连接，获得connection
 var e error
 conn, e = net.Dial("tcp", "127.0.0.1:8888")
 CHandleError(e, "net.Dial")
 defer func() {
 conn.Close()
 }()

 //在一条独立的协程中输入，并发送消息
 go handleSend(conn,name)

 //在一条独立的协程中接收服务端消息
 go handleReceive(conn)

 //设置优雅退出逻辑
 <-chanQuit

}

func handleReceive(conn net.Conn) {
 buffer := make([]byte, 1024)
 for {
 n, err := conn.Read(buffer)
 if err != io.EOF {
 CHandleError(err, "conn.Read")
 }

 if n > 0 {
 msg := string(buffer[:n])
 fmt.Println(msg)
 }
 }

}

func handleSend(conn net.Conn,name string) {
 //TODO:发送昵称到服务端
 _, err := conn.Write([]byte(name))
 CHandleError(err,"conn.Write([]byte(name))")

 reader := bufio.NewReader(os.Stdin)
 for {
 //读取标准输入
 lineBytes, _, _ := reader.ReadLine()

 //发送到服务端
 _, err := conn.Write(lineBytes)
 CHandleError(err, "conn.Write")

 //正常退出
 if string(lineBytes) == "exit" {
 os.Exit(0)
 }

 }
}

func GetCmdlineArgs(argInfos ...[3]interface{}) (retValuesMap map[string]interface{}) {

 fmt.Printf("type=%T,value=%v\n", argInfos, argInfos)

 //初始化返回结果
 retValuesMap = map[string]interface{}{}

 //预定义【用户可能输入的各种类型的指针】
 var strValuePtr *string
 var intValuePtr *int

 //预定义【用户可能输入的各种类型的指针】的容器
 //用户可能输入好几个string型的参数值，存放在好几个string型的指针中，将这些同种类型的指针放在同种类型的map中
 //例如：flag.Parse()了以后，可以根据【strValuePtrsMap["cmd"]】拿到【存放"cmd"值的指针】
 var strValuePtrsMap = map[string]*string{}
 var intValuePtrsMap = map[string]*int{}

 /* var floatValuePtr *float32
 var floatValuePtrsMap []*float32
 var boolValuePtr *bool
 var boolValuePtrsMap []*bool*/

 //遍历用户需要接受的所有命令定义
 for _, argArray := range argInfos {

 /*
 先把每个命令的名称和用法拿出来,
 这俩货都是string类型的，所有都可以通过argArray[i].(string)轻松愉快地获得其字符串
 一个叫“cmd”，一个叫“你想干嘛”
 "cmd"一会会用作map的key
 */
 //[3]interface{}
 //["cmd" "未知类型" "你想干嘛"]
 //["gid"  0  "要查询的商品ID"]
 //上面的破玩意类型[string 可能是任意类型 string]
 nameValue := argArray[0].(string) //拿到第一个元素的string值,是命令的name
 usageValue := argArray[2].(string) //拿到最后一个元素的string值，是命令的usage

 //判断argArray[1]的具体类型
 switch argArray[1].(type) {
 case string:
 //得到【存放cmd的指针】，cmd的值将在flag.Parse()以后才会有
 //cmdValuePtr = flag.String("cmd", argArray[1].(string), "你想干嘛")
 strValuePtr = flag.String(nameValue, argArray[1].(string), usageValue)

 //将这个破指针以"cmd"为键，存在【专门放置string型指针的map，即strValuePtrsMap】中
 strValuePtrsMap[nameValue] = strValuePtr

 case int:
 //得到【存放gid的指针】，gid的值将在flag.Parse()以后才会有
 //gidValuePtr = flag.String("gid", argArray[1].(int), "商品ID")
 intValuePtr = flag.Int(nameValue, argArray[1].(int), usageValue)

 //将这个破指针以"gid"为键，存在【专门放置int型指针的map，即intValuePtrsMap】中
 intValuePtrsMap[nameValue] = intValuePtr
 }

 }

 /*
 程序运行到这里，所有不同类型的【存值指针】都放在对相应类型的map中了
 flag.Parse()了以后，可以从map中以参数名字获取出【存值指针】，进而获得【用户输入的值】
 */

 //用户输入完了，解析，【用户输入的值】全都放在对应的【存值指针】中
 flag.Parse()

 /*
 遍历各种可能类型的【存值指针的map】
 */
 if len(strValuePtrsMap) > 0 {
 //从【cmd存值指针的map】中拿取cmd的值，还以cmd为键存入结果map中
 for k, vPtr := range strValuePtrsMap {
 retValuesMap[k] = *vPtr
 }
 }
 if len(intValuePtrsMap) > 0 {
 //从【gid存值指针的map】中拿取gid的值，还以gid为键存入结果map中
 for k, vPtr := range intValuePtrsMap {
 retValuesMap[k] = *vPtr
 }
 }

 //返回结果map
 return
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。

标签：
Go语言多人聊天室,Go多人聊天室,Go聊天室
免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！ 如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com
狼山资源网 Copyright www.pvsay.com

评论“Go语言多人聊天室项目实战”

暂无“Go语言多人聊天室项目实战”评论...
www.pvsay.com 狼山资源网
29,905影音资源
44,626技术资源
11,817软件资源
578,645站长资源

友情链接

杰晶网络 DDR爱好者之家 桃源资源网 杰网资源 富贵资源网 南强小屋 铁雪资源网 幽灵资源网 万梅资源网 狼山资源网 白云岛资源网 昆仑资源网 相思资源网 明霞山资源网 内蒙古资源网 黑松山资源网 饿虎岗资源网 大旗谷资源网 常春岛资源网 岱庙资源网 兴国资源网 快活林资源网 蝙蝠岛资源网 帝王谷资源网 白云城资源网 伏龙阁资源网 清风细雨楼 天枫庄资源网 圆月山庄资源网 无争山庄资源网 神水资源网 移花宫资源网 神剑山庄资源网 无为清净楼资源网 金钱帮资源网 丐帮资源网 华山资源网 极乐门资源网 小李飞刀资源网 凤求凰客栈 风云阁资源网 金狮镖局 鸳鸯亭资源网 千金楼资源网 更多链接
狼山资源网 Copyright www.pvsay.com