102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package server
|
||
|
||
import (
|
||
"sync"
|
||
|
||
"go-common/app/service/main/broadcast/libs/bufio"
|
||
"go-common/app/service/main/broadcast/model"
|
||
)
|
||
|
||
// Channel used by message pusher send msg to write goroutine.
|
||
type Channel struct {
|
||
Room *Room
|
||
CliProto Ring
|
||
signal chan *model.Proto
|
||
Writer bufio.Writer
|
||
Reader bufio.Reader
|
||
Next *Channel
|
||
Prev *Channel
|
||
|
||
Mid int64
|
||
Key string
|
||
IP string
|
||
Platform string
|
||
watchOps map[int32]struct{}
|
||
mutex sync.RWMutex
|
||
V1 bool
|
||
}
|
||
|
||
// NewChannel new a channel.
|
||
func NewChannel(cli, svr int) *Channel {
|
||
c := new(Channel)
|
||
c.CliProto.Init(cli)
|
||
c.signal = make(chan *model.Proto, svr)
|
||
c.watchOps = make(map[int32]struct{})
|
||
return c
|
||
}
|
||
|
||
// Watch watch a operation.
|
||
func (c *Channel) Watch(accepts ...int32) {
|
||
c.mutex.Lock()
|
||
for _, op := range accepts {
|
||
if op >= model.MinBusinessOp && op <= model.MaxBusinessOp {
|
||
c.watchOps[op] = struct{}{}
|
||
}
|
||
}
|
||
c.mutex.Unlock()
|
||
}
|
||
|
||
// UnWatch unwatch an operation
|
||
func (c *Channel) UnWatch(accepts ...int32) {
|
||
c.mutex.Lock()
|
||
for _, op := range accepts {
|
||
delete(c.watchOps, op)
|
||
}
|
||
c.mutex.Unlock()
|
||
}
|
||
|
||
// NeedPush verify if in watch.
|
||
func (c *Channel) NeedPush(op int32, platform string) bool {
|
||
if c.Platform != platform && platform != "" {
|
||
return false
|
||
}
|
||
if op >= 0 && op < model.MinBusinessOp {
|
||
return true
|
||
}
|
||
c.mutex.RLock()
|
||
if _, ok := c.watchOps[op]; ok {
|
||
c.mutex.RUnlock()
|
||
return true
|
||
}
|
||
c.mutex.RUnlock()
|
||
return false
|
||
}
|
||
|
||
// Push server push message.
|
||
func (c *Channel) Push(p *model.Proto) (err error) {
|
||
// NOTE: 兼容v1弹幕推送，等播放器接后可以去掉
|
||
if c.V1 && p.Operation != 5 {
|
||
return
|
||
}
|
||
select {
|
||
case c.signal <- p:
|
||
default:
|
||
}
|
||
return
|
||
}
|
||
|
||
// Ready check the channel ready or close?
|
||
func (c *Channel) Ready() *model.Proto {
|
||
return <-c.signal
|
||
}
|
||
|
||
// Signal send signal to the channel, protocol ready.
|
||
func (c *Channel) Signal() {
|
||
c.signal <- model.ProtoReady
|
||
}
|
||
|
||
// Close close the channel.
|
||
func (c *Channel) Close() {
|
||
c.signal <- model.ProtoFinish
|
||
}
|