[golang]GO语言编写的基于UDP协议的简易聊天软件

Go 前端之家
前端之家收集整理的这篇文章主要介绍了[golang]GO语言编写的基于UDP协议的简易聊天软件，前端之家小编觉得挺不错的，现在分享给大家，也给大家做个参考。

初学GO不到两周，本着熟悉语言的目标写了这个小程序，漏洞很多，实现上写的也有些渣渣，欢迎大家阅读指点。


下载地址：https://github.com/yinxin630/gochat


简单思路描述：

0、服务端监听客户端请求，完成会话转发的任务

1、服务端采用心跳包维护用户在线状态

2、客户端通知服务端自己的监听地址，创建服务端-客户端信息通道


服务端：

package main

import (
	"fmt"
	"net"
	"os"
	"strconv"
	"time"
)
//用户信息
type User struct {
	 userName string
	 userAddr *net.UDPAddr
	 userListenConn *net.UDPConn
	 chatToConn *net.UDPConn
}

//服务器监听端口
const LISTENPORT = 1616
//缓冲区
const BUFFSIZE = 1024
var buff = make([]byte,BUFFSIZE)
//在线用户
var onlineUser = make([]User,0)
//在线状态判断缓冲区
var onlineCheckAddr = make([]*net.UDPAddr,0)

//错误处理
func HandleError(err error) {
	if err != nil {
		fmt.Println(err.Error())
		os.Exit(2)
	}
}
//消息处理
func HandleMessage(udpListener *net.UDPConn) {
	n,addr,err := udpListener.ReadFromUDP(buff)
	HandleError(err)

	if n > 0 {
		msg := AnalyzeMessage(buff,n)
		
		switch msg[0] {
			//连接信息
			case "connect  ":
				//获取昵称+端口
				userName := msg[1]
				userListenPort := msg[2]
				//获取用户ip
				ip := AnalyzeMessage([]byte(addr.String()),len(addr.String()))
				//显示登录信息
				fmt.Println(" 昵称:",userName," 地址:",ip[0]," 用户监听端口:",userListenPort," 登录成功！")
				//创建对用户的连接，用于消息转发
				userAddr,err := net.ResolveUDPAddr("udp4",ip[0] + ":" + userListenPort)
				HandleError(err)
				
				userConn,err := net.DialUDP("udp4",nil,userAddr)
				HandleError(err)
				
				//因为连接要持续使用，不能在这里关闭连接
				//defer userConn.Close()
				//添加到在线用户
				onlineUser = append(onlineUser,User{userName,userConn,nil})
				
			case "online   ":
				//收到心跳包
				onlineCheckAddr = append(onlineCheckAddr,addr)
				
			case "outline  ":
				//退出消息，未实现
			case "chat     ":
				//会话请求
				//寻找请求对象
				index := -1
				for i := 0; i < len(onlineUser); i++ {
					if onlineUser[i].userName == msg[1] {
						index = i
					}
				}
				//将所请求对象的连接添加到请求者中
				if index != -1 {
					nowUser,_ := FindUser(addr)
					onlineUser[nowUser].chatToConn = onlineUser[index].userListenConn
				}
			case "get      ":
				//向请求者返回在线用户信息
				index,_ := FindUser(addr)
				onlineUser[index].userListenConn.Write([]byte("当前共有" + strconv.Itoa(len(onlineUser)) + "位用户在线"))
				for i,v := range onlineUser {
					onlineUser[index].userListenConn.Write([]byte("" + strconv.Itoa(i + 1) + ":" + v.userName))
				}
			default:
				//消息转发
				//获取当前用户
				index,_ := FindUser(addr)
				//获取时间
				nowTime := time.Now()
				nowHour := strconv.Itoa(nowTime.Hour())
				nowMinute := strconv.Itoa(nowTime.Minute())
				nowSecond := strconv.Itoa(nowTime.Second())
				//请求会话对象是否存在
				if onlineUser[index].chatToConn == nil {
					onlineUser[index].userListenConn.Write([]byte("对方不在线"))
				} else {
					onlineUser[index].chatToConn.Write([]byte(onlineUser[index].userName + " " + nowHour + ":" + nowMinute + ":" + nowSecond + "\n" + msg[0]))
				}
				
		}
	}
}
//消息解析，[]byte -> []string
func AnalyzeMessage(buff []byte,len int) ([]string) {
	analMsg := make([]string,0)
	strNow := ""
	for i := 0; i < len; i++ {
		if string(buff[i:i + 1]) == ":" {
			analMsg = append(analMsg,strNow)
			strNow = ""
		} else {
			strNow += string(buff[i:i + 1])
		}
	}
	analMsg = append(analMsg,strNow)
	return analMsg
}
//寻找用户，返回（位置，是否存在）
func FindUser(addr *net.UDPAddr) (int,bool) {
	alreadyhave := false
	index := -1
	for i := 0; i < len(onlineUser); i++ {
		
		if onlineUser[i].userAddr.String() == addr.String() {
			alreadyhave = true
			index = i
			break
		}
	}
	return index,alreadyhave
}
//处理用户在线信息（暂时仅作删除用户使用）
func HandleOnlineMessage(addr *net.UDPAddr,state bool) {
	index,alreadyhave := FindUser(addr)
	if state == false {
		if alreadyhave {
			onlineUser = append(onlineUser[:index],onlineUser[index + 1:len(onlineUser)]...) 
		}
	}
}
//在线判断，心跳包处理，每5s查看一次所有已在线用户状态
func OnlineCheck() {
	for {
		onlineCheckAddr = make([]*net.UDPAddr,0)
		sleepTimer := time.NewTimer(time.Second * 5)
		<- sleepTimer.C
		for i := 0; i < len(onlineUser); i++ {
			haved := false
			FORIN:for j := 0; j < len(onlineCheckAddr); j++ {
				if onlineUser[i].userAddr.String() == onlineCheckAddr[j].String() {
					haved = true
					break FORIN
				}
			}
			if !haved {
				fmt.Println(onlineUser[i].userAddr.String() + "退出！")
				HandleOnlineMessage(onlineUser[i].userAddr,false)
				i--
			}

		}
	}
}

func main() {
	//监听地址
	udpAddr,"127.0.0.1:" + strconv.Itoa(LISTENPORT))
	HandleError(err)
	//监听连接
	udpListener,err := net.ListenUDP("udp4",udpAddr)
	HandleError(err)

	defer udpListener.Close()

	fmt.Println("开始监听：")

	//在线状态判断
	go OnlineCheck()

	for {
		//消息处理
		HandleMessage(udpListener)
	}

}

客户端： 
package main

import (
	"fmt"
	"os"
	"strconv"
	"net"
	"bufio"
	"math/rand"
	"time"
)

//数据包头，标识数据内容
var reflectString = map[string]string {
	"连接": 		"connect  :","在线": 		"online   :","聊天": 		"chat     :","在线用户": 	"get      :",}

//服务器端口
const CLIENTPORT = 1616
//数据缓冲区
const BUFFSIZE = 1024
var buff = make([]byte,BUFFSIZE)

//错误消息处理
func HandleError(err error) {
	if err != nil {
		fmt.Println(err.Error())
		os.Exit(2)
	}
}
//发送消息
func SendMessage(udpConn *net.UDPConn) {
	scaner := bufio.NewScanner(os.Stdin)

	for scaner.Scan() {
		if scaner.Text() == "exit" {
			return
		}
		udpConn.Write([]byte(scaner.Text()))
	}
}
//接收消息
func HandleMessage(udpListener *net.UDPConn) {
	for {
		n,_,err := udpListener.ReadFromUDP(buff)
		HandleError(err)

		if n > 0 {
			fmt.Println(string(buff[:n]))
		}
	}
}
/*
func AnalyzeMessage(buff []byte,len int) ([]string) {
    analMsg := make([]string,0)
    strNow := ""
    for i := 0; i < len; i++ {
        if string(buff[i:i + 1]) == ":" {
            analMsg = append(analMsg,strNow)
            strNow = ""
        } else {
            strNow += string(buff[i:i + 1])
        }
    }
    analMsg = append(analMsg,strNow)
    return analMsg
}*/
//发送心跳包
func SendOnlineMessage(udpConn *net.UDPConn) {
	for {
		//每间隔1s向服务器发送一次在线信息
		udpConn.Write([]byte(reflectString["在线"]))
		sleepTimer := time.NewTimer(time.Second)
		<- sleepTimer.C
	}
}

func main() {
	//判断命令行参数，参数应该为服务器ip
	if len(os.Args) != 2 {
		fmt.Println("程序命令行参数错误！")
		os.Exit(2)
	}
	//获取ip
	host := os.Args[1]

	//udp地址
	udpAddr,host + ":" + strconv.Itoa(CLIENTPORT))
	HandleError(err)

	//udp连接
	udpConn,udpAddr)
	HandleError(err)

	//本地监听端口
	newSeed := rand.NewSource(int64(time.Now().Second()))
	newRand := rand.New(newSeed)
	randPort := newRand.Intn(30000) + 10000

	//本地监听udp地址
	udpLocalAddr,"127.0.0.1:" + strconv.Itoa(randPort))
	HandleError(err)

	//本地监听udp连接
	udpListener,udpLocalAddr)
	HandleError(err)

	//fmt.Println("监听",randPort,"端口")

	//用户昵称
	userName := ""
	fmt.Printf("请输入昵称：")
	fmt.Scanf("%s",&userName)
	
	//向服务器发送连接信息（昵称+本地监听端口）
	udpConn.Write([]byte(reflectString["连接"] + userName + ":" + strconv.Itoa(randPort)))

	//关闭端口
	defer udpConn.Close()
	defer udpListener.Close()

	//发送心跳包
	go SendOnlineMessage(udpConn)
	//接收消息
	go HandleMessage(udpListener)

	command := ""
	
	for {
		//获取命令
		fmt.Scanf("%s",&command)
		switch command {
			case "chat" :
				people := ""
				//fmt.Printf("请输入对方昵称：")
				fmt.Scanf("%s",&people)
				//向服务器发送聊天对象信息
				udpConn.Write([]byte(reflectString["聊天"] + people))
				//进入会话
				SendMessage(udpConn)
				//退出会话
				fmt.Println("退出与" + people + "的会话")
			case "get" :
				//请求在线情况信息
				udpConn.Write([]byte(reflectString["在线用户"]))
		}
	}
}


运行环境：本地多终端模拟运行

运行图：

服务端：


客户端：

用户一：


用户二：


用户三：

原文链接：https://www.f2er.com/go/191027.html

猜你在找的Go相关文章

golang简单实现jwt验证(beego、xorm、jwt)
程序目录结构 简单实现，用户登录后返回一个jwt的token，下次请求带上token请求用户信息接...
【工程实践】利用go语言写一个简单的Pow样例
本篇博客的主要内容是用go写一个简单的Proof-of-Work共识机制，不涉及到网络通信环节，只是...
【工程实践】go语言实现MerkleTree
简介 默克尔树（MerkleTree）是一种典型的二叉树结构，其主要特点为： 最下面的叶节点包含...
第九章 goroutine
接下来学习并发编程, 并发编程是go语言最有特色的地方, go对并发编程是原生支持. goroutin...
第十一章 运用广度优先搜索走迷宫
先普及一下, 什么是广度优先搜索 广度优先搜索类似于树的层次遍历。从图中的某一顶点出发，...
第十三章 go实现分布式网络爬虫---单机版爬虫
爬虫的分类 网络爬虫分为两类 1. 通用爬虫: 类似于baidu, google. 他们会把大量的数据挖下...
第十章 channel select 总结
感觉channel在理解起来还有点费劲的, 尤其是select的使用, 既要可以读数据, 又要可以写数据...
第十五章 并发版爬虫第二版 -- 完结
并发版爬虫, 在上一篇单机版爬虫的基础上演变而来 这里只有并发引擎的代码, 基本的解析器代...
第八章--性能优化--pprof详细研究
接着上次的博客:&#160;https://www.cnblogs.com/ITPower/articles/12315637.html ppro...
第八章 测试与性能调优--生成文档和测试示例代码
java也可以生成文档, 就像一个手册一样, 可以用来查看方法, 接口, 下面我们给go项目生成文...
PHPJavaJava SEPythonC#C&C++RubyVBasp.NetGoPerlnettyDjangoDelphiJsp.NET CoreSpringFlaskSpringbootSpringMVCLuaLaravelMybatisAspGroovyThinkPHPYiiswoole
文件时间pythonm相等性PHP Warning时间问题问题解决pcntl_signal采样点wav模块动态文本调用频率限制对外暴露多个访问请求更新数据表模型结构type()方法比较速度手写体sobel算子保存模型Image类nn.Conv2dpytorch1.0kaggleDCGAN交并比range()用法打印模型反卷积卷积