113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
package db
|
|
|
|
import (
|
|
"cc/util/log"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
var (
|
|
DB *sqlx.DB
|
|
// DBPath string = "file:weather.pp?_foreign_keys=1&&_journal_mode=WAL&&mode=rwc"
|
|
DBPath string = "file:weather.pp?_foreign_keys=1"
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
DB, err = NewSQLiteDB(DBPath)
|
|
if err != nil {
|
|
log.L.Fatalf("数据库连接失败: %v", err)
|
|
}
|
|
if !IsDBInitialized(DB) {
|
|
log.L.Info("数据库未初始化，正在初始化...")
|
|
createTables()
|
|
initData()
|
|
}
|
|
}
|
|
|
|
// 创建表
|
|
func createTables() {
|
|
// 创建天气类型表
|
|
weatherTypeTable := `
|
|
CREATE TABLE IF NOT EXISTS weather_types (
|
|
id INTEGER PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
description TEXT NOT NULL
|
|
);`
|
|
|
|
// 创建天气表
|
|
weatherTable := `
|
|
CREATE TABLE IF NOT EXISTS weathers (
|
|
project_id INTEGER PRIMARY KEY,
|
|
value TEXT NOT NULL,
|
|
update_at TEXT NOT NULL,
|
|
FOREIGN KEY(project_id) REFERENCES projects(id)
|
|
);`
|
|
|
|
// 创建位置表
|
|
localtionTable := `
|
|
CREATE TABLE IF NOT EXISTS localtions (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL,
|
|
code TEXT NOT NULL
|
|
);`
|
|
|
|
// 创建项目表
|
|
projectTable := `
|
|
CREATE TABLE IF NOT EXISTS projects (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL,
|
|
code TEXT NOT NULL,
|
|
user TEXT NOT NULL,
|
|
description TEXT NOT NULL,
|
|
localtion_id INTEGER NOT NULL,
|
|
weather_type_id INTEGER NOT NULL,
|
|
update_frequency INTEGER NOT NULL,
|
|
dev BOOLEAN NOT NULL,
|
|
FOREIGN KEY(localtion_id) REFERENCES localtions(id),
|
|
FOREIGN KEY(weather_type_id) REFERENCES weather_types(id)
|
|
);`
|
|
|
|
// 创建表
|
|
tables := []string{weatherTypeTable, weatherTable, localtionTable, projectTable}
|
|
for _, table := range tables {
|
|
_, err := DB.Exec(table)
|
|
if err != nil {
|
|
log.L.Fatalf("创建表失败: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 初始化数据
|
|
func initData() {
|
|
// 初始化天气类型
|
|
weatherTypes := map[int][]string{
|
|
1: {"now", "实时天气"},
|
|
3: {"3d", "3天预报"},
|
|
7: {"7d", "7天预报"},
|
|
10: {"10d", "10天预报"},
|
|
15: {"15d", "15天预报"},
|
|
30: {"30d", "30天预报"},
|
|
24: {"24h", "24小时预报"},
|
|
72: {"72h", "72小时预报"},
|
|
168: {"168h", "168小时预报"},
|
|
}
|
|
for id, v := range weatherTypes {
|
|
_, err := DB.Exec("INSERT INTO weather_types (id, name, description) VALUES (?,?,?)", id, v[0], v[1])
|
|
if err != nil {
|
|
log.L.Errorf("初始化天气类型失败: %v", err)
|
|
}
|
|
}
|
|
// 初始化位置
|
|
localtions := map[string]string{
|
|
"西安": "101010100",
|
|
"上海": "101020100",
|
|
}
|
|
for name, code := range localtions {
|
|
_, err := DB.Exec("INSERT INTO localtions (name, code) VALUES (?,?)", name, code)
|
|
if err != nil {
|
|
log.L.Errorf("初始化位置失败: %v", err)
|
|
}
|
|
}
|
|
}
|