diff --git a/api/api.go b/api/api.go index e2fa361..059d5ee 100644 --- a/api/api.go +++ b/api/api.go @@ -61,7 +61,7 @@ func RunServer() error { } // Star the SW service - err := sw.StartSWService(sw.FILEDB) + err := sw.StartSWService() if err != nil { return err } diff --git a/sw/sw.go b/sw/sw.go index 8ba32b3..a8c0f5a 100644 --- a/sw/sw.go +++ b/sw/sw.go @@ -4,15 +4,14 @@ package sw import ( "errors" + "github.com/ESilva15/PTInvCertSWDB/config" models "github.com/ESilva15/PTInvCertSWDB/sw/models" repo "github.com/ESilva15/PTInvCertSWDB/sw/repo" ) -type RepoType int - const ( - FILEDB RepoType = iota - POSTGRES + FILEDB = "YAML" + POSTGRES = "PG" ) var service *SWService @@ -21,18 +20,20 @@ type SWService struct { SWRepo repo.SWRepo } -func StartSWService(rt RepoType) error { +func StartSWService() error { + cfg := config.GetInstance() + var err error var swRepo repo.SWRepo - switch rt { + switch cfg.DataBackend { case FILEDB: - swRepo, err = repo.NewYAMLRepo("./swdb.yaml") + swRepo, err = repo.NewYAMLRepo(cfg.YamlBackend.Path) if err != nil { return errors.New("Failed to create yaml repo: " + err.Error()) } default: - return errors.New("repo type not implemented yet") + return errors.New("chosen persistence backend not implemented yet") } service = &SWService{