feat: improve driver

2022-06-07 22:02:41 +08:00
parent 0d93a6aa41
commit 677047c80b
9 changed files with 88 additions and 27 deletions
--- a/internal/driver/manage.go
+++ b/internal/driver/manage.go
@ -8,7 +8,7 @@ type New func() Driver

 var driversMap = map[string]New{}

-func RegisterDriver(name string, new New) {
-	log.Infof("register driver: [%s]", name)
-	driversMap[name] = new
+func RegisterDriver(config Config, driver New) {
+	log.Infof("register driver: [%s]", config.Name)
+	driversMap[config.Name] = driver
 }