140 lines
3.7 KiB
Go
140 lines
3.7 KiB
Go
package ext
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/status-im/status-go/common"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
"github.com/status-im/status-go/logutils"
|
|
"github.com/status-im/status-go/services/ext/mailservers"
|
|
)
|
|
|
|
// EnvelopeState in local tracker
|
|
type EnvelopeState int
|
|
|
|
const (
|
|
// NotRegistered returned if asked hash wasn't registered in the tracker.
|
|
NotRegistered EnvelopeState = -1
|
|
// MailServerRequestSent is set when p2p request is sent to the mailserver
|
|
MailServerRequestSent
|
|
)
|
|
|
|
// MailRequestMonitor is responsible for monitoring history request to mailservers.
|
|
type MailRequestMonitor struct {
|
|
eventSub mailservers.EnvelopeEventSubscriber
|
|
handler EnvelopeEventsHandler
|
|
|
|
mu sync.Mutex
|
|
cache map[types.Hash]EnvelopeState
|
|
|
|
requestsRegistry *RequestsRegistry
|
|
|
|
wg sync.WaitGroup
|
|
quit chan struct{}
|
|
}
|
|
|
|
func NewMailRequestMonitor(eventSub mailservers.EnvelopeEventSubscriber, h EnvelopeEventsHandler, reg *RequestsRegistry) *MailRequestMonitor {
|
|
return &MailRequestMonitor{
|
|
eventSub: eventSub,
|
|
handler: h,
|
|
cache: make(map[types.Hash]EnvelopeState),
|
|
requestsRegistry: reg,
|
|
}
|
|
}
|
|
|
|
// Start processing events.
|
|
func (m *MailRequestMonitor) Start() {
|
|
m.quit = make(chan struct{})
|
|
m.wg.Add(1)
|
|
go func() {
|
|
defer common.LogOnPanic()
|
|
m.handleEnvelopeEvents()
|
|
m.wg.Done()
|
|
}()
|
|
}
|
|
|
|
// Stop process events.
|
|
func (m *MailRequestMonitor) Stop() {
|
|
close(m.quit)
|
|
m.wg.Wait()
|
|
}
|
|
|
|
func (m *MailRequestMonitor) GetState(hash types.Hash) EnvelopeState {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
state, exist := m.cache[hash]
|
|
if !exist {
|
|
return NotRegistered
|
|
}
|
|
return state
|
|
}
|
|
|
|
// handleEnvelopeEvents processes whisper envelope events
|
|
func (m *MailRequestMonitor) handleEnvelopeEvents() {
|
|
events := make(chan types.EnvelopeEvent, 100) // must be buffered to prevent blocking whisper
|
|
sub := m.eventSub.SubscribeEnvelopeEvents(events)
|
|
defer sub.Unsubscribe()
|
|
for {
|
|
select {
|
|
case <-m.quit:
|
|
return
|
|
case event := <-events:
|
|
m.handleEvent(event)
|
|
}
|
|
}
|
|
}
|
|
|
|
// handleEvent based on type of the event either triggers
|
|
// confirmation handler or removes hash from MailRequestMonitor
|
|
func (m *MailRequestMonitor) handleEvent(event types.EnvelopeEvent) {
|
|
handlers := map[types.EventType]func(types.EnvelopeEvent){
|
|
types.EventMailServerRequestSent: m.handleRequestSent,
|
|
types.EventMailServerRequestCompleted: m.handleEventMailServerRequestCompleted,
|
|
types.EventMailServerRequestExpired: m.handleEventMailServerRequestExpired,
|
|
}
|
|
|
|
if handler, ok := handlers[event.Event]; ok {
|
|
handler(event)
|
|
}
|
|
}
|
|
|
|
func (m *MailRequestMonitor) handleRequestSent(event types.EnvelopeEvent) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.cache[event.Hash] = MailServerRequestSent
|
|
}
|
|
|
|
func (m *MailRequestMonitor) handleEventMailServerRequestCompleted(event types.EnvelopeEvent) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.requestsRegistry.Unregister(event.Hash)
|
|
state, ok := m.cache[event.Hash]
|
|
if !ok || state != MailServerRequestSent {
|
|
return
|
|
}
|
|
logutils.ZapLogger().Debug("mailserver response received", zap.Stringer("hash", event.Hash))
|
|
delete(m.cache, event.Hash)
|
|
if m.handler != nil {
|
|
if resp, ok := event.Data.(*types.MailServerResponse); ok {
|
|
m.handler.MailServerRequestCompleted(event.Hash, resp.LastEnvelopeHash, resp.Cursor, resp.Error)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *MailRequestMonitor) handleEventMailServerRequestExpired(event types.EnvelopeEvent) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
m.requestsRegistry.Unregister(event.Hash)
|
|
state, ok := m.cache[event.Hash]
|
|
if !ok || state != MailServerRequestSent {
|
|
return
|
|
}
|
|
logutils.ZapLogger().Debug("mailserver response expired", zap.Stringer("hash", event.Hash))
|
|
delete(m.cache, event.Hash)
|
|
if m.handler != nil {
|
|
m.handler.MailServerRequestExpired(event.Hash)
|
|
}
|
|
}
|