chore(backend-heuristics/db): remove redundant
This commit is contained in:
parent
62b36bd70a
commit
6ab6d342c0
5 changed files with 15 additions and 41 deletions
|
|
@ -60,7 +60,7 @@ func main() {
|
|||
}
|
||||
|
||||
// Handle heuristics and save to temporary queue in memory
|
||||
heurFinder.HandleMessage(session, msg)
|
||||
heurFinder.HandleMessage(sessionID, msg)
|
||||
|
||||
// Process saved heuristics messages as usual messages above in the code
|
||||
heurFinder.IterateSessionReadyMessages(sessionID, func(msg messages.Message) {
|
||||
|
|
|
|||
|
|
@ -1,43 +1,26 @@
|
|||
package heuristics
|
||||
|
||||
import (
|
||||
. "openreplay/backend/pkg/db/types"
|
||||
. "openreplay/backend/pkg/messages"
|
||||
)
|
||||
|
||||
type MessageHandler interface {
|
||||
HandleMessage(Message)
|
||||
}
|
||||
type ReadyMessagesIterator interface {
|
||||
IterateReadyMessages(func(Message))
|
||||
}
|
||||
|
||||
type Handler interface {
|
||||
MessageHandler
|
||||
ReadyMessagesIterator
|
||||
}
|
||||
|
||||
type mainHandler map[uint64]*sessHandler
|
||||
|
||||
func NewHandler() mainHandler {
|
||||
return make(mainHandler)
|
||||
}
|
||||
|
||||
func (m mainHandler) getSessHandler(session *Session) *sessHandler {
|
||||
if session == nil {
|
||||
//AAAA
|
||||
return nil
|
||||
}
|
||||
s := m[session.SessionID]
|
||||
func (m mainHandler) getSessHandler(sessionID uint64) *sessHandler {
|
||||
s := m[sessionID]
|
||||
if s == nil {
|
||||
s = newSessHandler(session)
|
||||
m[session.SessionID] = s
|
||||
s = newSessHandler()
|
||||
m[sessionID] = s
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func (m mainHandler) HandleMessage(session *Session, msg Message) {
|
||||
s := m.getSessHandler(session)
|
||||
func (m mainHandler) HandleMessage(sessionID uint64, msg Message) {
|
||||
s := m.getSessHandler(sessionID)
|
||||
s.HandleMessage(msg)
|
||||
}
|
||||
|
||||
|
|
@ -51,14 +34,3 @@ func (m mainHandler) IterateSessionReadyMessages(sessionID uint64, iter func(msg
|
|||
delete(m, sessionID)
|
||||
}
|
||||
}
|
||||
|
||||
func (m mainHandler) IterateReadyMessages(iter func(sessionID uint64, msg Message)) {
|
||||
for sessionID, s := range m {
|
||||
s.IterateReadyMessages(func(msg Message) {
|
||||
iter(sessionID, msg)
|
||||
})
|
||||
if s.IsEnded() {
|
||||
delete(m, sessionID)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
package builder
|
||||
package heuristics
|
||||
|
||||
import (
|
||||
. "openreplay/backend/pkg/messages"
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
package builder
|
||||
package heuristics
|
||||
|
||||
import (
|
||||
. "openreplay/backend/pkg/messages"
|
||||
|
|
|
|||
|
|
@ -1,19 +1,21 @@
|
|||
package heuristics
|
||||
|
||||
import (
|
||||
. "openreplay/backend/pkg/db/types"
|
||||
. "openreplay/backend/pkg/messages"
|
||||
)
|
||||
|
||||
type Handler interface {
|
||||
HandleMessage(Message)
|
||||
IterateReadyMessages(func(Message))
|
||||
}
|
||||
|
||||
type sessHandler struct {
|
||||
session *Session
|
||||
handlers []Handler
|
||||
ended bool
|
||||
}
|
||||
|
||||
func newSessHandler(session *Session) *sessHandler {
|
||||
func newSessHandler() *sessHandler {
|
||||
return &sessHandler{
|
||||
session: session,
|
||||
handlers: []Handler{
|
||||
new(clickrage),
|
||||
new(performanceAggregator),
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue