c6aac11cbf
* feat(backend): refactored heuristics service * feat(backend): refactored db service (moved several events to heuristics)
77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
package custom
|
|
|
|
import (
|
|
. "openreplay/backend/pkg/messages"
|
|
)
|
|
|
|
const InputEventTimeout = 1 * 60 * 1000
|
|
|
|
type inputLabels map[uint64]string
|
|
|
|
type inputEventBuilder struct {
|
|
inputEvent *InputEvent
|
|
inputLabels inputLabels
|
|
inputID uint64
|
|
}
|
|
|
|
func NewInputEventBuilder() *inputEventBuilder {
|
|
ieBuilder := &inputEventBuilder{}
|
|
ieBuilder.clearLabels()
|
|
return ieBuilder
|
|
}
|
|
|
|
func (b *inputEventBuilder) clearLabels() {
|
|
b.inputLabels = make(inputLabels)
|
|
}
|
|
|
|
func (b *inputEventBuilder) Handle(message Message, timestamp uint64) Message {
|
|
var inputEvent Message = nil
|
|
switch msg := message.(type) {
|
|
case *SetInputTarget:
|
|
if b.inputID != msg.ID {
|
|
inputEvent = b.Build()
|
|
b.inputID = msg.ID
|
|
}
|
|
b.inputLabels[msg.ID] = msg.Label
|
|
return inputEvent
|
|
case *SetInputValue:
|
|
if b.inputID != msg.ID {
|
|
inputEvent = b.Build()
|
|
b.inputID = msg.ID
|
|
}
|
|
if b.inputEvent == nil {
|
|
b.inputEvent = &InputEvent{
|
|
MessageID: message.MsgID(),
|
|
Timestamp: timestamp,
|
|
Value: msg.Value,
|
|
ValueMasked: msg.Mask > 0,
|
|
}
|
|
} else {
|
|
b.inputEvent.Value = msg.Value
|
|
b.inputEvent.ValueMasked = msg.Mask > 0
|
|
}
|
|
return inputEvent
|
|
case *CreateDocument:
|
|
inputEvent = b.Build()
|
|
b.clearLabels()
|
|
return inputEvent
|
|
case *MouseClick:
|
|
return b.Build()
|
|
}
|
|
|
|
if b.inputEvent != nil && b.inputEvent.Timestamp+InputEventTimeout < timestamp {
|
|
return b.Build()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b *inputEventBuilder) Build() Message {
|
|
if b.inputEvent == nil {
|
|
return nil
|
|
}
|
|
inputEvent := b.inputEvent
|
|
inputEvent.Label = b.inputLabels[b.inputID] // might be empty string
|
|
|
|
b.inputEvent = nil
|
|
return inputEvent
|
|
}
|