Harden LoadByIDs func

Signed-off-by: Bryan Frimin <bryan@probo.com>
This commit is contained in:
Bryan Frimin
2026-07-03 10:06:41 +02:00
parent a65d1096d6
commit 771edcc4fd
16 changed files with 64 additions and 16 deletions

View File

@@ -133,22 +133,6 @@ WHERE
return count, nil
}
func uniqueGIDs(values []gid.GID) []gid.GID {
set := make(map[gid.GID]struct{}, len(values))
unique := make([]gid.GID, 0, len(values))
for _, value := range values {
if _, ok := set[value]; ok {
continue
}
set[value] = struct{}{}
unique = append(unique, value)
}
return unique
}
func (f *Frameworks) LoadByOrganizationID(
ctx context.Context,
conn pg.Querier,
@@ -334,6 +318,10 @@ WHERE
*f = frameworks
if len(frameworks) != len(gid.NewSet(frameworkIDs...)) {
return ErrResourceNotFound
}
return nil
}