refactor: 重构数据库逻辑

service/jrrp/jrrp.go

@@ -2,34 +2,24 @@ package jrrp
 
 import (
 	"database/sql"
-	"fmt"
-	"log"
 	"math"
 	"math/rand"
 	"time"
 
+	"git.lxtend.com/qqbot/sqlite3"
+	"github.com/jmoiron/sqlx"
 	_ "github.com/mattn/go-sqlite3"
 )
 
 // 初始化 SQLite 数据库
 func initDB() {
-	db, err := sql.Open("sqlite3", "./jrrp.db")
-	if err != nil {
-		log.Print(err)
-	}
-	defer db.Close()
-
 	createTableSQL := `CREATE TABLE IF NOT EXISTS jrrp (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         qqid TEXT UNIQUE,
         date TEXT,
         rp_value INTEGER
     );`
-
-	_, err = db.Exec(createTableSQL)
-	if err != nil {
-		log.Print(err)
-	}
+	sqlite3.TryCreateTable(createTableSQL)
 }
 
 // 获取今天的日期
@@ -50,26 +40,21 @@ func rpValueConstructor() (string, int) {
 
 // Jrrp 结构体
 type Jrrp struct {
-	db *sql.DB
 }
 
 // NewJrrp 创建 Jrrp 实例
 func NewJrrp() *Jrrp {
 	initDB()
-	db, err := sql.Open("sqlite3", "./jrrp.db")
-	if err != nil {
-		log.Print(err)
-	}
-	return &Jrrp{db: db}
+	return &Jrrp{}
 }
 
 // findJrrp 查找RP数据
-func (j *Jrrp) findJrrp(qqid string) (string, int, error) {
+func (j *Jrrp) findJrrp(qqid string, tx *sqlx.Tx) (string, int, error) {
 	var date string
 	var rpValue int
 
 	query := "SELECT date, rp_value FROM jrrp WHERE qqid = ?"
-	err := j.db.QueryRow(query, qqid).Scan(&date, &rpValue)
+	err := tx.QueryRow(query, qqid).Scan(&date, &rpValue)
 	if err != nil {
 		if err == sql.ErrNoRows {
 			return "", 0, nil
@@ -80,47 +65,44 @@ func (j *Jrrp) findJrrp(qqid string) (string, int, error) {
 }
 
 // insertJrrp 插入RP数据
-func (j *Jrrp) insertJrrp(qqid string, date string, rpValue int) error {
+func (j *Jrrp) insertJrrp(qqid string, date string, rpValue int, tx *sqlx.Tx) error {
 	query := "INSERT OR REPLACE INTO jrrp (qqid, date, rp_value) VALUES (?, ?, ?)"
-	_, err := j.db.Exec(query, qqid, date, rpValue)
+	_, err := tx.Exec(query, qqid, date, rpValue)
 	return err
 }
 
-// updateJrrp 更新RP数据
-func (j *Jrrp) updateJrrp(qqid string) error {
-	date, _, err := j.findJrrp(qqid)
-	if err != nil {
-		return err
-	}
-	if date == "" {
-		return fmt.Errorf("updateJrrp()-> Error: the qqid does not exist!")
-	}
-	newDate, newRpValue := rpValueConstructor()
-	return j.insertJrrp(qqid, newDate, newRpValue)
-}
-
 // GetJrrp 获取RP值
 func (j *Jrrp) GetJrrp(qqid string) (int, error) {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return 0, err
+	}
+	defer tx.Rollback()
 	today := getTodayFullDate()
 
-	date, rpValue, err := j.findJrrp(qqid)
+	date, rpValue, err := j.findJrrp(qqid, tx)
 	if err != nil {
 		return 0, err
 	}
 
 	if date == "" || date != today {
 		newDate, newRpValue := rpValueConstructor()
-		err = j.insertJrrp(qqid, newDate, newRpValue)
+		err = j.insertJrrp(qqid, newDate, newRpValue, tx)
 		if err != nil {
 			return 0, err
 		}
+		if err = tx.Commit(); err != nil {
+			return 0, err
+		}
 		return newRpValue, nil
 	}
-
+	if err = tx.Commit(); err != nil {
+		return 0, err
+	}
 	return rpValue, nil
 }
 
 // Close 关闭数据库连接
 func (j *Jrrp) Close() {
-	j.db.Close()
+	sqlite3.GetDB().Close()
 }