Go语言实践_实现一(服务器端)对多(客户端)在线聊天室
一、 目的
运用Go语言中的goroutine和通道实现一个简单的一个服务器端对多个客户端的在线聊天
二、设计思路
与一对一的设计思路类似,就是加了个线程的操作。
1,服务器端声明一个map,并打开监听端口;
2,客户端打开监听端口,同时连入服务器端;
3,在客户端上给自己起一个昵称,并输出,同时启动一个线程;
4,服务器端接收一个昵称,并存入map;
5,声明一个空的字符串,并写入要群发的消息;
6,服务器端解析发送的消息(msg_str[0]的值):
nick:使该客户端加入聊天室并广播连上服务器端的所有其他客户端;
say:广播客户端发出的消息;
quit:使该客户端退出,断开与服务器端的连接,并将退出消息广播给其他连上服务器端的所有其他客户端;
三、代码
server.go
// one sever to more client chat room
//This is chat sever
package main
import (
"fmt"
"net"
"strings"
)
var ConnMap map[string]net.Conn = make(map[string]net.Conn) //声明一个集合
//ConnMap := make(map[string]net.Conn)
func main() {
listen_socket, err := net.Listen("tcp", "127.0.0.1:8000") //打开监听接口
if err != nil {
fmt.Println("server start error")
}
defer listen_socket.Close()
fmt.Println("server is wating ....")
for {
conn, err := listen_socket.Accept() //收到来自客户端发来的消息
if err != nil {
fmt.Println("conn fail ...")
}
fmt.Println(conn.RemoteAddr(), "connect successed")
go handle(conn) //创建线程
}
}
func handle(conn net.Conn) {
for {
data := make([]byte, 255) //创建字节流 (此处同 一对一 通信)
msg_read, err := conn.Read(data) //声明并将从客户端读取的消息赋给msg_read 和err
if msg_read == 0 || err != nil {
continue
}
//解析协议
msg_str := strings.Split(string(data[0:msg_read]), "|") //将从客户端收到的字节流分段保存到msg_str这个数组中
switch msg_str[0] {
case "nick": //加入聊天室
fmt.Println(conn.RemoteAddr(), "-->", msg_str[1]) //nick占在数组下标0上,客户端上写的昵称占在数组下标1上
for k, v := range ConnMap { //遍历集合中存储的客户端消息
if k != msg_str[1] {
v.Write([]byte("[" + msg_str[1] + "]: join..."))
}
}
ConnMap[msg_str[1]] = conn
case "say": //转发消息
for k, v := range ConnMap { //k指客户端昵称 v指客户端连接服务器端后的地址
if k != msg_str[1] { //判断是不是给自己发,如果不是
fmt.Println("Send "+msg_str[2]+" to ", k) //服务器端将消息转发给集合中的每一个客户端
v.Write([]byte("[" + msg_str[1] + "]: " + msg_str[2])) //给除了自己的每一个客户端发送自己之前要发送的消息
}
}
case "quit": //退出
for k, v := range ConnMap { //遍历集合中的客户端昵称
if k != msg_str[1] { //如果昵称不是自己
v.Write([]byte("[" + msg_str[1] + "]: quit")) //给除了自己的其他客户端昵称发送退出的消息,并使Write方法阻塞
}
}
delete(ConnMap, msg_str[1]) //退出聊天室
}
}
}
client.go
// one sever to more client chat room
//This is chat client
package main
import (
"fmt"
"net"
)
var nick string = "" //声明聊天室的昵称
func main() {
conn, err := net.Dial("tcp", "127.0.0.1:8000") //打开监听端口
if err != nil {
fmt.Println("conn fail...")
}
defer conn.Close()
fmt.Println("client connect server successed \n")
//给自己取一个聊天室的昵称
fmt.Printf("Make a nickname:")
fmt.Scanf("%s", &nick) //输入昵称
fmt.Println("hello : ", nick) //客户端输出
conn.Write([]byte("nick|" + nick)) //将信息发送给服务器端
go Handle(conn) //创建线程
var msg string
for {
msg = "" //声明一个空的消息
fmt.Scan(&msg) //输入消息
conn.Write([]byte("say|" + nick + "|" + msg)) //三段字节流 say | 昵称 | 发送的消息
if msg == "quit" { //如果消息为quit
conn.Write([]byte("quit|" + nick)) //将quit字节流发送给服务器端
break //程序结束运行
}
}
}
func Handle(conn net.Conn) {
for {
data := make([]byte, 255) //创建一个字节流
msg_read, err := conn.Read(data) //将读取的字节流赋值给msg_read和err
if msg_read == 0 || err != nil { //如果字节流为0或者有错误
break
}
fmt.Println(string(data[0:msg_read])) //把字节流转换成字符串
}
}
编译运行服务端与客户端
go run server.go
go run client.go
运行结果展示
转载自Go语言实践_实现一(服务器端)对多(客户端)在线聊天室
作者:Zoctopus_Zhang
出处:http://www.cnblogs.com/OctoptusLian/