并发聊天服务器

今天终于调通了一个并发版的聊天服务器，这个服务器主要是管理用户，通知用户上线或者下线，然后消息转发，广播到客户端，有点像聊天室那种，非常有趣。

我先用函数编程把程序调通了，然后又用面向对象的方法重写了一般，感觉业务逻辑更清晰一些。

// 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)
	}
}

欢迎不吝赐教。