feat: add system config item for wechat qrcode

This commit is contained in:
RockYang
2024-01-19 16:58:13 +08:00
parent c9cd082855
commit 15b2ec9721
7 changed files with 320 additions and 254 deletions

View File

@@ -195,5 +195,6 @@ type SystemConfig struct {
InviteChatCalls int `json:"invite_chat_calls"` // 邀请用户注册奖励对话次数
InviteImgCalls int `json:"invite_img_calls"` // 邀请用户注册奖励绘图次数
ShowDemoNotice bool `json:"show_demo_notice"` // 显示演示站公告
ShowDemoNotice bool `json:"show_demo_notice"` // 显示演示站公告
WechatCardURL string `json:"wechat_card_url"` // 微信客服地址
}

View File

@@ -26,7 +26,8 @@ import (
)
const ErrorMsg = "抱歉AI 助手开小差了,请稍后再试。"
const ErrImg = "![](/images/wx.png)"
var ErrImg = "![](/images/wx.png)"
var logger = logger2.GetLogger()
@@ -45,6 +46,13 @@ func NewChatHandler(app *core.AppServer, db *gorm.DB, redis *redis.Client) *Chat
return &h
}
func (h *ChatHandler) Init() {
// 如果后台有上传微信客服微信二维码,则覆盖
if h.App.SysConfig.WechatCardURL != "" {
ErrImg = fmt.Sprintf("![](%s)", h.App.SysConfig.WechatCardURL)
}
}
var chatConfig types.ChatConfig
// ChatHandle 处理聊天 WebSocket 请求

View File

@@ -59,11 +59,13 @@ func main() {
}
debug, _ := strconv.ParseBool(os.Getenv("APP_DEBUG"))
logger.Info("Loading config file: ", configFile)
defer func() {
if err := recover(); err != nil {
logger.Error("Panic Error:", err)
}
}()
if !debug {
defer func() {
if err := recover(); err != nil {
logger.Error("Panic Error:", err)
}
}()
}
app := fx.New(
// 初始化配置应用配置
@@ -378,7 +380,9 @@ func main() {
log.Fatal(err)
}
}),
fx.Invoke(func(h *chatimpl.ChatHandler) {
h.Init()
}),
// 注册生命周期回调函数
fx.Invoke(func(lifecycle fx.Lifecycle, lc *AppLifecycle) {
lifecycle.Append(fx.Hook{