openclaw
0c1a4f06f7
- Telegram Bot + QQ Bot (WebSocket) 双平台支持 - 150+ 预设分类关键词，jieba 智能分词 - Web 管理后台（记录查看/删除/CSV导出） - 金额精确存储（分/int64） - 版本信息嵌入（编译时注入） - Docker 支持 - 优雅关闭（context + signal）
103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
"time"
|
||
|
||
"xiaji-go/config"
|
||
"xiaji-go/internal/bot"
|
||
"xiaji-go/internal/qq"
|
||
"xiaji-go/internal/service"
|
||
"xiaji-go/internal/web"
|
||
"xiaji-go/models"
|
||
"xiaji-go/version"
|
||
|
||
"gorm.io/driver/sqlite"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
func main() {
|
||
// 版本信息
|
||
if len(os.Args) > 1 && (os.Args[1] == "-v" || os.Args[1] == "--version" || os.Args[1] == "version") {
|
||
fmt.Println(version.Info())
|
||
return
|
||
}
|
||
|
||
log.Printf("🦞 %s", version.Info())
|
||
|
||
// 1. 加载配置
|
||
cfgPath := "config.yaml"
|
||
if len(os.Args) > 1 {
|
||
cfgPath = os.Args[1]
|
||
}
|
||
|
||
cfg, err := config.LoadConfig(cfgPath)
|
||
if err != nil {
|
||
log.Fatalf("无法加载配置: %v", err)
|
||
}
|
||
|
||
// 2. 初始化数据库
|
||
db, err := gorm.Open(sqlite.Open(cfg.Database.Path), &gorm.Config{})
|
||
if err != nil {
|
||
log.Fatalf("无法连接数据库: %v", err)
|
||
}
|
||
|
||
// 3. 自动迁移表结构
|
||
if err := models.Migrate(db); err != nil {
|
||
log.Fatalf("数据库迁移失败: %v", err)
|
||
}
|
||
|
||
// 4. 初始化核心服务
|
||
finance := service.NewFinanceService(db)
|
||
defer finance.Close()
|
||
|
||
// 全局 context，用于优雅退出
|
||
ctx, cancel := context.WithCancel(context.Background())
|
||
defer cancel()
|
||
|
||
// 5. 启动 Telegram Bot
|
||
if cfg.Telegram.Enabled {
|
||
tgBot, err := bot.NewTGBot(cfg.Telegram.Token, finance)
|
||
if err != nil {
|
||
log.Printf("⚠️ TG Bot 启动失败: %v", err)
|
||
} else {
|
||
go tgBot.Start(ctx)
|
||
}
|
||
}
|
||
|
||
// 6. 启动 QQ Bot
|
||
if cfg.QQBot.Enabled {
|
||
qqBot := qq.NewQQBot(cfg.QQBot.AppID, cfg.QQBot.Secret, finance)
|
||
go qqBot.Start(ctx)
|
||
}
|
||
|
||
// 7. 启动 Web 后台
|
||
webServer := web.NewWebServer(db, cfg.Server.Port, cfg.Admin.Username, cfg.Admin.Password)
|
||
go webServer.Start()
|
||
|
||
// 8. 优雅关闭
|
||
log.Println("🦞 Xiaji-Go 已全面启动")
|
||
sig := make(chan os.Signal, 1)
|
||
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
|
||
<-sig
|
||
|
||
log.Println("⏳ 正在关闭服务...")
|
||
cancel() // 通知所有 goroutine 停止
|
||
|
||
// 等待一点时间让 goroutine 退出
|
||
time.Sleep(2 * time.Second)
|
||
|
||
// 关闭数据库连接
|
||
sqlDB, err := db.DB()
|
||
if err == nil {
|
||
sqlDB.Close()
|
||
}
|
||
|
||
log.Println("👋 Xiaji-Go 已关闭")
|
||
}
|