feat: load source

entry/entry.go

@@ -15,6 +15,7 @@ type Entry struct {
 	localExp         time.Duration
 	data             []byte
 	ttl              float64
+	loadFrom int
 	codec            codec.Codec
 	enableLocalCache bool
 }
@@ -98,6 +99,18 @@ func (e *Entry) SetTTL(ttl float64) {
 	e.ttl = ttl
 }
 
+func (e *Entry) SourceLocal() bool {
+	return e.loadFrom == SourceLocal
+}
+
+func (e *Entry) SetSource(s int) {
+	e.loadFrom = s
+}
+
+func (e *Entry) SourceCenter() bool {
+	return e.loadFrom == SourceCenter
+}
+
 // CheckSet checks set conditions.
 func checkSet(e *Entry) error {
 	if int64(e.exp) == 0 {

entry/load_source.go

@@ -0,0 +1,7 @@
+package entry
+
+const (
+	_  int  = iota
+	SourceLocal
+	SourceCenter
+)

entry/setting.go

@@ -31,6 +31,14 @@ func WithLocalExpiration(d time.Duration) Setting {
 	}
 }
 
+// EnableLocalCache enables local cache.
+func EnableLocalCache() Setting {
+	return func(e *Entry) error {
+		e.enableLocalCache = true
+		return nil
+	}
+}
+
 // WithValue sets the corresponding value.
 func WithValue(value interface{}) Setting {
 	return func(e *Entry) error {

freesia.go

@@ -111,6 +111,7 @@ func (f *Freesia) Get(e *entry.Entry) error {
 			if err = e.Decode(b); ok && err != nil {
 				return err
 			}
+			e.SetSource(entry.SourceLocal)
 			return nil
 		default:
 			return err
@@ -123,6 +124,7 @@ func (f *Freesia) Get(e *entry.Entry) error {
 		if err != nil {
 			return redis.Nil
 		}
+		e.SetSource(entry.SourceCenter)
 		return nil
 	default:
 		return err
@@ -160,6 +162,7 @@ func (f *Freesia) GetWithTTL(e *entry.Entry) error {
 	default:
 		return err
 	}
+	return nil
 }
 
 func (f *Freesia) batchGet(es ...*entry.Entry) ([]*entry.Entry, error) {