Server.go
package main
import (
"fmt"
"io"
"net"
"os"
"strings"
)
func main() {
addr, err := net.ResolveTCPAddr("tcp", ":4040")
checkErr(err)
listen, err := net.ListenTCP("tcp", addr)
checkErr(err)
fmt.Println("Start server...")
for {
conn, err := listen.Accept()
checkErr(err)
go Handle(conn)// 每次建立一个连接就放到单独的线程内做处理
}
}
const BufLength = 128
var users map[string]net.Conn = make(map[string]net.Conn, 10)
func Handle(conn net.Conn) {
conn.Write([]byte("欢迎加入2B聊天组~"))
for {
data := make([]byte, 0)//此处做一个输入缓冲以免数据过长读取到不完整的数据
buf := make([]byte, BufLength)
for {
n, err := conn.Read(buf)
if err != nil && err != io.EOF {
checkErr(err)
}
data = append(data, buf[:n]...)
if n != BufLength {
break
}
}
cmd := strings.Split(string(data), "|")
fmt.Println("命令:", cmd)
switch cmd[0] {
case "nick":
fmt.Println("注册名称:" + cmd[1])
users[cmd[1]] = conn
case "say":
for k, v := range users {
if k != cmd[1] {
fmt.Println("给" + k + "发送消息:" + cmd[2])
v.Write([]byte(cmd[1] + ":[" + cmd[2] + "]"))
}
}
}
}
}
func checkErr(err error) {
if err != nil {
fmt.Println(err)
os.Exit(-1)
}
}
Client.go
package main
import (
"fmt"
"io"
"net"
"os"
)
var nick string = ""
func main() {
addr, err := net.ResolveTCPAddr("tcp", ":4040")
checkErr(err)
conn, err := net.DialTCP("tcp", nil, addr)
checkErr(err)
// 读取提示
data := make([]byte, 1024)
conn.Read(data)
fmt.Println(string(data))
// 输入昵称
fmt.Print("输入昵称:")
fmt.Scanf("%v", &nick)
fmt.Println("Hello " + nick)
conn.Write([]byte("nick|" + nick))
go Handle(conn)
for {
someTex := ""
fmt.Scanf("%v", &someTex)
conn.Write([]byte("say|" + nick + "|" + someTex))
}
}
const BufLength = 128
func Handle(conn net.Conn) {
for {
data := make([]byte, 1024)
buf := make([]byte, BufLength)
for {
n, err := conn.Read(buf)
if err != nil && err != io.EOF {
checkErr(err)
}
data = append(data, buf[:n]...)
if n != BufLength {
break
}
}
fmt.Println(string(data))
}
}
func checkErr(err error) {
if err != nil {
fmt.Println(err)
os.Exit(-1)
}
}
有疑问加站长微信联系(非本文作者)