use an atomic counter.

This commit is contained in:
fiatjaf
2023-05-30 14:01:07 -03:00
parent 39f09e6bb9
commit 0f7a3f01f2
2 changed files with 5 additions and 5 deletions

View File

@@ -18,7 +18,7 @@ const (
PublishStatusSucceeded Status = 1 PublishStatusSucceeded Status = 1
) )
var subscriptionIdCounter = 0 var subscriptionIdCounter xsync.Counter
func (s Status) String() string { func (s Status) String() string {
switch s { switch s {
@@ -402,8 +402,8 @@ func (r *Relay) QuerySync(ctx context.Context, filter Filter) ([]*Event, error)
} }
func (r *Relay) PrepareSubscription(ctx context.Context) *Subscription { func (r *Relay) PrepareSubscription(ctx context.Context) *Subscription {
current := subscriptionIdCounter current := subscriptionIdCounter.Value()
subscriptionIdCounter++ subscriptionIdCounter.Inc()
ctx, cancel := context.WithCancel(ctx) ctx, cancel := context.WithCancel(ctx)

View File

@@ -9,7 +9,7 @@ import (
type Subscription struct { type Subscription struct {
label string label string
counter int counter int64
conn *Connection conn *Connection
mutex sync.Mutex mutex sync.Mutex
@@ -38,7 +38,7 @@ func (sub *Subscription) SetLabel(label string) {
// GetID return the Nostr subscription ID as given to the relay, it will be a sequential number, stringified. // GetID return the Nostr subscription ID as given to the relay, it will be a sequential number, stringified.
func (sub *Subscription) GetID() string { func (sub *Subscription) GetID() string {
return sub.label + ":" + strconv.Itoa(sub.counter) return sub.label + ":" + strconv.FormatInt(sub.counter, 10)
} }
// Unsub closes the subscription, sending "CLOSE" to relay as in NIP-01. // Unsub closes the subscription, sending "CLOSE" to relay as in NIP-01.