From 5624862d0df6ded2c961cec085e2c9aa28b45def Mon Sep 17 00:00:00 2001
From: GiangTran <secmask@users.noreply.github.com>
Date: Wed, 28 Feb 2024 15:37:05 +0700
Subject: [PATCH] add background sync worker (#24)

* add background sync worker
---
 pkg/bsync/bsync.go             | 65 ++++++++++++++++++++++++++++++++++
 pkg/bsync/bsync_worker_test.go | 23 ++++++++++++
 2 files changed, 88 insertions(+)
 create mode 100644 pkg/bsync/bsync.go
 create mode 100644 pkg/bsync/bsync_worker_test.go

diff --git a/pkg/bsync/bsync.go b/pkg/bsync/bsync.go
new file mode 100644
index 0000000..17a4de1
--- /dev/null
+++ b/pkg/bsync/bsync.go
@@ -0,0 +1,65 @@
+package bsync
+
+import (
+	"sync"
+	"time"
+)
+
+type LocalReader[T any] interface {
+	Get() (T, time.Time)
+}
+
+// Worker support for background sync local state.
+type Worker[T any] struct {
+	interval   time.Duration
+	stop       chan struct{}
+	fetchFunc  func() (T, error)
+	localValue T
+	localTime  time.Time
+	lock       sync.Mutex
+}
+
+func New[T any](updateInternal time.Duration, fetchFunc func() (T, error)) *Worker[T] {
+	return &Worker[T]{
+		interval:  updateInternal,
+		fetchFunc: fetchFunc,
+		stop:      make(chan struct{}, 1),
+	}
+}
+
+func (w *Worker[T]) setValue(v T) {
+	w.lock.Lock()
+	defer w.lock.Unlock()
+	w.localValue = v
+	w.localTime = time.Now()
+}
+
+func (w *Worker[T]) Get() (T, time.Time) {
+	w.lock.Lock()
+	defer w.lock.Unlock()
+	return w.localValue, w.localTime
+}
+
+func (w *Worker[T]) Start() {
+	ticker := time.NewTicker(w.interval)
+	defer ticker.Stop()
+	if v, err := w.fetchFunc(); err == nil {
+		w.setValue(v)
+	}
+	for {
+		select {
+		case <-w.stop:
+			break
+		case <-ticker.C:
+			v, err := w.fetchFunc()
+			if err != nil { // user should handle error themself.
+				continue
+			}
+			w.setValue(v)
+		}
+	}
+}
+
+func (w *Worker[T]) Stop() {
+	w.stop <- struct{}{}
+}
diff --git a/pkg/bsync/bsync_worker_test.go b/pkg/bsync/bsync_worker_test.go
new file mode 100644
index 0000000..8a1117c
--- /dev/null
+++ b/pkg/bsync/bsync_worker_test.go
@@ -0,0 +1,23 @@
+package bsync_test
+
+import (
+	"testing"
+	"time"
+
+	"github.com/KyberNetwork/tradinglib/pkg/bsync"
+)
+
+func TestWorker(t *testing.T) {
+	v := 0
+	w := bsync.New(time.Millisecond, func() (int, error) {
+		v++
+		return v, nil
+	})
+	go w.Start()
+	for i := 0; i < 10; i++ {
+		c, at := w.Get()
+		t.Log(c, at)
+		time.Sleep(time.Millisecond)
+	}
+	w.Stop()
+}