使用go语言怎么实现一个聊天服务器
这篇文章将为大家详细讲解有关使用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()
}
}()
}关于使用go语言怎么实现一个聊天服务器就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
当前名称：使用go语言怎么实现一个聊天服务器
本文网址：http://azwzsj.com/article/pjeeeh.html
