From ed10574deacf5a27be9572a94c1ae2e39e11ff6c Mon Sep 17 00:00:00 2001 From: Ophestra Date: Fri, 20 Dec 2024 19:05:39 +0900 Subject: state: store join util Signed-off-by: Ophestra --- internal/state/util.go | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 internal/state/util.go (limited to 'internal/state/util.go') diff --git a/internal/state/util.go b/internal/state/util.go new file mode 100644 index 00000000..473ae51d --- /dev/null +++ b/internal/state/util.go @@ -0,0 +1,49 @@ +package state + +import ( + "errors" + "maps" +) + +var ( + ErrDuplicate = errors.New("store contains duplicates") +) + +// Join returns joined state entries of all active aids. +func Join(s Store) (Entries, error) { + var ( + aids []int + entries = make(Entries) + + el int + res Entries + loadErr error + ) + + if ln, err := s.List(); err != nil { + return nil, err + } else { + aids = ln + } + + for _, aid := range aids { + if _, err := s.Do(aid, func(c Cursor) { + res, loadErr = c.Load() + }); err != nil { + return nil, err + } + + if loadErr != nil { + return nil, loadErr + } + + // save expected length + el = len(entries) + len(res) + maps.Copy(entries, res) + if len(entries) != el { + return nil, ErrDuplicate + } + } + + return entries, nil +} -- cgit v1.3.1