61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package builder
|
|
|
|
import (
|
|
"openreplay/backend/internal/handlers"
|
|
. "openreplay/backend/pkg/messages"
|
|
)
|
|
|
|
type builderMap struct {
|
|
handlers []handlers.MessageProcessor
|
|
sessions map[uint64]*builder
|
|
}
|
|
|
|
func NewBuilderMap(handlers ...handlers.MessageProcessor) *builderMap {
|
|
return &builderMap{
|
|
handlers: handlers,
|
|
sessions: make(map[uint64]*builder),
|
|
}
|
|
}
|
|
|
|
func (m *builderMap) GetBuilder(sessionID uint64) *builder {
|
|
b := m.sessions[sessionID]
|
|
if b == nil {
|
|
b = NewBuilder(m.handlers...)
|
|
m.sessions[sessionID] = b
|
|
}
|
|
return b
|
|
}
|
|
|
|
func (m *builderMap) HandleMessage(sessionID uint64, msg Message, messageID uint64) {
|
|
b := m.GetBuilder(sessionID)
|
|
b.handleMessage(msg, messageID)
|
|
}
|
|
|
|
func (m *builderMap) IterateReadyMessages(iter func(sessionID uint64, msg Message)) {
|
|
for sessionID, b := range m.sessions {
|
|
if b.ended {
|
|
for _, p := range b.processors {
|
|
if rm := p.Build(); rm != nil {
|
|
b.readyMsgs = append(b.readyMsgs, rm)
|
|
}
|
|
}
|
|
}
|
|
b.iterateReadyMessage(func(msg Message) {
|
|
iter(sessionID, msg)
|
|
})
|
|
if b.ended {
|
|
delete(m.sessions, sessionID)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *builderMap) IterateSessionReadyMessages(sessionID uint64, iter func(msg Message)) {
|
|
session, ok := m.sessions[sessionID]
|
|
if !ok {
|
|
return
|
|
}
|
|
session.iterateReadyMessage(iter)
|
|
if session.ended {
|
|
delete(m.sessions, sessionID)
|
|
}
|
|
}
|