Go语言多人聊天室项目实战

本文为大家分享了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
}

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

(0)

相关推荐

  • 利用GO语言实现多人聊天室实例教程

    前言 运用go里面的net包中的相关方法来实现一个基于tcp的简单多人聊天室,用一个服务器来管理,主要反馈客户端是否连接成功并显示客户端输入的内容,并且发送给每一个在服务器上连接的客服端,下面话不多说了,来一起看看详细的介绍吧. 示例代码 服务器代码 // server package main import ( "fmt" "net" ) var ConnMap map[string]*net.TCPConn func checkErr(err error) in

  • 基于django channel实现websocket的聊天室的方法示例

    websocket 网易聊天室? ​ web微信? ​ 直播? 假如你工作以后,你的老板让你来开发一个内部的微信程序,你需要怎么办?我们先来分析一下里面的技术难点 消息的实时性? 实现群聊 现在有这样一个需求,老板给到你了,关乎你是否能转正?你要怎么做? 我们先说消息的实时性,按照我们目前的想法是我需要用http协议来做,那么http协议怎么来做那? 是不是要一直去访问我们的服务器,问服务器有没有人给我发消息,有没有人给我发消息?那么大家认为我多长时间去访问一次服务比较合适那? 1分钟1次?1分

  • golang基于websocket实现的简易聊天室程序

    本文实例讲述了golang基于websocket实现的简易聊天室.分享给大家供大家参考,具体如下: 先说点无关的,最近忙于工作没有更新博客,今天休息顺便把golang websocket研究了一下,挺好玩的,写了一个聊天室,分享给大家. websocket包 : code.google.com/p/go.net/websocket 文档 : http://go.pkgdoc.org/code.google.com/p/go.net/websocket 首先安装websocket包 复制代码 代码

  • Go语言多人聊天室项目实战

    本文为大家分享了Go语言多人聊天室项目实战,供大家参考,具体内容如下 功能需求 实现单撩 实现群撩 实现用户上线的全网通知 实现用户昵称 实现聊天日志的存储和查看 服务端实现 type Client struct { conn net.Conn name string addr string } var ( //客户端信息,用昵称为键 //clientsMap = make(map[string]net.Conn) clientsMap = make(map[string]Client) ) f

  • Java NIO实战之多人聊天室

    本文实例为大家分享了Java NIO实战之多人聊天室的具体代码,供大家参考,具体内容如下 NIO服务端 public class NioServer { /** * 启动 */ public void start() throws IOException { /** * 1. 创建Selector */ Selector selector = Selector.open(); /** * 2. 通过ServerSocketChannel创建channel通道 */ ServerSocketCha

  • nodejs+express搭建多人聊天室步骤

    前言 本文主要是笔者在学习node的时候,作为练手的一个小项目,花了几天空余时间,边码边写教程的一个过程.适用于对node理论知识看的多,实战少的同学,那么现在就让我们开始吧! 准备工作 新建一个文件夹 chatroom 在终端输入以下命令,按照步骤npm(没装过的去官网安装下node和npm)会自动给你生成一个package.json文件 安装express和socket.io package.json文件如下: //package.json { "name": "chat

  • 使用Angular和Nodejs、socket.io搭建聊天室及多人聊天室

    一,利用Node搭建静态服务器 这个是这个项目的底层支撑部分.用来支持静态资源文件像html, css, gif, jpg, png, javascript, json, plain text等等静态资源的访问.这里面是有一个mime类型的文件映射. mime.js /** * mime类型的 map * @ author Cheng Liufeng * @ date 2014/8/30 * 当请求静态服务器文件的类型 html, css, gif, jpg, png, javascript,

  • java课程设计做一个多人聊天室(socket+多线程)

    目录 课设要求 相关知识点 1.服务端能够看到所有在线用户 2.服务端能够强制用户下线 3.客户端能够看到所有在线用户 4.客户端要求能够向某个用户发送消息 5.运用JDBC实现持久化存储用户信息 6.使用JSONObject对象封装数据 7.使用Maven构建管理项目 类图 项目框架 核心代码 1.maven配置文件pom.xml 2.服务器端Server.java 4.客户端注册界面Register.java 5.客户端聊天界面Chat.java 6.用户实体User.java 7.JDBC

  • 300行代码实现go语言即时通讯聊天室

    学了2年Java,因为工作原因需要转Golang,3天时间学习了下go的基本语法,做这样一个聊天室小项目来巩固串联一下语法. 实现的功能:公聊,私聊,修改用户名 只用到了四个类: main.go:用来启动服务器 server.go:服务器相关代码 client.go:客户端相关代码,用户可以直接操作的可视化界面 user.go:用户类,用来封装用户的业务逻辑 架构图 完整代码 server.go package main import ( "fmt" "io" &q

  • 基于Nodejs利用socket.io实现多人聊天室

    socket.io简介 在Html5中存在着这样的一个新特性,引入了websocket,关于websocket的内部实现原理可以看这篇文章,这篇文章讲述了websocket无到有,根据协议,分析数据帧的头,进行构建websocket.虽然代码短,但可以很好地体现websocket的原理. ,这个特性提供了浏览器端和服务器端的基于TCP连接的双向通道.但是并不是所有的浏览器都支持websocket特性,故为了磨平浏览器间的差异,为开发者提供统一的接口,引入了socket.io模块.在不支持webs

  • Java基于中介者模式实现多人聊天室功能示例

    本文实例讲述了Java基于中介者模式实现多人聊天室功能.分享给大家供大家参考,具体如下: 一 模式定义 中介者模式,用一个中介对象来封装一系列对象之间的交互,使各个对象中不需要显示地引用其他对象实例,从而降低各个对象之间的耦合度,并且可以独立地改变对象间的交互关系. 二 模式举例 1 模式分析 我们借用多人聊天室来说明这一模式 2 中介模式静态类图 3 代码示例 3.1中介者接口--IMediator package com.demo.mediator; import com.demo.coll

  • java编程实现多人聊天室功能

    本文实例为大家分享了java实现多人聊天室的具体代码,供大家参考,具体内容如下 程序源代码及运行截图: server.java //server.java package Socket; import java.io.BufferedReader; import java.io.IOException; import java.io.InputStreamReader; import java.io.PrintWriter; import java.net.ServerSocket; impor

随机推荐