feat(database): 实现数据库功能并优化数据导出
- 新增数据库相关 API 和服务 - 实现数据导出功能,支持导出到 JSON 文件 - 优化数据导入流程,增加数据校验 - 新增数据库页面,展示解析数据和统计信息 - 更新捕获页面,支持导入数据到数据库
This commit is contained in:
163
internal/model/database.go
Normal file
163
internal/model/database.go
Normal file
@@ -0,0 +1,163 @@
|
||||
package model
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
// Database 数据库管理器
|
||||
type Database struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// NewDatabase 创建新的数据库连接
|
||||
func NewDatabase() (*Database, error) {
|
||||
dbPath := getDatabasePath()
|
||||
|
||||
// 确保目录存在
|
||||
dir := filepath.Dir(dbPath)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return nil, fmt.Errorf("创建数据库目录失败: %w", err)
|
||||
}
|
||||
|
||||
// 连接数据库
|
||||
db, err := sql.Open("sqlite3", dbPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("连接数据库失败: %w", err)
|
||||
}
|
||||
|
||||
// 测试连接
|
||||
if err := db.Ping(); err != nil {
|
||||
return nil, fmt.Errorf("数据库连接测试失败: %w", err)
|
||||
}
|
||||
|
||||
database := &Database{db: db}
|
||||
|
||||
// 初始化表结构
|
||||
if err := database.initTables(); err != nil {
|
||||
return nil, fmt.Errorf("初始化数据库表失败: %w", err)
|
||||
}
|
||||
|
||||
return database, nil
|
||||
}
|
||||
|
||||
// Close 关闭数据库连接
|
||||
func (d *Database) Close() error {
|
||||
return d.db.Close()
|
||||
}
|
||||
|
||||
// getDatabasePath 获取数据库文件路径
|
||||
func getDatabasePath() string {
|
||||
homeDir, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return "equipment_analyzer.db"
|
||||
}
|
||||
return filepath.Join(homeDir, ".equipment-analyzer", "equipment_analyzer.db")
|
||||
}
|
||||
|
||||
// initTables 初始化数据库表结构
|
||||
func (d *Database) initTables() error {
|
||||
// 解析数据表 - 存储抓包解析后的装备和角色数据
|
||||
parsedDataTable := `
|
||||
CREATE TABLE IF NOT EXISTS parsed_data (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
session_name TEXT NOT NULL,
|
||||
items_json TEXT NOT NULL,
|
||||
heroes_json TEXT NOT NULL,
|
||||
created_at INTEGER NOT NULL
|
||||
);`
|
||||
|
||||
// 应用设置表
|
||||
settingsTable := `
|
||||
CREATE TABLE IF NOT EXISTS app_settings (
|
||||
key TEXT PRIMARY KEY,
|
||||
value TEXT NOT NULL,
|
||||
updated_at INTEGER NOT NULL
|
||||
);`
|
||||
|
||||
tables := []string{
|
||||
parsedDataTable,
|
||||
settingsTable,
|
||||
}
|
||||
|
||||
for _, table := range tables {
|
||||
if _, err := d.db.Exec(table); err != nil {
|
||||
return fmt.Errorf("创建表失败: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// SaveParsedData 保存解析后的数据
|
||||
func (d *Database) SaveParsedData(sessionName string, itemsJSON, heroesJSON string) error {
|
||||
stmt := `
|
||||
INSERT INTO parsed_data (session_name, items_json, heroes_json, created_at)
|
||||
VALUES (?, ?, ?, ?)`
|
||||
|
||||
_, err := d.db.Exec(stmt, sessionName, itemsJSON, heroesJSON, time.Now().Unix())
|
||||
return err
|
||||
}
|
||||
|
||||
// GetLatestParsedData 获取最新的解析数据
|
||||
func (d *Database) GetLatestParsedData() (string, string, error) {
|
||||
stmt := `
|
||||
SELECT items_json, heroes_json
|
||||
FROM parsed_data
|
||||
ORDER BY created_at DESC
|
||||
LIMIT 1`
|
||||
|
||||
var itemsJSON, heroesJSON string
|
||||
err := d.db.QueryRow(stmt).Scan(&itemsJSON, &heroesJSON)
|
||||
if err != nil {
|
||||
return "", "", err
|
||||
}
|
||||
|
||||
return itemsJSON, heroesJSON, nil
|
||||
}
|
||||
|
||||
|
||||
|
||||
// SaveSetting 保存应用设置
|
||||
func (d *Database) SaveSetting(key, value string) error {
|
||||
stmt := "INSERT OR REPLACE INTO app_settings (key, value, updated_at) VALUES (?, ?, ?)"
|
||||
_, err := d.db.Exec(stmt, key, value, time.Now().Unix())
|
||||
return err
|
||||
}
|
||||
|
||||
// GetSetting 获取应用设置
|
||||
func (d *Database) GetSetting(key string) (string, error) {
|
||||
stmt := "SELECT value FROM app_settings WHERE key = ?"
|
||||
var value string
|
||||
err := d.db.QueryRow(stmt, key).Scan(&value)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return value, nil
|
||||
}
|
||||
|
||||
// GetAllSettings 获取所有设置
|
||||
func (d *Database) GetAllSettings() (map[string]string, error) {
|
||||
stmt := "SELECT key, value FROM app_settings"
|
||||
rows, err := d.db.Query(stmt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
settings := make(map[string]string)
|
||||
for rows.Next() {
|
||||
var key, value string
|
||||
if err := rows.Scan(&key, &value); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
settings[key] = value
|
||||
}
|
||||
|
||||
return settings, nil
|
||||
}
|
||||
Reference in New Issue
Block a user