From 71e22d9f6e49ce0c3bc3b177323b17652a1c45a2 Mon Sep 17 00:00:00 2001
From: Michael Muré <batolettre@gmail.com>
Date: Thu, 11 Feb 2021 09:52:09 +0100
Subject: entity: clock loader

---
 entity/dag/clock.go | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 entity/dag/clock.go

(limited to 'entity/dag')

diff --git a/entity/dag/clock.go b/entity/dag/clock.go
new file mode 100644
index 00000000..fa944b33
--- /dev/null
+++ b/entity/dag/clock.go
@@ -0,0 +1,41 @@
+package dag
+
+import (
+	"fmt"
+
+	"github.com/MichaelMure/git-bug/identity"
+	"github.com/MichaelMure/git-bug/repository"
+)
+
+// ClockLoader is the repository.ClockLoader for Entity
+func ClockLoader(defs ...Definition) repository.ClockLoader {
+	clocks := make([]string, len(defs)*2)
+	for _, def := range defs {
+		clocks = append(clocks, fmt.Sprintf(creationClockPattern, def.namespace))
+		clocks = append(clocks, fmt.Sprintf(editClockPattern, def.namespace))
+	}
+
+	return repository.ClockLoader{
+		Clocks: clocks,
+		Witnesser: func(repo repository.ClockedRepo) error {
+			// We don't care about the actual identity so an IdentityStub will do
+			resolver := identity.NewStubResolver()
+
+			for _, def := range defs {
+				// override the resolver
+				def := def
+				def.identityResolver = resolver
+
+				// we actually just need to read all entities,
+				// as that will create and update the clocks
+				// TODO: concurrent loading to be faster?
+				for b := range ReadAll(def, repo) {
+					if b.Err != nil {
+						return b.Err
+					}
+				}
+			}
+			return nil
+		},
+	}
+}
-- 
cgit