diff --git a/store/datastore/store.go b/store/datastore/store.go index 4ff85e0a1..107b8092e 100644 --- a/store/datastore/store.go +++ b/store/datastore/store.go @@ -31,16 +31,30 @@ func Load(env envconfig.Env) store.Store { } func New(driver, config string) store.Store { - conn := Open(driver, config) + db := Open(driver, config) return store.New( driver, - &nodestore{conn}, - &userstore{conn}, - &repostore{conn}, - &keystore{conn}, - &buildstore{conn}, - &jobstore{conn}, - &logstore{conn}, + &nodestore{db}, + &userstore{db}, + &repostore{db}, + &keystore{db}, + &buildstore{db}, + &jobstore{db}, + &logstore{db}, + ) +} + +func From(db *sql.DB) store.Store { + var driver string + return store.New( + driver, + &nodestore{db}, + &userstore{db}, + &repostore{db}, + &keystore{db}, + &buildstore{db}, + &jobstore{db}, + &logstore{db}, ) }