Hesham/sft
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

main: move db code to func

Browse Source 
Signed-off-by: HeshamTB <hishaminv@gmail.com>
This commit is contained in:
HeshamTB 2026-03-13 02:39:21 +03:00
parent f3215035bd
commit e3f12f9c57
Signed by: Hesham
GPG Key ID: 74876157D199B09E
1 changed files with 44 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
cmd/sft/sft.go
View File

@ -34,9 +34,6 @@ func main() {
rootLog.Info("init sft server")
// TODO: Init DB
// This creates the db file if it does not exist
db, err := sql.Open(DB_DRIVER, DB_PATH)
if err != nil {
@ -46,17 +43,25 @@ func main() {
defer db.Close()
db.SetMaxOpenConns(0)
err = initDb(db, *rootLog.With("sub", "initDb"))
if err != nil {
rootLog.Error("error in initDb", "err", err.Error())
os.Exit(EXIT_FAIL)
}
// sqlStmt := `
// create table if not exists _sft_db (db_rev integer not null, sft_ver text);
// delete from _sft_db;
// `
//
// _, err = db.Exec(sqlStmt)
// if err != nil {
// rootLog.Error("error in init db", "err", err.Error())
// os.Exit(EXIT_FAIL)
// }
// TODO: Create local admin users
// TODO: File transfer, quarintine, scan with local accounts
}
/*
initilize database with sft parameters and table from starting point
*/
func initDb(db *sql.DB, logger slog.Logger) error {
/*
- Case 1: Database file does not exist. Created with sql.Open()
@ -68,41 +73,40 @@ func main() {
It is critical to run in order. In this case we don't need the next column
*/
// TODO: apply rev 0
f, _ := migrations.DbMigrations.Open("0_init.sql")
defer f.Close()
initSql, err := io.ReadAll(f)
if err != nil {
rootLog.Error("can not read embedFS db rev0", "err", err.Error())
logger.Error("can not read embedFS db rev0", "err", err.Error())
os.Exit(EXIT_FAIL)
}
rootLog.Debug("migrations read", "bytes", len(initSql))
logger.Debug("migrations read rev0", "bytes", len(initSql))
_, err = db.Exec(string(initSql))
if err != nil {
if strings.Contains(err.Error(), "UNIQUE constraint failed") {
rootLog.Info("db is already initilized to rev0")
logger.Info("db is already initilized to rev0")
} else {
rootLog.Error("can not exec on db", "err", err.Error())
logger.Error("can not exec on db", "err", err.Error())
os.Exit(EXIT_FAIL)
}
}
rootLog.Info("db initilized")
logger.Info("db initilized")
rootLog.Debug("read on _sft_db")
logger.Debug("read on _sft_db")
rows, err := db.Query("select db_rev, sft_ver from _sft_db;")
if err != nil {
rootLog.Error("error in init db", "err", err.Error())
logger.Error("error in init db", "err", err.Error())
os.Exit(EXIT_FAIL)
}
if !rows.Next() {
rootLog.Warn("db not migrated to 1")
logger.Warn("db not migrated to 1")
}
rows.Close()
@ -111,7 +115,7 @@ func main() {
rows, err = db.Query("SELECT db_rev FROM _sft_db ORDER BY db_rev;")
if err != nil {
rootLog.Error("error", "err", err.Error())
logger.Error("error", "err", err.Error())
os.Exit(EXIT_FAIL)
}
@ -121,11 +125,11 @@ func main() {
var val int
if err := rows.Scan(&val); err != nil {
rootLog.Error("revision stored in db is not an integer")
logger.Error("revision stored in db is not an integer")
os.Exit(EXIT_FAIL)
}
rootLog.Debug("found applied database revision", "rev", val)
logger.Debug("found applied database revision", "rev", val)
appliedMigs = append(appliedMigs, val)
}
@ -134,7 +138,7 @@ func main() {
migrationFilesDir, err := migrations.DbMigrations.ReadDir(".")
if err != nil {
rootLog.Error("could not list files in migrations embedFS", "err", err.Error())
logger.Error("could not list files in migrations embedFS", "err", err.Error())
os.Exit(EXIT_FAIL)
}
@ -148,19 +152,19 @@ func main() {
for i, file := range migrationFilesDir {
if file.IsDir() {
rootLog.Debug("migration dir. skipping", "name", file.Name())
logger.Debug("migration dir. skipping", "name", file.Name())
continue
}
rootLog.Debug("migration file", "name", file.Name())
logger.Debug("migration file", "name", file.Name())
migStmts, err := migrations.DbMigrations.ReadFile(file.Name())
if err != nil {
rootLog.Error("could not read migration file", "err", err.Error(), "name", file.Name())
logger.Error("could not read migration file", "err", err.Error(), "name", file.Name())
os.Exit(EXIT_FAIL)
}
mRev, err := strconv.Atoi(strings.Split(file.Name(), "_")[0])
if err != nil {
rootLog.Error(
logger.Error(
"unexpected db revision. invalid file naming in migrations",
"err", err.Error(),
"name", file.Name())
@ -168,7 +172,7 @@ func main() {
}
if mRev != i {
rootLog.Error("unexpected migration sequence", "found", mRev, "expected", i)
logger.Error("unexpected migration sequence", "found", mRev, "expected", i)
os.Exit(EXIT_FAIL)
}
@ -178,7 +182,7 @@ func main() {
Rev: mRev,
}
migs = append(migs, rev)
rootLog.Debug("loaded db migration", "name", rev.Filename)
logger.Debug("loaded db migration", "name", rev.Filename)
// NOTE: Expect migs to be ordered 0..1..2...
@ -187,11 +191,11 @@ func main() {
// We have migs from migrations dir and appliedMigs from the db
if len(appliedMigs) > len(migs) {
rootLog.Error("wtf", "migs", len(migs), "appliedmigs", len(appliedMigs))
logger.Error("wtf", "migs", len(migs), "appliedmigs", len(appliedMigs))
os.Exit(EXIT_FAIL)
}
rootLog.Info("database currently",
logger.Info("database currently",
"applied_count", appliedMigs[len(appliedMigs)-1] + 1,
"available_count", migs[len(migs)-1].Rev + 1)
@ -204,44 +208,31 @@ func main() {
for {
if !needMigration { break }
rev := migs[pRev]
rlog := rootLog.With("rev", pRev, "name", rev.Filename)
rlog := logger.With("rev", pRev, "name", rev.Filename)
rlog.Info("applying database migration")
rlog.Debug("apply migrations", "sql", string(rev.Data))
tx, err := db.Begin()
if err != nil {
rootLog.Error("error: create db transaction", "err", err.Error())
logger.Error("error: create db transaction", "err", err.Error())
os.Exit(EXIT_FAIL)
}
_, err = tx.Exec(string(rev.Data))
if err != nil {
rootLog.Error("error: tx exec migration", "err", err.Error())
logger.Error("error: tx exec migration", "err", err.Error())
os.Exit(EXIT_FAIL)
}
err = tx.Commit()
if err != nil {
rootLog.Error("error: commit migration tx", "rev", rev.Rev, "err", err.Error())
logger.Error("error: commit migration tx", "rev", rev.Rev, "err", err.Error())
os.Exit(EXIT_FAIL)
}
pRev++
if pRev > migs[len(migs)-1].Rev { break }
}
rootLog.Info("database up-to-date")
// TODO: Create local admin users
// TODO: File transfer, quarintine, scan with local accounts
}
/*
initilize database with sft parameters and table from starting point
*/
func initDb(db sql.DB) {
logger.Info("database up-to-date")
return nil
}