goodmorningbitcoin/wot-relay
1
0
Fork 0
mirror of https://github.com/bitvora/wot-relay.git synced 2025-04-22 16:51:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #38 from bitvora/dev-profileRefresh

Browse Source 
broadcast events, process in goroutine
This commit is contained in:
Barry Deen 2024-09-12 20:37:02 -04:00 committed by GitHub
commit 47f1b9422b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
main.go
View File

@ -38,6 +38,8 @@ var booted bool
var oneHopNetwork []string
var trustNetworkMap map[string]bool
var pubkeyFollowerCount = make(map[string]int)
var trustedNotes uint64
var untrustedNotes uint64
func main() {
nostr.InfoLogger = log.New(io.Discard, "", 0)
@ -78,17 +80,16 @@ func main() {
relay.RejectEvent = append(relay.RejectEvent,
policies.RejectEventsWithBase64Media,
policies.EventIPRateLimiter(5, time.Minute*2, 30),
policies.EventIPRateLimiter(5, time.Minute*1, 30),
)
relay.RejectFilter = append(relay.RejectFilter,
policies.NoEmptyFilters,
policies.NoComplexFilters,
// policies.FilterIPRateLimiter(50, time.Minute, 250),
)
relay.RejectConnection = append(relay.RejectConnection,
policies.ConnectionRateLimiter(10, time.Minute*1, 30),
policies.ConnectionRateLimiter(10, time.Minute*2, 30),
)
relay.StoreEvent = append(relay.StoreEvent, db.SaveEvent)
@ -349,8 +350,6 @@ func archiveTrustedNotes(relay *khatru.Relay, ctx context.Context) {
}}
log.Println("📦 archiving trusted notes...")
var trustedNotes uint64
var untrustedNotes uint64
eventChan := pool.SubMany(ctx, seedRelays, filters)
@ -359,31 +358,43 @@ func archiveTrustedNotes(relay *khatru.Relay, ctx context.Context) {
case <-timeout:
log.Println("⏰ Archive process terminated due to timeout")
log.Println("📦 archived", trustedNotes, "trusted notes and discarded", untrustedNotes, "untrusted notes")
trustedNotes = 0
untrustedNotes = 0
return
case <-ctx.Done():
log.Println("⏰ Archive process terminated due to context cancellation")
log.Println("📦 archived", trustedNotes, "trusted notes and discarded", untrustedNotes, "untrusted notes")
trustedNotes = 0
untrustedNotes = 0
return
case ev, ok := <-eventChan:
if !ok {
log.Println("📦 subscription channel closed")
log.Println("📦 archived", trustedNotes, "trusted notes and discarded", untrustedNotes, "untrusted notes")
trustedNotes = 0
untrustedNotes = 0
return
}
if trustNetworkMap[ev.Event.PubKey] {
if len(ev.Event.Tags) > 3000 {
continue
}
relay.AddEvent(ctx, ev.Event)
trustedNotes++
//log.Println("📦 archived note: ", ev.Event.ID)
} else {
untrustedNotes++
}
go processEvent(ctx, ev.Event, relay)
}
}
}
func processEvent(ctx context.Context, ev *nostr.Event, relay *khatru.Relay) {
if trustNetworkMap[ev.PubKey] {
if len(ev.Tags) > 3000 {
return
}
relay.AddEvent(ctx, ev)
relay.BroadcastEvent(ev)
trustedNotes++
log.Println("📦 archived note: ", ev.ID)
} else {
log.Println("📦 discarded note: ", ev.ID)
untrustedNotes++
}
}