scottk/databunker
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/databunker.git synced 2025-11-01 18:38:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

review DBExists function

Browse Source
This commit is contained in:
root
2020-09-15 18:47:22 +00:00
parent 2131d48ed8
commit 864bd76a5a
2 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/storage/mysql-storage.go
View File

@@ -68,8 +68,10 @@ func (dbobj MySQLDB) DBExists(dbname *string) bool {
dbobj2 := MySQLDB{db}
record, err := dbobj2.GetRecord2(TblName.Xtokens, "token", "", "type", "root")
if record == nil || err != nil {
dbobj2.CloseDB()
return false
}
dbobj2.CloseDB()
return true
}

15
src/storage/sqlite-storage.go
View File

@@ -41,6 +41,21 @@ func (dbobj SQLiteDB) DBExists(filepath *string) bool {
if _, err := os.Stat(dbfile); os.IsNotExist(err) {
return false
}
db, err := sql.Open("sqlite3", "file:"+dbfile+"?_journal_mode=WAL")
if err != nil {
return false
}
err = db.Ping()
if err != nil {
return false
}
dbobj2 := SQLiteDB{db}
record, err := dbobj2.GetRecord2(TblName.Xtokens, "token", "", "type", "root")
if record == nil || err != nil {
dbobj2.CloseDB()
return false
}
dbobj2.CloseDB()
return true
}