Merge fix drivers

thiagozs · Nov 25, 2021 · 0fec0b1 · 0fec0b1
2 parents ffb0c21 + d7b1e38
commit 0fec0b1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/v1/cache/drivers/drivers.go b/v1/cache/drivers/drivers.go
@@ -32,13 +32,18 @@ type DriverPort interface {
 }
 
 func NewDriver(driver Driver, opts ...Options) (DriverPort, error) {
+	var db DriverPort
+	var err error
 	switch driver {
 	case BUNTDB:
-		return buntdblayer.NewBuntDB(opts...), nil
+		db, err = buntdblayer.NewBuntDB(opts...)
 	case REDIS:
-		return redislayer.NewRedis(opts...), nil
+		db, err = redislayer.NewRedis(opts...)
 	}
-	return &Drivers{}, fmt.Errorf("unknown driver type: %s", driver.String())
+	if err != nil {
+		return nil, err
+	}
+	return &Drivers{db: db}, fmt.Errorf("unknown driver type: %s", driver.String())
 }
 
 func (d *Drivers) WriteKeyVal(key string, val string) error {