其实从上学开始就一直想写一个im。 最近深入go,真是学会了太多,感觉人森虽然苦短,但是也不能只用python。很多知识是不用编译型语言无法了解的。
该来的还是会来,现在会一步一步用go把这个服务器完善起来 先从这个demo开始。
这个demo 我们要求所有连上服务器的用户都会知道有用户的离开,有用户的加入(除了第一个加入的用户),每个人说话就像聊天室一样,房间里的所有人都能看到。
由于接收tcp请求,get accept的conn步骤都差不多所以先上main部分的代码:
func main() {
listener, err := net.Listen("tcp", "0.0.0.0:8888")
if err != nil {
log.Fatal(err)
} go broadcaster() for{
conn, err := listener.Accept()
if err != nil {
fmt.Fprintf(os.Stdout, "you got something wrong %v", err)
continue
}
go handleConn(conn)
}
}
使用net包里面提供的Listen监听tcp来自8888端口的数据。并获得一个listener对象。
发起一个goroutine用于消息广播使用
然后进入监听循环,使用listener对象提供的Accept方法来获取连接。
每获得一个连接就重新启一个goroutine去handle这个链接。
main里面写的代码非常简单,其实服务器要做的事情总结一下无非就是获得listener对象,然后不停的获取获取链接上来的conn对象,然后把这些对象丢给处理链接函数去进行处理。在使用handleConn方法处理conn对象的时候,我们同样对不同的链接都启一个goroutine去并发处理每个conn这样则无需等待。
用于要给在线的所有用户发送消息,而不同的用户的conn对象都在不同的goroutine里面,我们很容易想到使用队列这种东西来做消息的传递,但是golang里面有channel来处理各不同goroutine之间的消息传递,所以在这个demo我选择使用channel在各不同的goroutine中传递广播消息。
先申明一些要用到的channel
type client chan<- string // send only channel var (
entering = make(chan client)
leaving = make(chan client)
messages = make(chan string)
)
这里要注意一点的是,重新定义了一个client类型,他是一个单向chennel,只能往里面写消息。
下面申请的entering和leaving都是client类型的channel。
什么意思呢?
就是说下面的entering和leaving都是装channel的channel。这里有点绕要注意,装channel的channel在<-的时候,会直接将channel对象装进去。
这里拓展开说说这个问题,以免下面的代码难以理解,来看一个例子:
package main import (
"fmt"
"time"
) type client chan string
var entering = make(chan client) func main() {
ch := make(chan string)
go func() {ch <- "那你很棒棒哦