feat: 初步设计工单系统

This commit is contained in:
lixiangwuxian
2024-10-17 02:16:33 +08:00
parent c42a7e5dd7
commit e27eebda77
4 changed files with 125 additions and 0 deletions

View File

@@ -0,0 +1,51 @@
package ticket
import "git.lxtend.com/qqbot/sqlite3"
func init() {
createTableQuery := `CREATE TABLE IF NOT EXISTS ticket (
id INTEGER PRIMARY KEY AUTOINCREMENT,
content TEXT NOT NULL,
fromQQ TEXT NOT NULL,
fromGroup TEXT DEFAULT NULL,
isDone BOOLEAN DEFAULT FALSE,
isDeleted BOOLEAN DEFAULT FALSE,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
`
sqlite3.TryCreateTable(createTableQuery)
}
func storeTicket(content string, fromQQ int64, fromGroup int64) (int64, error) {
ticket := Ticket{
Content: content,
FromQQ: fromQQ,
FromGroup: fromGroup,
}
tx, err := sqlite3.GetTran()
if err != nil {
return 0, err
}
defer tx.Rollback()
res, err := tx.Exec("INSERT INTO ticket (content, fromQQ, fromGroup) VALUES (:content, :fromQQ, :fromGroup)", ticket.Content, ticket.FromQQ, ticket.FromGroup)
if err != nil {
return 0, err
}
id, _ := res.LastInsertId()
tx.Commit()
return id, nil
}
func setTicketDone(id int) error {
tx, err := sqlite3.GetTran()
if err != nil {
return err
}
defer tx.Rollback()
_, err = tx.Exec("UPDATE ticket SET isDone = TRUE WHERE id = ?", id)
if err != nil {
return err
}
tx.Commit()
return nil
}