highperfocused/khatru
1
0
Fork 0
mirror of https://github.com/fiatjaf/khatru.git synced 2025-03-28 18:51:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix pre-search on policies.

Browse Source
This commit is contained in:
fiatjaf 2024-03-30 14:23:13 -03:00
parent 5efadf6256
commit 3214dac302
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
policies/events.go
View File

@ -14,6 +14,9 @@ import (
// If ignoreKinds is given this restriction will not apply to these kinds (useful for allowing a bigger). // If ignoreKinds is given this restriction will not apply to these kinds (useful for allowing a bigger).
// If onlyKinds is given then all other kinds will be ignored. // If onlyKinds is given then all other kinds will be ignored.
func PreventTooManyIndexableTags(max int, ignoreKinds []int, onlyKinds []int) func(context.Context, *nostr.Event) (bool, string) { func PreventTooManyIndexableTags(max int, ignoreKinds []int, onlyKinds []int) func(context.Context, *nostr.Event) (bool, string) {
slices.Sort(ignoreKinds)
slices.Sort(onlyKinds)
ignore := func(kind int) bool { return false } ignore := func(kind int) bool { return false }
if len(ignoreKinds) > 0 { if len(ignoreKinds) > 0 {
ignore = func(kind int) bool { ignore = func(kind int) bool {
@ -74,6 +77,9 @@ func RestrictToSpecifiedKinds(kinds ...uint16) func(context.Context, *nostr.Even
} }
} }
// sort the kinds in increasing order
slices.Sort(kinds)
return func(ctx context.Context, event *nostr.Event) (reject bool, msg string) { return func(ctx context.Context, event *nostr.Event) (reject bool, msg string) {
// these are cheap and very questionable optimizations, but they exist for a reason: // these are cheap and very questionable optimizations, but they exist for a reason:
// we would have to ensure that the kind number is within the bounds of a uint16 anyway // we would have to ensure that the kind number is within the bounds of a uint16 anyway
@ -84,9 +90,6 @@ func RestrictToSpecifiedKinds(kinds ...uint16) func(context.Context, *nostr.Even
return true, fmt.Sprintf("event kind not allowed (it should be higher than %d)", min) return true, fmt.Sprintf("event kind not allowed (it should be higher than %d)", min)
} }
// Sort the kinds in increasing order
slices.Sort(kinds)
// hopefully this map of uint16s is very fast // hopefully this map of uint16s is very fast
if _, allowed := slices.BinarySearch(kinds, uint16(event.Kind)); allowed { if _, allowed := slices.BinarySearch(kinds, uint16(event.Kind)); allowed {
return false, "" return false, ""