67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package lmdb
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"fiatjaf.com/nostr"
|
|
"fiatjaf.com/nostr/eventstore"
|
|
"fiatjaf.com/nostr/eventstore/codec/betterbinary"
|
|
"github.com/PowerDNS/lmdb-go/lmdb"
|
|
)
|
|
|
|
func (b *LMDBBackend) SaveEvent(evt nostr.Event) error {
|
|
return b.lmdbEnv.Update(func(txn *lmdb.Txn) error {
|
|
if b.EnableHLLCacheFor != nil {
|
|
// modify hyperloglog caches relative to this
|
|
useCache, skipSaving := b.EnableHLLCacheFor(evt.Kind)
|
|
|
|
if useCache {
|
|
err := b.updateHyperLogLogCachedValues(txn, evt)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to update hll cache: %w", err)
|
|
}
|
|
if skipSaving {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
// check if we already have this id
|
|
_, err := txn.Get(b.indexId, evt.ID[0:8])
|
|
if operr, ok := err.(*lmdb.OpError); ok && operr.Errno == lmdb.NotFound {
|
|
// we will only proceed if we get a NotFound
|
|
return b.save(txn, evt)
|
|
}
|
|
|
|
if err == nil {
|
|
return eventstore.ErrDupEvent
|
|
}
|
|
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (b *LMDBBackend) save(txn *lmdb.Txn, evt nostr.Event) error {
|
|
// encode to binary form so we'll save it
|
|
buf := make([]byte, betterbinary.Measure(evt))
|
|
if err := betterbinary.Marshal(evt, buf); err != nil {
|
|
return err
|
|
}
|
|
|
|
idx := b.serial(txn)
|
|
// raw event store
|
|
if err := txn.Put(b.rawEventStore, idx, buf, 0); err != nil {
|
|
return err
|
|
}
|
|
|
|
// put indexes
|
|
for k := range b.getIndexKeysForEvent(evt) {
|
|
err := txn.Put(k.dbi, k.key, idx, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|