You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package chat
|
|
|
|
import (
|
|
"fmt"
|
|
"git.noahlan.cn/northlan/ngs"
|
|
"git.noahlan.cn/northlan/ngs/component"
|
|
"git.noahlan.cn/northlan/ngs/examples/cluster/protocol"
|
|
"git.noahlan.cn/northlan/ngs/internal/log"
|
|
"git.noahlan.cn/northlan/ngs/session"
|
|
)
|
|
|
|
type RoomService struct {
|
|
component.Base
|
|
group *ngs.Group
|
|
}
|
|
|
|
func newRoomService() *RoomService {
|
|
return &RoomService{
|
|
group: ngs.NewGroup("all-users"),
|
|
}
|
|
}
|
|
|
|
func (rs *RoomService) JoinRoom(s *session.Session, msg *protocol.JoinRoomRequest) error {
|
|
if err := s.Bind(msg.MasterUid); err != nil {
|
|
return err
|
|
}
|
|
|
|
broadcast := &protocol.NewUserBroadcast{
|
|
Content: fmt.Sprintf("User user join: %v", msg.Nickname),
|
|
}
|
|
if err := rs.group.Broadcast("onNewUser", broadcast); err != nil {
|
|
return err
|
|
}
|
|
return rs.group.Add(s)
|
|
}
|
|
|
|
type SyncMessage struct {
|
|
Name string `json:"name"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
func (rs *RoomService) SyncMessage(s *session.Session, msg *SyncMessage) error {
|
|
// Send an RPC to master server to stats
|
|
if err := s.RPC("TopicService.Stats", &protocol.MasterStats{Uid: s.UID()}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Sync message to all members in this room
|
|
return rs.group.Broadcast("onMessage", msg)
|
|
}
|
|
|
|
func (rs *RoomService) userDisconnected(s *session.Session) {
|
|
if err := rs.group.Leave(s); err != nil {
|
|
log.Println("Remove user from group failed", s.UID(), err)
|
|
return
|
|
}
|
|
log.Println("User session disconnected", s.UID())
|
|
}
|