get method for atomic counters

drpcorg · Dec 16, 2024 · 016693d · 016693d
1 parent cde0087
commit 016693d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/atomic_counter.go b/atomic_counter.go
@@ -69,6 +69,17 @@ func (a *AtomicCounter) load() error {
 	return nil
 }
 
+func (a *AtomicCounter) Get(ctx context.Context) (int64, error) {
+	err := a.load()
+	if err != nil {
+		return 0, err
+	}
+	if !a.loaded.Load() {
+		return 0, ErrCounterNotLoaded
+	}
+	return a.localValue.Load(), nil
+}
+
 // Loads (if needed) and increments counter
 func (a *AtomicCounter) Increment(ctx context.Context, val int64) (int64, error) {
 	err := a.load()