137 lines
2.9 KiB
Go
137 lines
2.9 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"mrvacommander/pkg/common"
|
|
"sync"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
DBmutex sync.Mutex
|
|
)
|
|
|
|
func (s *StorageContainer) NextID() int {
|
|
// TODO update via db
|
|
return 12345
|
|
}
|
|
|
|
func (s *StorageContainer) SaveQueryPack(tgz []byte, sessionID int) (storagePath string, error error) {
|
|
// TODO save and return path
|
|
return "todo:no-path-yet", nil
|
|
}
|
|
|
|
func (s *StorageContainer) FindAvailableDBs(analysisReposRequested []common.OwnerRepo) (notFoundRepos []common.OwnerRepo, analysisRepos *map[common.OwnerRepo]DBLocation) {
|
|
// TODO s.FindAvailableDBs() via postgres
|
|
analysisRepos = &map[common.OwnerRepo]DBLocation{}
|
|
notFoundRepos = []common.OwnerRepo{}
|
|
|
|
return notFoundRepos, analysisRepos
|
|
}
|
|
|
|
func NewQLDBStore(startingID int, v *Visibles) (*StorageContainer, error) {
|
|
// TODO drop the startingID
|
|
db, err := ConnectDB(DBSpec{
|
|
Host: "postgres",
|
|
Port: 5432,
|
|
User: "exampleuser",
|
|
Password: "examplepass",
|
|
DBname: "querypack_db",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := StorageContainer{RequestID: startingID, DB: db}
|
|
// TODO XX: set up qldb_db tables
|
|
if err := s.SetupTables(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err = s.loadState(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// TODO XX: set up qldb_db
|
|
s.modules = v
|
|
|
|
return &s, nil
|
|
}
|
|
|
|
func NewServerStore(startingID int, v *Visibles) (*StorageContainer, error) {
|
|
db, err := ConnectDB(DBSpec{
|
|
Host: "postgres",
|
|
Port: 5432,
|
|
User: "exampleuser",
|
|
Password: "examplepass",
|
|
DBname: "server_db",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := StorageContainer{RequestID: startingID, DB: db}
|
|
if err := s.SetupTables(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err = s.loadState(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// TODO XX: set up qldb_db
|
|
s.modules = v
|
|
|
|
return &s, nil
|
|
}
|
|
|
|
func ConnectDB(s DBSpec) (*gorm.DB, error) {
|
|
// Open the database connection
|
|
dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
|
|
s.Host, s.Port, s.User, s.Password, s.DBname)
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
slog.Error("Error connecting to the database", "err", err)
|
|
return nil, err
|
|
}
|
|
return db, nil
|
|
}
|
|
|
|
func (s *StorageContainer) SetupTables() error {
|
|
msg := "Failed to initialize database "
|
|
|
|
if err := s.DB.AutoMigrate(&DBInfo{}); err != nil {
|
|
slog.Error(msg, "table", "dbinfo")
|
|
return err
|
|
}
|
|
if err := s.DB.AutoMigrate(&DBJobs{}); err != nil {
|
|
slog.Error(msg, "table", "dbjobs")
|
|
return err
|
|
}
|
|
if err := s.DB.AutoMigrate(&DBResult{}); err != nil {
|
|
slog.Error(msg, "table", "dbresult")
|
|
return err
|
|
}
|
|
if err := s.DB.AutoMigrate(&DBStatus{}); err != nil {
|
|
slog.Error(msg, "table", "dbstatus")
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *StorageContainer) loadState() error {
|
|
// XX:
|
|
// TODO load the state
|
|
return nil
|
|
}
|
|
|
|
func (s *StorageContainer) hasTables() bool {
|
|
// XX:
|
|
// TODO sql query to check for tables
|
|
return false
|
|
}
|