rainerosion/E5BotForSQLite
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change Struct Again

Browse Source
This commit is contained in:
iyear 2020-03-27 18:13:14 +08:00
parent c8015361f9
commit ba04f275dc
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
sqlite.go
View File

@ -10,6 +10,7 @@ import (
type MSData struct { type MSData struct {
tgId int tgId int
refreshToken string refreshToken string
msId string
uptime time.Time uptime time.Time
other string other string
} }
@ -17,14 +18,14 @@ type MSData struct {
func init() { func init() {
} }
//update data by refreshtoken //update data by msId
func UpdateData(db *sql.DB, u MSData) (bool, error) { func UpdateData(db *sql.DB, u MSData) (bool, error) {
sqlString := `UPDATE users set tg_id=?,uptime=?,other=? where refresh_token=?` sqlString := `UPDATE users set tg_id=?,refresh_token=?,uptime=?,other=? where ms_id=?`
stmt, err := db.Prepare(sqlString) stmt, err := db.Prepare(sqlString)
if err != nil { if err != nil {
return false, err return false, err
} }
res, err := stmt.Exec(u.tgId, u.uptime, u.other, u.refreshToken) res, err := stmt.Exec(u.tgId, u.refreshToken, u.uptime, u.other, u.msId)
if err != nil { if err != nil {
return false, err return false, err
} }
@ -35,22 +36,22 @@ func UpdateData(db *sql.DB, u MSData) (bool, error) {
//add data //add data
func AddData(db *sql.DB, u MSData) (bool, error) { func AddData(db *sql.DB, u MSData) (bool, error) {
sqlString := ` sqlString := `
INSERT INTO users (tg_id, refresh_token, uptime,other) INSERT INTO users (tg_id, refresh_token,ms_id, uptime,other)
VALUES (?,?,?,?)` VALUES (?,?,?,?,?)`
stmt, err := db.Prepare(sqlString) stmt, err := db.Prepare(sqlString)
if err != nil { if err != nil {
return false, err return false, err
} }
_, err = stmt.Exec(u.tgId, u.refreshToken, u.uptime, u.other) _, err = stmt.Exec(u.tgId, u.refreshToken, u.msId, u.uptime, u.other)
if err != nil { if err != nil {
return false, err return false, err
} }
return true, nil return true, nil
} }
//del data by refresh_token //del data by ms_id
func DelData(db *sql.DB, refreshToken string) (bool, error) { func DelData(db *sql.DB, refreshToken string) (bool, error) {
sqlString := `delete from users where refresh_token=?` sqlString := `delete from users where ms_id=?`
stmt, err := db.Prepare(sqlString) stmt, err := db.Prepare(sqlString)
if err != nil { if err != nil {
return false, err return false, err
@ -68,17 +69,17 @@ func DelData(db *sql.DB, refreshToken string) (bool, error) {
//query data by tg_id //query data by tg_id
func QueryData(db *sql.DB, tgid int) []MSData { func QueryData(db *sql.DB, tgid int) []MSData {
rows, err := db.Query("select * from users where tg_id = ?;", tgid) rows, err := db.Query("select * from users where tg_id = ?", tgid)
CheckErr(err) CheckErr(err)
var result = make([]MSData, 0) var result = make([]MSData, 0)
defer rows.Close() defer rows.Close()
for rows.Next() { for rows.Next() {
var refresht, othert string var refresht, othert, msidt string
var tgIdt int var tgIdt int
var uptimet time.Time var uptimet time.Time
rows.Scan(&tgIdt, &refresht, &uptimet, &othert) rows.Scan(&tgIdt, &refresht, &msidt, &uptimet, &othert)
//fmt.Println(string(tgNamet) + "=>" + uptimet.Format("2006-01-02 15:04:05")) //fmt.Println(string(tgNamet) + "=>" + uptimet.Format("2006-01-02 15:04:05"))
result = append(result, MSData{tgIdt, refresht, uptimet, othert}) result = append(result, MSData{tgIdt, refresht, msidt, uptimet, othert})
} }
return result return result
} }
@ -89,6 +90,7 @@ func CreateTB(db *sql.DB) (bool, error) {
( (
tg_id INTEGER, tg_id INTEGER,
refresh_token TEXT, refresh_token TEXT,
ms_id TEXT,
uptime DATE, uptime DATE,
other TEXT other TEXT
);` );`