Golang学习笔记之简易聊天系统服务器的搭建
下面先列举一下程序使用到的函数,省的大家去找,直接拷贝官方api的解释吧。
func DialTCP(net string, laddr, raddr *TCPAddr) (*TCPConn, error)
DialTCP在网络协议net上连接本地地址laddr和远端地址raddr。
net必须是"tcp"、"tcp4"、"tcp6";如果laddr不是nil,将使用它作为本地地址,否则自动选择一个本地地址。
func ResolveTCPAddr(net, addr string) (*TCPAddr, error)
ResolveTCPAddr将addr作为TCP地址解析并返回。
参数addr格式为"host:port"或"[ipv6-host%zone]:port",解析得到网络名和端口名;net必须是"tcp"、"tcp4"或"tcp6"。
func ListenTCP(net string, laddr *TCPAddr) (*TCPListener, error)
ListenTCP在本地TCP地址laddr上声明并返回一个*TCPListener,
net参数必须是"tcp"、"tcp4"、"tcp6",如果laddr的端口字段为0,函数将选择一个当前可用的端口,可以用Listener的Addr方法获得该端口。TCPListener代表一个TCP网络的监听者。使用者应尽量使用Listener接口而不是假设(网络连接为)TCP。
func (l *TCPListener) AcceptTCP() (*TCPConn, error)
//AcceptTCP接收下一个呼叫,并返回一个新的*TCPConn。
//TCPConn代表一个TCP网络连接,实现了Conn接口。
Write(b []byte) (n int, err error)
Write从连接中写入数据
Write方法可能会在超过某个固定时间限制后超时返回错误,该错误的Timeout()方法返回真
Read(b []byte) (n int, err error)
Read从连接中读取数据
Read方法可能会在超过某个固定时间限制后超时返回错误,该错误的Timeout()方法返回真
var Args []string
Args保管了命令行参数,第一个是程序名。
服务端代码
server.go
package main
import (
"encoding/json"
"fmt"
"log"
"net"
)
func main() {
//端口号
StartServer("8080")
}
//结构体
type person struct {
news string
ip string
}
//StartServer 启动服务器
func StartServer(s string) {
// 获取tcp地址
tcpAddr, err := net.ResolveTCPAddr("tcp4", ":"+s)
if err != nil {
log.Printf("resolve tcp addr failed: %v\n", err)
return
}
//连接池,用来保存所有人的数据
conns := make(map[string]net.Conn)
//消息信道,为有缓冲信道,当然缓冲内存也可以增大
messages := make(chan person, 10)
// 监听
listener, err := net.ListenTCP("tcp", tcpAddr)
if err != nil {
log.Printf("listen tcp port failed: %v\n", err)
return
}
//开启发送消息的协程
go BroadCastMessage(conns, messages)
//时刻监测有没有新的消息发送过来
for {
/*
func (l *TCPListener) AcceptTCP() (*TCPConn, error)
AcceptTCP接收下一个呼叫,并返回一个新的*TCPConn。
TCPConn代表一个TCP网络连接,实现了Conn接口。
*/
conn, err := listener.AcceptTCP()
if err != nil {
fmt.Println("链接失败")
continue
}
conns[conn.RemoteAddr().String()] = conn
go HandlerMessage(conn, conns, messages)
}
}
//HandlerMessage 检查发送来的消息
/*
//conn:返回的新的*TCPConn
conns:连接池
messages:消息通道
*/
func HandlerMessage(conn net.Conn, conns map[string]net.Conn, messages chan person) {
//切片用来暂存消息
buf := make([]byte, 1024)
for {
//从链接里面读取数据,写给buf
len, err := conn.Read(buf)
if err != nil {
conn.Close()
delete(conns, conn.RemoteAddr().String())
break
}
//消息写入结构体,并且把发送者的ip一块写入
p := person{
news: string(buf[:len]),
ip: conn.RemoteAddr().String(),
}
//发送给信道
messages <- p
fmt.Println(string(buf[:len]))
}
}
//BroadCastMessage 发送消息
/*
conns:连接池
messages:消息通道
*/
func BroadCastMessage(conns map[string]net.Conn, messages chan person) {
for {
//读取信道里面的消息
mess := <-messages
//发送给所有人
for k, v := range conns {
//不发送给自己
if k != mess.ip {
var m map[string]interface{}
//将发送过来消息的news序列化为map
json.Unmarshal([]byte(mess.news), &m)
//拼接字符串
msg := m["time"].(string) + "\n" + m["userID"].(string) + ":" + m["message"].(string)
//发送
_, err := v.Write([]byte(msg))
//如果失败关闭v所属人的协程,继续监测
if err != nil {
delete(conns, k)
v.Close()
continue
}
}
}
}
}
个人端代码
net.go
package main
import (
"encoding/json"
"fmt"
"log"
"net"
"os"
"time"
)
func main() {
/*
var Args []string
Args保管了命令行参数,第一个是程序名。
*/
tcpAddr, _ := net.ResolveTCPAddr("tcp", os.Args[1])
//拨号
conn, _ := net.DialTCP("tcp", nil, tcpAddr)
var p = make(map[string]interface{})
p["userID"] = os.Args[2]
go HandlerMessage(conn, p)
ReceivesMessage(conn)
}
//HandlerMessage 向服务器发送数据
func HandlerMessage(conn net.Conn, p map[string]interface{}) {
for {
var input string
// 接收输入消息,放到input变量中
fmt.Scanln(&input)
//用户端退出啊
if input == "/q" || input == "/quit" {
fmt.Println("Byebye ...")
conn.Close()
os.Exit(0)
}
//发送的信息
p["message"] = input
//时间戳
p["time"] = time.Now()
// 只处理有内容的消息
if len(input) > 0 {
//序列化为json
msg, err := json.Marshal(p)
if err != nil {
//如果不成功,返回错误
fmt.Println("错误", err)
} else {
//发送数据
_, err := conn.Write([]byte(string(msg)))
//没有发送成功
if err != nil {
conn.Close()
break
}
}
}
}
}
//ReceivesMessage 接收服务器消息
func ReceivesMessage(conn net.Conn) {
// 接收来自服务器端的广播消息
buf := make([]byte, 1024)
for {
length, err := conn.Read(buf)
if err != nil {
log.Printf("recv server msg failed: %v\n", err)
conn.Close()
os.Exit(0)
break
}
fmt.Println(string(buf[0:length]))
}
}