114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
// 这个工具用来修复旧数据库没有 message_seq 字段的问题
|
|
func main() {
|
|
// Open SQLite database
|
|
db, err := sql.Open("sqlite3", "../data.db")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// Create messages table if not exists
|
|
_, err = db.Exec(`
|
|
CREATE TABLE IF NOT EXISTS messages (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
message_id INTEGER UNIQUE NOT NULL,
|
|
message_seq INTEGER
|
|
)
|
|
`)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Fetch messageIDs from database
|
|
rows, err := db.Query("SELECT message_id FROM messages WHERE message_seq IS NULL")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var messageIDs []int
|
|
for rows.Next() {
|
|
var messageID int
|
|
err := rows.Scan(&messageID)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
messageIDs = append(messageIDs, messageID)
|
|
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
for _, messageID := range messageIDs {
|
|
messageSeq, err := fetchMessageSeq(messageID)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
continue
|
|
}
|
|
err = insertOrUpdateMessageSeq(db, messageID, messageSeq)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
continue
|
|
}
|
|
fmt.Printf("MessageSeq for message_id %d is %d\n", messageID, messageSeq)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func fetchMessageSeq(messageID int) (int, error) {
|
|
url := fmt.Sprintf("http://127.0.0.1:5700/get_msg?message_id=%d", messageID)
|
|
fmt.Println("URL:", url)
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
println("Body:", string(body))
|
|
|
|
var messageResp map[string]interface{}
|
|
err = json.Unmarshal(body, &messageResp)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
data, ok := messageResp["data"].(map[string]interface{})
|
|
if !ok {
|
|
return 0, errors.New("data is not a map")
|
|
}
|
|
message_seq := int(data["message_seq"].(float64))
|
|
// println(message_seq)
|
|
return message_seq, nil
|
|
}
|
|
|
|
func insertOrUpdateMessageSeq(db *sql.DB, messageID, messageSeq int) error {
|
|
// Try to insert messageSeq
|
|
// _, err := db.Exec("INSERT INTO messages (message_id, message_seq) VALUES (?, ?)", messageID, messageSeq)
|
|
// if err != nil {
|
|
// If messageID already exists, update messageSeq
|
|
_, err := db.Exec("UPDATE messages SET message_seq = ? WHERE message_id = ?", messageSeq, messageID)
|
|
if err != nil {
|
|
return err
|
|
// }
|
|
}
|
|
|
|
return nil
|
|
}
|