go语言实现聊天服务器的示例代码
作者:Y_xx 发布时间:2024-04-26 17:17:39
标签:go,聊天服务器
看了两天 go 语言,是时候练练手了。
go 的 routine(例程) 和 chan(通道) 简直是神器,实现多线程(在 go 里准确的来说是 多例程)简直不要太轻松。
于是动手码了一个傻瓜版的黑框聊天器。
server 端:
监听 TCP 连接;支持自定义客户端命令;支持消息分发;理论上支持广播;...
package main
import (
"fmt"
"net"
"io"
"strconv"
"time"
"strings"
)
const (
NORMAL_MESSAGE = iota
LIST_MESSAGE
)
var clientSenders = make(map[string] chan string)
func send (addr string, conn *net.Conn){
senderChan := clientSenders[addr]
for s := range senderChan{
(*conn).Write([]byte(s))
}
}
func sendUsersInfo(addr string){
senderChan := clientSenders[addr]
if nil != senderChan{
ls := strconv.Itoa(LIST_MESSAGE)
cs := strconv.Itoa(NORMAL_MESSAGE) + "已登录客户端列表:\n"
i := 1
for k := range clientSenders{
a := ""
if k == addr {
a = "(我)"
}
cs = cs + strconv.Itoa(i) + ")" + k + a + "\n"
ls += k + "\n"
i ++
}
cs += "发送消息,可使用 1<-这是给1号客户端的消息\n(请使用英文以获取最佳体验)\n"
senderChan <- cs
time.Sleep(time.Millisecond * 300)
senderChan <- ls
// 发送格式化的列表
fmt.Println("已发送“登录用户信息”", addr)
} else{
fmt.Println("客户端接受通道不存在", addr)
}
}
func serve (conn *net.Conn){
connect := *conn
addr := connect.RemoteAddr().String()
fmt.Println(addr, "接入服务")
senderChan := make(chan string, 3)
clientSenders[addr] = senderChan
// 启动发送
go send(addr, conn)
// 发送当前用户信息
go sendUsersInfo(addr)
buff := make([]byte, 10240)
for {
n, err := connect.Read(buff)
if err != nil {
if err == io.EOF {
fmt.Println("客户端断开链接,", addr)
delete(clientSenders, addr)
return
} else{
fmt.Println(err)
}
}
msg := string(buff[:n])
// 刷新客户端列表
if msg == "ls\n" {
go sendUsersInfo(addr)
continue
}
// 提取数据
msgs := strings.Split(msg, "<-")
if len(msg) < 2{
senderChan <- string("数据格式不正确,请联系开发者")
continue
}
aimAddr := msgs[0]
aimSender := clientSenders[aimAddr]
if aimSender == nil {
senderChan <- string("客户端已下线,使用 ls 命令获取最新的客户端列表")
continue
}
aimSender <- strconv.Itoa(NORMAL_MESSAGE) + "[from:" + addr + "]:" + strings.Join(msgs[1:], "<-")
}
}
func main(){
addr := ":8080"
listener, err := net.Listen("tcp", addr)
if err != nil{
fmt.Println(err)
return
}
// 启动消息调度器
defer listener.Close()
// 启动连接监听
for {
conn, err := listener.Accept()
if err != nil {
fmt.Println(err)
continue
}
go serve(&conn)
}
}
客户端:
支持断线重连;支持给特定其他客户端发信息
package main
import (
"net"
"fmt"
"io"
"os"
"bufio"
"sync"
"time"
"strings"
"strconv"
)
var conn *net.Conn
var addrs []string
const (
NORMAL_MESSAGE = iota
LIST_MESSAGE
)
func read(conn2 *net.Conn){
defer func() {
fmt.Println("尝试重连")
go connectServer()
}()
connect := *conn2
buff := make([]byte, 20140)
for {
n, err := connect.Read(buff)
if err != nil {
if err == io.EOF{
fmt.Println("结束")
(*conn2).Close()
conn = nil
return
} else{
fmt.Println(err)
}
}
msg := string(buff[:n])
t, err := strconv.Atoi(string(msg[0]))
msg = msg[1:]
switch t {
case NORMAL_MESSAGE:
fmt.Print(msg)
break
case LIST_MESSAGE:
// 解析客户端列表数据
addrs = strings.Split(msg, "\n")
fmt.Println("已接收客户端列表。\n")
break
default:
fmt.Print(msg)
break
}
}
}
func connectServer(){
addr := "192.168.99.236:8080"
fmt.Println("等待服务器开启中")
conn2, err := net.Dial("tcp", addr)
if err != nil {
fmt.Print(err)
fmt.Println("连接失败,10s后尝试")
time.Sleep(10 * time.Second)
go connectServer()
return
}
fmt.Println("已连接")
conn = &conn2
go read(&conn2)
}
func send (){
inputReader := bufio.NewReader(os.Stdout)
for {
input, err := inputReader.ReadString('\n')
if err != nil {
if err == io.EOF{
return
} else{
fmt.Println(err)
}
}
if input == "ls\n" {
(*conn).Write([]byte(input))
continue
}
msgs := strings.Split(input, "<-")
if len(msgs) < 2 {
fmt.Println("发送的姿势不正确,应该像这样 1<-给1号发送消息\n")
continue
}
index, err := strconv.Atoi(msgs[0])
if err != nil {
fmt.Println("发送的姿势不正确,应该像这样 1<-给1号发送消息\n")
continue
}
if len(addrs) <= index {
fmt.Println("不存在第" + strconv.Itoa(index) + "个客户端\n")
continue
}
addr := addrs[index-1]
input = addr + "<-" + strings.Join(msgs[1:], "<-")
if nil != conn {
(*conn).Write([]byte(input))
}
}
}
func main (){
var wg sync.WaitGroup
wg.Add(2)
go connectServer()
go send()
wg.Wait()
defer func() {
if nil != conn {
(*conn).Close()
}
}()
}
来源:https://segmentfault.com/a/1190000015950984
0
投稿
猜你喜欢
- 这年头如果用 python3 做条形码的,肯定(推荐)用 pystrich 。这货官方文档貌似都没写到支持 Code128 ,但是居然有这个
- MySQL启动报错window10上安装了MySQL之前使用都是执行net start mysql启动,执行net stop mysql关闭
- git 报错信息:OpenSSL SSL_read: Connection was reset, errno 10054Git 中 push
- 所谓异常指的是程序的执行出现了非预期行为,就好比现实中的做一件事过程中总会出现一些意外的事。异常的处理是跨越编程语言的,和具体的编程细节相比
- 因为python打包的时候需要用到ico图片格式,网上看好的图片又没有ico,于是自己寻找了一下python转换图片格式的方法,彻底解决这个
- 字符函数——返回字符值这些函数全都接收的是字符族类型的参数(CHR除外)并且返回字符值.除了特别说明的之外,这些函数大部分返回VARCHAR
- 本文实例讲述了Python数学形态学。分享给大家供大家参考,具体如下:一 原始随机图像1、代码import numpy as npimpor
- 前言图像分割是许多计算机视觉应用中的关键处理步骤,通常用于将图像划分为不同的区域,这些区域常常对应于真实世界的对象。因此,图像分割是图像识别
- 一、数据引擎简介在MySQL 5.1中,MySQL AB引入了新的插件式存储引擎体系结构,允许将存储引擎加载到正在运新的MySQL
- 目录pyspark创建DataFrameRDD和DataFrame使用二元组创建DataFrame使用键值对创建DataFrame使用rdd
- 使用python生成一个图片验证码,随机的,可以由于验证人机和别的啊,很方便很简单导入模块import randomfrom PIL imp
- MySQL Version确认(版本确认)的几个方法1.SHOW VARIABLES LIKE 'VERSION';mysq
- 本文实例讲述了javascript insertAfter()定义与用法。分享给大家供大家参考,具体如下:HTML部分:<div id
- pynput这个库让你可以控制和监控输入设备。对于每一种输入设备,它包含一个子包来控制和监控该种输入设备:pynput.mouse:包含控制
- 在 Python 的项目中,如何管理所用的全部依赖库呢?最主流的做法是维护一份“requirements.txt”,记录下依赖库的名字及其版
- 把中文字符转换回Unicode字符。见下列代码:<%function unicode(str)dim i,j,c,
- 在PHP中,有两种包含外部文件的方式,分别是include和require。他们之间有什么不同呢?如果文件不存在或发生了错误,require
- 前言这篇文章主要记录一下平时自己实践得到的, 博客中学习的以及在一些项目源码中看到的 javascript 技巧。有些东西可以说是奇淫技巧,
- 看python社区大妈组织的内容里边有一篇讲python内存优化的,用到了__slots__。然后查了一下,总结一下。感觉非常有用pytho
- 有时候想同时(同级)展示多个视图,而不是嵌套展示,例如创建一个布局,有 sidebar(侧导航) 和 main(主内容) 两个视图,这个时候