@@ -23,6 +23,7 @@ import (
2323// Badger provider type.
2424type Badger struct {
2525 * badger.DB
26+
2627 stale time.Duration
2728 logger core.Logger
2829}
@@ -37,7 +38,7 @@ type badgerLogger struct {
3738}
3839
3940func (b * badgerLogger ) Warningf (msg string , params ... interface {}) {
40- b .SugaredLogger . Warnf (msg , params ... )
41+ b .Warnf (msg , params ... )
4142}
4243
4344// Factory function create new Badger instance.
@@ -48,6 +49,7 @@ func Factory(badgerConfiguration core.CacheProvider, logger core.Logger, stale t
4849
4950 if badgerConfiguration .Configuration != nil {
5051 var parsedBadger badger.Options
52+
5153 if b , e := json .Marshal (badgerConfiguration .Configuration ); e == nil {
5254 if e = json .Unmarshal (b , & parsedBadger ); e != nil {
5355 logger .Error ("Impossible to parse the configuration for the default provider (Badger)" , e )
@@ -112,7 +114,7 @@ func (provider *Badger) Uuid() string {
112114func (provider * Badger ) MapKeys (prefix string ) map [string ]string {
113115 keys := map [string ]string {}
114116
115- _ = provider .DB . View (func (txn * badger.Txn ) error {
117+ _ = provider .View (func (txn * badger.Txn ) error {
116118 opts := badger .DefaultIteratorOptions
117119 opts .PrefetchValues = false
118120 iterator := txn .NewIterator (opts )
@@ -139,7 +141,7 @@ func (provider *Badger) MapKeys(prefix string) map[string]string {
139141func (provider * Badger ) ListKeys () []string {
140142 keys := []string {}
141143
142- err := provider .DB . View (func (txn * badger.Txn ) error {
144+ err := provider .View (func (txn * badger.Txn ) error {
143145 opts := badger .DefaultIteratorOptions
144146 opts .PrefetchValues = false
145147 it := txn .NewIterator (opts )
@@ -174,7 +176,7 @@ func (provider *Badger) Get(key string) []byte {
174176
175177 var result []byte
176178
177- err := provider .DB . View (func (txn * badger.Txn ) error {
179+ err := provider .View (func (txn * badger.Txn ) error {
178180 i , err := txn .Get ([]byte (key ))
179181 item = i
180182
@@ -198,7 +200,7 @@ func (provider *Badger) Get(key string) []byte {
198200
199201// GetMultiLevel tries to load the key and check if one of linked keys is a fresh/stale candidate.
200202func (provider * Badger ) GetMultiLevel (key string , req * http.Request , validator * core.Revalidator ) (fresh * http.Response , stale * http.Response ) {
201- _ = provider .DB . View (func (tx * badger.Txn ) error {
203+ _ = provider .View (func (tx * badger.Txn ) error {
202204 result , err := tx .Get ([]byte (core .MappingKeyPrefix + key ))
203205 if err != nil && ! errors .Is (err , badger .ErrKeyNotFound ) {
204206 return err
@@ -226,7 +228,7 @@ func (provider *Badger) GetMultiLevel(key string, req *http.Request, validator *
226228func (provider * Badger ) SetMultiLevel (baseKey , variedKey string , value []byte , variedHeaders http.Header , etag string , duration time.Duration , realKey string ) error {
227229 now := time .Now ()
228230
229- err := provider .DB . Update (func (btx * badger.Txn ) error {
231+ err := provider .Update (func (btx * badger.Txn ) error {
230232 var err error
231233
232234 compressed := new (bytes.Buffer )
@@ -280,7 +282,7 @@ func (provider *Badger) SetMultiLevel(baseKey, variedKey string, value []byte, v
280282
281283// Set method will store the response in Badger provider.
282284func (provider * Badger ) Set (key string , value []byte , duration time.Duration ) error {
283- err := provider .DB . Update (func (txn * badger.Txn ) error {
285+ err := provider .Update (func (txn * badger.Txn ) error {
284286 return txn .SetEntry (badger .NewEntry ([]byte (key ), value ).WithTTL (duration ))
285287 })
286288 if err != nil {
@@ -292,20 +294,19 @@ func (provider *Badger) Set(key string, value []byte, duration time.Duration) er
292294
293295// Delete method will delete the response in Badger provider if exists corresponding to key param.
294296func (provider * Badger ) Delete (key string ) {
295- _ = provider .DB . Update (func (txn * badger.Txn ) error {
297+ _ = provider .Update (func (txn * badger.Txn ) error {
296298 return txn .Delete ([]byte (key ))
297299 })
298300}
299301
300302// DeleteMany method will delete the responses in Badger provider if exists corresponding to the regex key param.
301303func (provider * Badger ) DeleteMany (key string ) {
302304 rgKey , e := regexp .Compile (key )
303-
304305 if e != nil {
305306 return
306307 }
307308
308- _ = provider .DB . View (func (txn * badger.Txn ) error {
309+ _ = provider .View (func (txn * badger.Txn ) error {
309310 opts := badger .DefaultIteratorOptions
310311 opts .PrefetchValues = false
311312 it := txn .NewIterator (opts )
@@ -330,13 +331,11 @@ func (provider *Badger) Init() error {
330331
331332// Reset method will reset or close provider.
332333func (provider * Badger ) Reset () error {
333- err := provider .DB .DropAll ()
334- if err != nil {
334+ if err := provider .DropAll (); err != nil {
335335 provider .logger .Errorf ("Impossible to reset the Badger DB, %v" , err )
336336 }
337337
338- err = provider .DB .Close ()
339- if err != nil {
338+ if err := provider .Close (); err != nil {
340339 provider .logger .Errorf ("Impossible to close the Badger DB, %v" , err )
341340 }
342341
0 commit comments