yves/sqldb
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change ids to int64

Browse Source
This commit is contained in:
ycc
2023-07-05 14:37:41 +02:00
parent 4a6840fd2f
commit cde5fe7366
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
db.go
View File

@@ -462,7 +462,7 @@ func (t *TableInfo) buildSelect(key string, columns []string, restriction string
return query return query
} }
func (t *TableInfo) Insert(record AssRow) (int, error) { func (t *TableInfo) Insert(record AssRow) (int64, error) {
columns := "" columns := ""
values := "" values := ""
t, err := t.GetSchema() t, err := t.GetSchema()
@@ -470,7 +470,7 @@ func (t *TableInfo) Insert(record AssRow) (int, error) {
log.Println(err) log.Println(err)
return -1, err return -1, err
} }
var id int var id int64
for key, element := range record { for key, element := range record {
columns += key + "," columns += key + ","
@@ -480,11 +480,17 @@ func (t *TableInfo) Insert(record AssRow) (int, error) {
err = t.db.conn.QueryRow("INSERT INTO " + t.Name + "(" + removeLastChar(columns) + ") VALUES (" + removeLastChar(values) + ") RETURNING id").Scan(&id) err = t.db.conn.QueryRow("INSERT INTO " + t.Name + "(" + removeLastChar(columns) + ") VALUES (" + removeLastChar(values) + ") RETURNING id").Scan(&id)
} }
if t.db.Driver == "mysql" { if t.db.Driver == "mysql" {
_, err = t.db.conn.Query("INSERT INTO " + t.Name + "(" + removeLastChar(columns) + ") VALUES (" + removeLastChar(values) + ")") /* _, err = t.db.conn.Query("INSERT INTO " + t.Name + "(" + removeLastChar(columns) + ") VALUES (" + removeLastChar(values) + ")")
if err != nil {
return id, err
}
err = t.db.conn.QueryRow("SELECT LAST_INSERT_ID()").Scan(&id)*/
stmt, err := t.db.conn.Prepare("INSERT INTO " + t.Name + "(" + removeLastChar(columns) + ") VALUES (" + removeLastChar(values) + ")")
if err != nil { if err != nil {
return id, err return id, err
} }
err = t.db.conn.QueryRow("SELECT LAST_INSERT_ID();").Scan(&id) res, err := stmt.Exec()
id, err = res.LastInsertId()
} }
return id, err return id, err
} }
@@ -542,12 +548,13 @@ func (t *TableInfo) Delete(record AssRow) error {
return nil return nil
} }
func (t *TableInfo) UpdateOrInsert(record AssRow) (int, error) { func (t *TableInfo) UpdateOrInsert(record AssRow) (int64, error) {
id := -1 var id int64
id = -1
for key, element := range record { for key, element := range record {
if key == "id" { if key == "id" {
sid := fmt.Sprintf("%v", element) sid := fmt.Sprintf("%v", element)
id, _ = strconv.Atoi(sid) id, _ = strconv.ParseInt(sid, 10, 64)
break break
} }
} }