Initial commit

service/cacheSer/writer.go

@@ -0,0 +1,43 @@
+package cacheSer
+
+import (
+	"sync"
+)
+
+type SyncFunc func(list []interface{}) error
+
+type BaseWriter interface {
+	Add(doc interface{})
+	Sync() error
+}
+
+type Writer struct {
+	lock sync.Mutex
+	list []interface{}
+	sync SyncFunc
+}
+
+func NewWriter(syncFunc SyncFunc) BaseWriter {
+	return &Writer{
+		lock: sync.Mutex{},
+		list: make([]interface{}, 0),
+		sync: syncFunc,
+	}
+}
+
+func (w *Writer) Add(doc interface{}) {
+	w.lock.Lock()
+	defer w.lock.Unlock()
+	w.list = append(w.list, doc)
+}
+
+func (w *Writer) Sync() error {
+	list := make([]interface{}, 0)
+	func() {
+		w.lock.Lock()
+		defer w.lock.Unlock()
+		list = w.list
+		w.list = make([]interface{}, 0)
+	}()
+	return w.sync(list)
+}