最近被后台日志弄的很烦,看到有个项目简简单单,又能满足需要,顺便试下看看效果,做下记录。只是记录下一部分内容,就不全部读了,关于源码可以去https://github.com/xmge/seelog。
结构设计
// websocket客户端
type client struct {
id string
socket *websocket.Conn
send chan []byte
}
// 客户端管理
type clientManager struct {
clients map[*client]bool
broadcast chan []byte
register chan *client
unregister chan *client
}
WebSocket 是 HTML5 开始提供的一种在单个 TCP 连接上进行全双工通讯的协议。
WebSocket 使得客户端和服务器之间的数据交换变得更加简单,允许服务端主动向客户端推送数据。在 WebSocket API 中,浏览器和服务器只需要完成一次握手,两者之间就直接可以创建持久性的连接,并进行双向数据传输。
在 WebSocket API 中,浏览器和服务器只需要做一个握手的动作,然后,浏览器和服务器之间就形成了一条快速通道。两者之间就直接可以数据互相传送。
程序使用管道作为通信基础
- clients 用来保存当前全部的Websocket
- broadcast 作为广播使用的管道,当收到消息,向所有的clients中的websocket进行传输信息
- register 当新的链接建立,将client指针放入注册管道
- unregister 当链接断开,将断开的链接对象放入取消管道
- client结构体内的send管道,当broadcast收到,将信息发到每个client的send管道中
func (manager *clientManager) start() {
defer func() {
if err := recover(); err != nil {
log.Printf("[seelog] error:%+v", err)
}
}()
for {
select {
case conn := <-manager.register:
manager.clients[conn] = true
case conn := <-manager.unregister:
if _, ok := manager.clients[conn]; ok {
close(conn.send)
delete(manager.clients, conn)
}
case message := <-manager.broadcast:
for conn := range manager.clients {
select {
case conn.send <- message:
default:
close(conn.send)
delete(manager.clients, conn)
}
}
}
}
}
使用select-case进行管道的数据处理,外部加一个for循环保持轮询的状态。
func (c *client) write() {
defer func() {
manager.unregister <- c
c.socket.Close()
}()
for {
select {
case message, ok := <-c.send:
if !ok {
c.socket.WriteClose(1)
return
}
c.socket.Write(message)
}
}
}
这个是在每个websocket启动的时候使用,每个socket保持一个for循环,使用defer用于关闭操作,当for被打断(即关闭网页之类的操作),socket被关闭,则会插入到取消管道中,clients键值对会删除这个连接的信息。
监控文件的内容变化
通过os.Stat获取文件信息,返回值为fileInfo的接口
fileInfo, err = os.Stat(filePath)
func (f *File) Stat() (FileInfo, error)
type FileInfo interface {
Name() string // base name of the file 文件名
Size() int64 // length in bytes for regular files; system-dependent for others 文件大小(byte长度)
Mode() FileMode // file mode bits 文件模式(只读、只写之类)
ModTime() time.Time // modification time
IsDir() bool // abbreviation for Mode().IsDir() 是否是目录
Sys() interface{} // underlying data source (can return nil) 基础数据源(可以返回nil)
}
获取当前的文件的截止位置
offset := fileInfo.Size()
获取新的文件大小,然后根据文件大小和之前的区别,构建一个新的byte数组,大小为新的字节数减去旧的字节数
msg := make([]byte, newOffset-offset)
使用Open方法打开一个文件,Open方法是以只读的方式读取数据
file, err := os.Open(filePath)
func Open(name string) (*File, error)
可以将文件读取的起点设置到某个位置,在seelog中,将读取起点设置到文件末尾,当文件的大小发生变化,则文件从上个起点开始读取文件内容
_, err = file.Seek(offset, 0)
func (f *File) Seek(offset int64, whence int) (ret int64, err error)
whence 存在3个参数
0:文件头的绝对位置偏移offset的距离
1:文件的相对位置,即当前位置偏移offset的距离
2:文件末尾的绝对位置偏移offset的距离
这个特性当文件以O_APPEND的模式打开是没有效果的
msg是之前构造的字节数组,将新增的内容读取到字节数组中
_, err = file.Read(msg)
使用管道作为消息传输的方式,manager在这里是一个全局的manager,当管道收到消息,就打印处理
manager.broadcast <- msg
最后记得将文件关闭,否则下次打开会出错
file.Close()
有疑问加站长微信联系(非本文作者)