刚来公司实习，每天都在划水，划水狗一只。自己写点东西，大家有意见可以直接留言。
功能：基于go语言和协程池，实现一个简单的聊天服务器。
服务器代码：
/*/////////////////////////////
date：2018-3-17
writer：fengzi
function:用go语言，基于一个可自增的协程池，实现多人聊天
功能。函数首先创建协程池，然后根据用户连接，创建
多余的协程加入协程池，如果长时间未使用多余的协程，
系统自动回收多余协程。
/////////////////////////////*/
import (
"fmt"
// "io"
"net"
)
var player map[net.Conn]int = map[net.Conn]int{} //存储用户连接
var message = make(chan net.Conn, 20) //用户连接的信道
var num int = 0 //连接的用户数量
var channelnum int = 0 //开辟的协程数量
var boolexit int = 0 //是否需要关闭多余协程
func main() {
socket, err := net.Listen("tcp", "localhost:8080")
if err != nil {
fmt.Println("net listen error:", err)
return
}
defer socket.Close()
for i := 0; i < 2; i++ {
channelnum++
go talkChannel()
}
for {
conn, err := socket.Accept()
if err != nil {
continue
}
message <- conn
//扩充协程池
if num >= 2 {
fmt.Println("get new channel", num)
channelnum++
go talkChannel()
}
//如果用户数量长时间小于现在的协程数量，那么关闭多余协程
if (channelnum >= 2) && (num < channelnum) {
boolexit++
} else {
boolexit = 0
}
fmt.Println("boolexit num:", boolexit)
//连续十次未用到多余协程，关闭多余协程
if boolexit > 10 {
message <- nil
boolexit = 0
}
//go talk(conn)
}
}
func talkChannel() {
for {
conn := <-message
//如果传递的是空，则关闭该协程
if conn == nil {
fmt.Println("close channel")
break
}
fmt.Println("channel get one")
num++
fmt.Println("talkchannel used:", num)
defer conn.Close()
player[conn] = 0
for {
buf := make([]byte, 128)
n, err := conn.Read(buf)
if err != nil {
delete(player, conn)
fmt.Println("one client close:", conn)
break
}
buf = buf[0:n]
sendall(conn, buf)
}
num--
}
}
func talk(conn net.Conn) {
defer conn.Close()
player[conn] = 0
for {
buf := make([]byte, 8192)
n, err := conn.Read(buf)
if err != nil {
delete(player, conn)
fmt.Println("one client close:", conn)
break
}
buf = buf[0:n]
fmt.Println(buf)
sendall(conn, buf)
}
}
//广播发送
func sendall(conn net.Conn, buf []byte) {
for tmp, _ := range player {
if tmp != conn {
tmp.Write(buf)
}
}
}
客户端：
import (
"bufio"
"fmt"
"net"
"os"
)
func main() {
socket, err := net.Dial("tcp", "localhost:8080")
if err != nil {
fmt.Println("net.dial error:", err)
return
}
defer socket.Close()
go get(socket)
for {
var buff string
tmp := bufio.NewReader(os.Stdin)
buff, _ = tmp.ReadString('\n')
buff = buff[0 : len(buff)-2]
//fmt.Println(buff)
socket.Write([]byte(buff))
}
}
func get(conn net.Conn) {
for {
buf := make([]byte, 1024)
n, err := conn.Read(buf)
if err != nil {
fmt.Println("socket link error:", err)
break
}
buf = buf[0:n]
fmt.Println("others send:", string(buf))
}
}