From ea69d02af6e691a1b32e5198a1143aa142ba82ac Mon Sep 17 00:00:00 2001 From: Julio Capote Date: Fri, 6 Jan 2023 14:31:20 -0500 Subject: subscription lookup on outbox item create --- models/persister.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'models/persister.go') diff --git a/models/persister.go b/models/persister.go index f877488..b4b9c49 100644 --- a/models/persister.go +++ b/models/persister.go @@ -98,6 +98,23 @@ func (p *Persister) Collect(model model) ([][]byte, error) { return result, err } +// Much faster, "keys only" version of Collect() +func (p *Persister) CollectKeys(model model) ([][]byte, error) { + var keys [][]byte + err := p.db.View(func(txn *badger.Txn) error { + opts := badger.DefaultIteratorOptions + opts.PrefetchValues = false + it := txn.NewIterator(opts) + defer it.Close() + prefix := []byte(model.Keybase()) + for it.Seek(prefix); it.ValidForPrefix(prefix); it.Next() { + keys = append(keys, it.Item().Key()) + } + return nil + }) + return keys, err +} + func (p *Persister) Find(model model) ([]byte, error) { var result []byte var item *badger.Item -- cgit v1.2.3