今天终于调通了一个并发版的聊天服务器,这个服务器主要是管理用户,通知用户上线或者下线,然后消息转发,广播到客户端,有点像聊天室那种,非常有趣。
我先用函数编程把程序调通了,然后又用面向对象的方法重写了一般,感觉业务逻辑更清晰一些。
// goserver.go
package main
import (
"fmt"
"net"
"strings"
)
type client struct {
C chan string
addr string
}
var onlineMap map[string]client
var message = make(chan string)
func handleconn(conn net.Conn) {
addr := conn.RemoteAddr().String()
cli := client{C: make(chan string), addr: addr}
onlineMap[addr] = cli
defer func() {
message <- addr + "outline\n"
delete(onlineMap, addr)
conn.Close()
}()
fmt.Println(addr, "login")
message <- addr + "login successful" + "\r\n"
buf := make([]byte, 1024)
go func() {
for ch := range cli.C {
conn.Write([]byte(ch + "\n"))
}
}()
for {
n, err := conn.Read(buf)
if err != nil {
return
}
str := string(buf[:n])
str = strings.ToUpper(str)
message <- str
}
}
func main() {
onlineMap = make(map[string]client)
lsn, err := net.Listen("tcp", "127.0.0.1:8000")
if err != nil {
panic(err)
}
defer lsn.Close()
fmt.Println("服务器启动了...")
go func() {
for {
msg := <-message
for _, cli := range onlineMap {
cli.C <- msg
}
}
}()
for {
conn, err1 := lsn.Accept()
if err1 != nil {
fmt.Println(err1)
return
}
go handleconn(conn)
}
}
下面是面向对象的方式封装了一下。
// chatserver.go
package main
import (
"fmt"
"net"
"strings"
)
var message = make(chan string)
type client struct {
C chan string
addr string
}
func createClient(conn net.Conn) client {
addr := conn.RemoteAddr().String()
return client{C: make(chan string), addr: addr}
}
func (cli client) writeToclient(conn net.Conn) {
for ch := range cli.C {
conn.Write([]byte(ch + "\n"))
}
}
type onlineMap map[string]client
func (om onlineMap) add(cli client, addr string) {
om[addr] = cli
message <- addr + "login" + "\n"
}
func (om onlineMap) remove(key, addr string) {
delete(om, key)
message <- addr + "logout" + "\n"
}
func (om onlineMap) broadcast(message chan string) {
for {
msg := <-message
for _, v := range om {
v.C <- msg
}
}
}
func handleconn(conn net.Conn, om onlineMap) {
addr := conn.RemoteAddr().String()
cli := createClient(conn)
om.add(cli, addr)
defer func() {
om.remove(addr, addr)
conn.Close()
}()
fmt.Println(addr, "login")
buf := make([]byte, 1024)
go cli.writeToclient(conn)
for {
n, err := conn.Read(buf)
if err != nil {
return
}
str := string(buf[:n])
str = strings.ToUpper(str)
message <- str
}
}
func main() {
om := make(onlineMap)
lsn, err := net.Listen("tcp", "127.0.0.1:8000")
if err != nil {
panic(err)
}
defer lsn.Close()
fmt.Println("服务器启动了...")
go om.broadcast(message)
for {
conn, err1 := lsn.Accept()
if err1 != nil {
fmt.Println(err1)
return
}
go handleconn(conn, om)
}
}
欢迎不吝赐教。