一百行Golang代码实现简单并发聊天室

项目介绍:Golang100行代码实现高并发聊天室,其中实现的功能有:上下线广播,私聊,用户改名,超时强踢,在线用户检测等

在开始项目前,我们需要理解贯穿这整个项目的两个重要变量,若能理解这两个变量的使用,那么并发聊天室项目会变得手到擒来。第一个是onlinemap全局map,第二个是Message全局channel。

取名为onlinemap的全局map类型为map[string][client],这个全局字典是用来存储当前在此聊天室的用户的,key值是string类型,为用户的ip地址+Port端口,对应的value值为一个结构体,结构体内有此用户的姓名,地址和管道(用来给每一个用户传输信息,服务于Message全局通道)

取名为Message的全局channel也贯穿在整段代码中,向其中传送数据时,Message会在另一个go程里向其他每一个在线用户的管道中发送内容,随后在另一个go程里每一个用户的管道会向对应用户转发内容。如此可以实现上下线广播,群聊的功能。而每一个用户私有的管道可以实现私聊功能。

这个图详细阐述了这段代码的工作流程。

理解以上内容 下面我们再来看代码,就会很轻松,如果还是一头雾水也不要着急,小编会在下面每一行代码都加上精准通俗的注释,不多说,上代码。

package main
import (
  "net"
  "fmt"
  "strings"
  "time"
)
//定义的此结构体为全局map的value值,包括每一个用户的姓名,ip地址和私人管道
type client struct {
  name string
  addr string
  C  chan string
}
/*这个函数是将私人管道中的内容发送给用户,配合全局管道Message使用可以实现广播的功能,
单独使用可以实现私聊的功能*/
func writemsg2client(clinet client,conn net.Conn) {
  for m := range clinet.C {
    conn.Write([]byte(m + "\n"))
  }
}
//这只是一个封装好用来统一(发送信息格式)的小函数,不用在意
func makemsg(name string, addr string, s string) string {
  return "[" + addr + "]" + name + s
}
//每一个进入聊天室的用户都将启动一个handleconn的go程来处理事件
func handleconn(conn net.Conn) {
  defer conn.Close()
/*用户连接进来以后要初始化全局map,把自己的信息加入到字典里,相当于进到聊天室里之前要登
记一下个人信息,注意姓名初始为ip地址。*/
  addr := conn.RemoteAddr().String()
  fmt.Printf("用户%s进入了房间\n", addr)
  client := client{addr, addr, make(chan string)}
  //在这里启动子go程,功能上面已经提及
  go writemsg2client(client,conn)
  onlinemap[addr] = client
  //登录进来一切准备就绪后就给所有人广播上线信息啦
  Message <- makemsg(client.name, addr, "login")
  //下面这三个变量服务于下面一些小功能
  var haschat=make(chan bool)
  var ifquit=make(chan bool)
  var flag bool
  //从这单独开启一个go程来读取用户输入的信息
  go func() {
    buf:=make([]byte,4096)
    for {
      n,_:=conn.Read(buf)
      if n==0 {
        fmt.Printf("%s离开了房间\n",client.name)
        ifquit<-true
        return
      }
      //改名功能的实现
      if string(buf[:7])=="Rename|" {
        client.name=strings.Split(string(buf[:n-1]),"|")[1]
        onlinemap[addr]=client
        conn.Write([]byte("rename success\n"))
      }else if string(buf[:n-1])=="/who"{
      //查询在线用户信息的功能
        for _,s:=range onlinemap{
          conn.Write([]byte(s.name+"online\n"))
        }
      }else if string(buf[:2])=="m|"&&strings.Count(string(buf[:n]),"|")==2 {
/*私聊功能的实现,其实私聊功能就是跳过了往全局Message里传输信息,
改为直接向私人管道里传输信息*/
        flag=false
        slice:=strings.Split(string(buf[:n-1]),"|")
        for _,a:=range onlinemap{
        //遍历所有在线用户,向指定的用户管道中发送信息
          if a.name==slice[1]{
            flag=true
            a.C<-makemsg(client.name,addr,slice[2])
            conn.Write([]byte("send success"))
          }
        }
        if flag {
          conn.Write([]byte("no such man or not online"))
        }
      } else {
        Message<-makemsg(client.name,addr,string(buf[:n-1]))
      }
      haschat<-true
    }
  }()
  for {
    select {
    case <-haschat:
    //超时强踢
    case <-time.After(time.Minute*3):
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    case <-ifquit:
    //退出处理
      delete(onlinemap,addr)
      Message<-makemsg(client.name,addr,"out time to leave")
      close(client.C)
      return
    }
  }
}
//这个函数用来将全局Message中的内容全部塞到私人管道C里,实现上下线广播和群聊的功能
func Manager() {
  for {
    msg := <-Message
    for _, s := range onlinemap {
      s.C <- msg
    }
  }
}
var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)
//主函数
func main() {
  listener, _ := net.Listen("tcp", "127.0.0.1:6666")
  defer listener.Close()
  //提前开启全局Message的go程,防止被阻塞
  go Manager()
  for {
    conn, err := listener.Accept()
    if err != nil {
      fmt.Println("accept err", err)
      continue
    }
    //每一个连接进来的用户都会被分配进入一个子go程,用来处理上面我们提到的各种功能
    go handleconn(conn)
  }
}

以上就是一个简单的高并发聊天室了,依托于go语言的强大,去掉注释只剩下不到一百行,虽然功能简单,但是涉及到channel,socket,select,map,string及go的使用,有利于此阶段在学的小伙伴们学习交流,大家有什么疑问或者想法可以在下面给我留言哦。

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

(0)

相关推荐

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

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

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

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

  • 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

  • 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包 复制代码 代码

  • 一百行Golang代码实现简单并发聊天室

    项目介绍:Golang100行代码实现高并发聊天室,其中实现的功能有:上下线广播,私聊,用户改名,超时强踢,在线用户检测等 在开始项目前,我们需要理解贯穿这整个项目的两个重要变量,若能理解这两个变量的使用,那么并发聊天室项目会变得手到擒来.第一个是onlinemap全局map,第二个是Message全局channel. 取名为onlinemap的全局map类型为map[string][client],这个全局字典是用来存储当前在此聊天室的用户的,key值是string类型,为用户的ip地址+Po

  • 玩转NODE.JS(四)-搭建简单的聊天室的代码

    Nodejs好久没有跟进了,最近想用它搞一个聊天室,然后便偶遇了socket.io这个东东,说是可以用它来简单的实现实时双向的基于事件的通讯机制.我便看了一些个教程使用它来搭建一个超级简单的聊天室. 初始化项目 在电脑里新建一个文件夹,叫做"chatroom",然后使用npm进行初始化: $ npm init 然后根据提示以及相关信息一步一步输入,当然也可以一路回车下去,之后会在项目里生成一个package.json文件,里面的信息如下: $ cat package.json { &q

  • Go语言实现一个简单的并发聊天室的项目实战

    目录 写在前面 并发聊天服务器 具体代码 服务端 客户端 总结 写在前面 Go语言在很多方面天然的具备很多便捷性,譬如网络编程,并发编程.而通道则又是Go语言实现并发编程的重要工具,因为其承担着通道之间互相通信的重任.并且因为其本身就是并发安全的,所以在某些场景下是非常好用的. 并发聊天服务器 这里主要是实现一个简单的并发聊天服务器.首先,客户端可以在服务器中注册自己的信息(登录以及退出),客户端发出的所有的信息由服务器向各个客户端进行转发,或者换句话说是广播. 具体代码 服务端 说的再多,没有

  • 一百行JS代码实现一个校验工具

    做过校验需求的小伙伴们都知道,校验其实是个麻烦事. 规则多,需要校验的字段多,都给我们前端带来巨大的工作量. 一个不小心,代码里就出现了不少if else等不可维护的代码. 因此,我觉得一个团队或者是一个项目,需要一个校验工具,简化我们的工作. 首先,参考一下 Joi.只看这一小段代码: Joi.string().alphanum().min(3).max(30).required() 我希望我的校验工具Coi也是链式调用,链式调用可以极大的简化代码. 校验呢,其实主要就3个入参:需要校验的数据

  • ASP建立一个简单的聊天室

    经过一个阶段的asp学习,下面我们结合所学过的内容建立一个最简单的聊天室,虽然很简单,但是大家可以通过它来掌握一个聊天室建立的基本过程,并且可以不断的完善其功能. 下面介绍其主要步骤: 1,添加Global.asa文件里面的代码.这部分代码主要处理Application_onStart事件,在此事件中,定义了一个有15个元素的数据,并把它赋给了一个Application对象的属性.Global.asa文件的内容如下. <SCRIPT LANGUAGE="VBScript" RUN

  • Java基于UDP协议实现简单的聊天室程序

    最近比较闲,一直在抽空回顾一些Java方面的技术应用. 今天没什么事做,基于UDP协议,写了一个非常简单的聊天室程序. 现在的工作,很少用到socket,也算是对Java网络编程方面的一个简单回忆. 先看一下效果: 实现的效果可以说是非常非常简单,但还是可以简单的看到一个实现原理.  "聊天室001"的用户,小红和小绿相互聊了两句,"聊天室002"的小黑无人理会,在一旁寂寞着. 看一下代码实现: 1.首先是消息服务器的实现,功能很简单: •将客户端的信息(进入了哪一

  • Java实现简单局域网聊天室

    本文实例为大家分享了Java实现简单局域网聊天室的具体代码,供大家参考,具体内容如下 Java 的Socket编程: 1.TCP协议是面向连接的.可靠的.有序的.以字节流的方式发送数据,通过三次握手方式建立连接,形成传输数据的通道,在连接中进行大量数据的传输,效率会稍低 2.Java中基于TCP协议实现网络通信的类 客户端的Socket类 服务器端的ServerSocket类 3.Socket通信的步骤 ① 创建ServerSocket和Socket ② 打开连接到Socket的输入/输出流 ③

  • GO使用socket和channel实现简单控制台聊天室

    使用socket和channel,实现简单控制台聊天室 这里使用socket和channel,演示在GO中如何编写一个简单网络程序 功能分析 聊天室主要功能:用户可以加入/离开聊天室:每个用户发送的消息,广播给所有人 聊天室分为客户端和服务端,客户端负责发送消息和打印服务器消息,服务器负责接收客户端消息,并广播给所有人 客户端可以使用telnet程序 服务端是需要实现的.需要实现的功能, 如何保存多个客户端的连接,管理连接的接入与断开 如何接收和广播客户端消息 实现思路 通过功能分析,拆分为聊天

  • Android编写简单的聊天室应用

    最近写了一个简单的聊天室应用,可以发送表情,更改头像这些功能.主要技术点就是怎样把表情图片放到textview等Ui控件中展示.这里废话不多说,下面是效果图: 这里主要讲下怎样把文本替换到表情,先说下思路,首先我们的图片是保存在本地资源目录drawable中而所有的资源文件都是R这个类来管理,所以我们可以利用正则表达式找出图片id包装成ImageSpan然后把ImageSpan放到SpannableString中,最后把SpannableString放入edittext中,下面是源码: pack

  • c#基于WinForm的Socket实现简单的聊天室 IM

    1:什么是Socket 所谓套接字(Socket),就是对网络中不同主机上的应用进程之间进行双向通信的端点的抽象. 一个套接字就是网络上进程通信的一端,提供了应用层进程利用网络协议交换数据的机制. 从所处的地位来讲,套接字上联应用进程,下联网络协议栈,是应用程序通过网络协议进行通信的接口,是应用程序与网络协议根进行交互的接口. 2:客服端和服务端的通信简单流程 3:服务端Code: using System; using System.Collections.Generic; using Sys

随机推荐