scottk/databunker
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/databunker.git synced 2025-10-30 01:22:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename storgae functions

Browse Source
This commit is contained in:
yuli
2023-01-02 18:02:50 +02:00
parent fa6b0d91dc
commit 22cc4db0ea
5 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -458,7 +458,7 @@ func (dbobj MySQLDB) LookupRecord(t Tbl, row bson.M) (bson.M, error) {
values = append(values, keyValue) values = append(values, keyValue)
num = num + 1 num = num + 1
} }
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord returns specific record from database // GetRecord returns specific record from database
@@ -467,15 +467,15 @@ func (dbobj MySQLDB) GetRecord(t Tbl, keyName string, keyValue string) (bson.M,
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=?" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=?"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecordInTable returns specific record from database // GetRecordFromTable returns specific record from database
func (dbobj MySQLDB) GetRecordInTable(table string, keyName string, keyValue string) (bson.M, error) { func (dbobj MySQLDB) GetRecordFromTable(table string, keyName string, keyValue string) (bson.M, error) {
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=?" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=?"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord2 returns specific record from database // GetRecord2 returns specific record from database
@@ -487,10 +487,10 @@ func (dbobj MySQLDB) GetRecord2(t Tbl, keyName string, keyValue string,
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
values = append(values, keyValue2) values = append(values, keyValue2)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
func (dbobj MySQLDB) getRecordInTableDo(q string, values []interface{}) (bson.M, error) { func (dbobj MySQLDB) getOneRecord(q string, values []interface{}) (bson.M, error) {
//fmt.Printf("query: %s\n", q) //fmt.Printf("query: %s\n", q)
tx, err := dbobj.db.Begin() tx, err := dbobj.db.Begin()

14
src/storage/pgsql-storage.go
View File

@@ -456,7 +456,7 @@ func (dbobj PGSQLDB) LookupRecord(t Tbl, row bson.M) (bson.M, error) {
values = append(values, keyValue) values = append(values, keyValue)
num = num + 1 num = num + 1
} }
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord returns specific record from database // GetRecord returns specific record from database
@@ -465,15 +465,15 @@ func (dbobj PGSQLDB) GetRecord(t Tbl, keyName string, keyValue string) (bson.M,
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecordInTable returns specific record from database // GetRecordFromTable returns specific record from database
func (dbobj PGSQLDB) GetRecordInTable(table string, keyName string, keyValue string) (bson.M, error) { func (dbobj PGSQLDB) GetRecordFromTable(table string, keyName string, keyValue string) (bson.M, error) {
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord2 returns specific record from database // GetRecord2 returns specific record from database
@@ -485,10 +485,10 @@ func (dbobj PGSQLDB) GetRecord2(t Tbl, keyName string, keyValue string,
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
values = append(values, keyValue2) values = append(values, keyValue2)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
func (dbobj PGSQLDB) getRecordInTableDo(q string, values []interface{}) (bson.M, error) { func (dbobj PGSQLDB) getOneRecord(q string, values []interface{}) (bson.M, error) {
//fmt.Printf("query: %s\n", q) //fmt.Printf("query: %s\n", q)
tx, err := dbobj.db.Begin() tx, err := dbobj.db.Begin()

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

@@ -445,7 +445,7 @@ func (dbobj SQLiteDB) LookupRecord(t Tbl, row bson.M) (bson.M, error) {
values = append(values, keyValue) values = append(values, keyValue)
num = num + 1 num = num + 1
} }
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord returns specific record from database // GetRecord returns specific record from database
@@ -454,15 +454,15 @@ func (dbobj SQLiteDB) GetRecord(t Tbl, keyName string, keyValue string) (bson.M,
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecordInTable returns specific record from database // GetRecordFromTable returns specific record from database
func (dbobj SQLiteDB) GetRecordInTable(table string, keyName string, keyValue string) (bson.M, error) { func (dbobj SQLiteDB) GetRecordFromTable(table string, keyName string, keyValue string) (bson.M, error) {
q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1" q := "select * from " + table + " WHERE " + dbobj.escapeName(keyName) + "=$1"
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
// GetRecord2 returns specific record from database // GetRecord2 returns specific record from database
@@ -474,10 +474,10 @@ func (dbobj SQLiteDB) GetRecord2(t Tbl, keyName string, keyValue string,
values := make([]interface{}, 0) values := make([]interface{}, 0)
values = append(values, keyValue) values = append(values, keyValue)
values = append(values, keyValue2) values = append(values, keyValue2)
return dbobj.getRecordInTableDo(q, values) return dbobj.getOneRecord(q, values)
} }
func (dbobj SQLiteDB) getRecordInTableDo(q string, values []interface{}) (bson.M, error) { func (dbobj SQLiteDB) getOneRecord(q string, values []interface{}) (bson.M, error) {
//fmt.Printf("query: %s\n", q) //fmt.Printf("query: %s\n", q)
tx, err := dbobj.db.Begin() tx, err := dbobj.db.Begin()

2
src/storage/storage.go
View File

@@ -87,7 +87,7 @@ type BackendDB interface {
UpdateRecordInTable2(string, string, string, string, string, *bson.M, *bson.M) (int64, error) UpdateRecordInTable2(string, string, string, string, string, *bson.M, *bson.M) (int64, error)
LookupRecord(Tbl, bson.M) (bson.M, error) LookupRecord(Tbl, bson.M) (bson.M, error)
GetRecord(Tbl, string, string) (bson.M, error) GetRecord(Tbl, string, string) (bson.M, error)
GetRecordInTable(string, string, string) (bson.M, error) GetRecordFromTable(string, string, string) (bson.M, error)
GetRecord2(Tbl, string, string, string, string) (bson.M, error) GetRecord2(Tbl, string, string, string, string) (bson.M, error)
DeleteRecord(Tbl, string, string) (int64, error) DeleteRecord(Tbl, string, string) (int64, error)
DeleteRecordInTable(string, string, string) (int64, error) DeleteRecordInTable(string, string, string) (int64, error)

10
src/userapps_db.go
View File

@@ -17,7 +17,7 @@ func (dbobj dbcon) getUserApp(userTOKEN string, appName string, conf Config) ([]
var record bson.M var record bson.M
var err error var err error
if conf.Generic.UseSeparateAppTables == true { if conf.Generic.UseSeparateAppTables == true {
record, err = dbobj.store.GetRecordInTable(appNameFull, "token", userTOKEN) record, err = dbobj.store.GetRecordFromTable(appNameFull, "token", userTOKEN)
} else { } else {
record, err = dbobj.store.GetRecord2(storage.TblName.Userapps, "token", userTOKEN, "appname", appName) record, err = dbobj.store.GetRecord2(storage.TblName.Userapps, "token", userTOKEN, "appname", appName)
} }
@@ -78,7 +78,7 @@ func (dbobj dbcon) createAppRecord(jsonData []byte, userTOKEN string, appName st
} }
//fmt.Println("creating new app") //fmt.Println("creating new app")
if conf.Generic.UseSeparateAppTables == true { if conf.Generic.UseSeparateAppTables == true {
record, err := dbobj.store.GetRecordInTable(appNameFull, "token", userTOKEN) record, err := dbobj.store.GetRecordFromTable(appNameFull, "token", userTOKEN)
if err != nil { if err != nil {
return userTOKEN, err return userTOKEN, err
} }
@@ -118,7 +118,7 @@ func (dbobj dbcon) updateAppRecord(jsonDataPatch []byte, userTOKEN string, appNa
} }
var record bson.M var record bson.M
if conf.Generic.UseSeparateAppTables == true { if conf.Generic.UseSeparateAppTables == true {
record, err = dbobj.store.GetRecordInTable(appNameFull, "token", userTOKEN) record, err = dbobj.store.GetRecordFromTable(appNameFull, "token", userTOKEN)
} else { } else {
record, err = dbobj.store.GetRecord2(storage.TblName.Userapps, "token", userTOKEN, "appname", appName) record, err = dbobj.store.GetRecord2(storage.TblName.Userapps, "token", userTOKEN, "appname", appName)
} }
@@ -208,7 +208,7 @@ func (dbobj dbcon) listUserApps(userTOKEN string, conf Config) ([]byte, error) {
var result []string var result []string
for _, colName := range allCollections { for _, colName := range allCollections {
if strings.HasPrefix(colName, "app_") { if strings.HasPrefix(colName, "app_") {
record, err := dbobj.store.GetRecordInTable(colName, "token", userTOKEN) record, err := dbobj.store.GetRecordFromTable(colName, "token", userTOKEN)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -247,7 +247,7 @@ func (dbobj dbcon) dumpUserApps(userTOKEN string, conf Config) ([]byte, error) {
} }
for _, colName := range allCollections { for _, colName := range allCollections {
if strings.HasPrefix(colName, "app_") { if strings.HasPrefix(colName, "app_") {
record, err := dbobj.store.GetRecordInTable(colName, "token", userTOKEN) record, err := dbobj.store.GetRecordFromTable(colName, "token", userTOKEN)
if err != nil { if err != nil {
return nil, err return nil, err
} }