🎨 逻辑修改,把依赖从写死改为配置

This commit is contained in:
李寻欢 2023-10-26 10:07:08 +08:00
parent 581faf974c
commit 8f15b58825
12 changed files with 179 additions and 19 deletions

View File

@ -1,6 +1,7 @@
package client
import (
"go-wechat/config"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"log"
@ -10,11 +11,9 @@ import (
var MySQL *gorm.DB
func InitMySQLClient() {
dsn := "wechat:wechat123@tcp(10.0.0.31:3307)/wechat?charset=utf8mb4&parseTime=True&loc=Local"
// 创建连接对象
mysqlConfig := mysql.Config{
DSN: dsn,
DSN: config.Conf.MySQL.GetDSN(),
DontSupportRenameIndex: true, // 重命名索引时采用删除并新建的方式
DontSupportRenameColumn: true, // 用 `change` 重命名列
}

View File

@ -1,13 +1,32 @@
# 微信HOOK配置
wechat:
# 微信HOOK接口地址
host: 10.0.0.73:19088
# 是否在启动的时候自动设置hook服务的回调
autoSetCallback: true
# 回调IP如果是Docker运行本参数必填如果Docker修改了映射格式为 ip:port
callback: 10.0.0.51
# 数据库
mysql:
host: 10.0.0.31
port: 3307
user: wechat
password: wechat123
db: wechat
task:
enable: true
enable: false
syncFriends:
enable: true
cron: '0 * * * *'
waterGroup:
enable: true
cron: '30 9 * * *'
# 需要发送水群排行榜的群Id
groups:
- '18958257758@chatroom'
- '49448748645@chatroom'
# 不计入统计范围的用户Id
blacklist:
- 'wxid_778868788691exit2'
- 'wxid_7788687886912'

View File

@ -6,6 +6,8 @@ var Conf Config
// @description: 配置
type Config struct {
Task task `json:"task" yaml:"task"` // 定时任务配置
MySQL mysql `json:"mysql" yaml:"mysql"` // MySQL 配置
Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
}
// task

24
config/mysql.go Normal file
View File

@ -0,0 +1,24 @@
package config
import (
"fmt"
)
// mysql
// @description: MySQL配置
type mysql struct {
Host string `mapstructure:"host" yaml:"host"` // 主机
Port int `mapstructure:"port" yaml:"port"` // 端口
User string `mapstructure:"user" yaml:"user"` // 用户名
Password string `mapstructure:"password" yaml:"password"` // 密码
Db string `mapstructure:"db" yaml:"db"` // 数据库名称
}
// GetDSN
// @description: 返回 MySQL 连接字符串
// @receiver c
// @return string
func (c mysql) GetDSN() string {
return fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=utf8mb4&parseTime=True&loc=Local",
c.User, c.Password, c.Host, c.Port, c.Db)
}

33
config/wechat.go Normal file
View File

@ -0,0 +1,33 @@
package config
import "strings"
// wxHelper
// @description: 微信助手
type wechat struct {
Host string `json:"host" yaml:"host"` // 接口地址
AutoSetCallback bool `json:"autoSetCallback" yaml:"autoSetCallback"` // 是否自动设置回调地址
Callback string `json:"callback" yaml:"callback"` // 回调地址
}
// Check
// @description: 检查配置是否可用
// @receiver w
// @return bool
func (w wechat) Check() bool {
if w.Host == "" {
return false
}
if w.AutoSetCallback && w.Callback == "" {
return false
}
return true
}
func (w wechat) GetURL(uri string) string {
host := w.Host
if !strings.HasPrefix(w.Host, "http://") {
host = "http://" + w.Host
}
return host + uri
}

1
go.mod
View File

@ -3,6 +3,7 @@ module go-wechat
go 1.21
require (
github.com/duke-git/lancet/v2 v2.2.7
github.com/fsnotify/fsnotify v1.6.0
github.com/go-co-op/gocron v1.34.1
github.com/go-resty/resty/v2 v2.8.0

2
go.sum
View File

@ -51,6 +51,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/duke-git/lancet/v2 v2.2.7 h1:u9zr6HR+MDUvZEtTlAFtSTIgZfEFsN7cKi27n5weZsw=
github.com/duke-git/lancet/v2 v2.2.7/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=

View File

@ -1,8 +1,10 @@
package config
package initialization
import (
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
"go-wechat/client"
"go-wechat/config"
"log"
)
@ -21,23 +23,29 @@ func InitConfig() {
log.Panicf("读取配置文件失败: %v", err)
}
// 绑定配置文件
if err := vp.Unmarshal(&Conf); err != nil {
if err := vp.Unmarshal(&config.Conf); err != nil {
log.Panicf("配置文件解析失败: %v", err)
}
log.Printf("配置文件解析完成: %+v", Conf)
log.Printf("配置文件解析完成: %+v", config.Conf)
if !config.Conf.Wechat.Check() {
log.Panicf("微信HOOK配置缺失")
}
// 初始化数据库连接
//db.Init()
client.InitMySQLClient()
//redis.Init()
// 下面的代码是配置变动之后自动刷新的
vp.WatchConfig()
vp.OnConfigChange(func(e fsnotify.Event) {
// 绑定配置文件
if err := vp.Unmarshal(&Conf); err != nil {
if err := vp.Unmarshal(&config.Conf); err != nil {
log.Printf("配置文件更新失败: %v", err)
} else {
if !config.Conf.Wechat.Check() {
log.Panicf("微信HOOK配置缺失")
}
// 初始化数据库连接
//db.Init()
client.InitMySQLClient()
//redis.Init()
}
})

14
main.go
View File

@ -2,18 +2,19 @@ package main
import (
"bytes"
"go-wechat/client"
"go-wechat/config"
"go-wechat/handler"
"go-wechat/initialization"
"go-wechat/tasks"
"go-wechat/utils"
"io"
"log"
"net"
"time"
)
func init() {
config.InitConfig()
client.InitMySQLClient()
initialization.InitConfig()
tasks.InitTasks()
}
@ -37,6 +38,13 @@ func process(conn net.Conn) {
}
func main() {
// 如果启用了自动配置回调,就设置一下
if config.Conf.Wechat.AutoSetCallback {
utils.ClearCallback()
time.Sleep(500 * time.Millisecond) // 休眠五百毫秒再设置
utils.SetCallback(config.Conf.Wechat.Callback)
}
// 建立 tcp 服务
listen, err := net.Listen("tcp", "0.0.0.0:19099")
if err != nil {

View File

@ -4,6 +4,7 @@ import (
"encoding/json"
"github.com/go-resty/resty/v2"
"go-wechat/client"
"go-wechat/config"
"go-wechat/constant"
"go-wechat/entity"
"go-wechat/model"
@ -27,7 +28,7 @@ func syncFriends() {
resp, err := hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetResult(&base).
Post("http://10.0.0.73:19088/api/getContactList")
Post(config.Conf.Wechat.GetURL("/api/getContactList"))
if err != nil {
log.Printf("获取好友列表失败: %s", err.Error())
return
@ -105,7 +106,7 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&baseResp).
Post("http://10.0.0.73:19088/api/getMemberFromChatRoom")
Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
if err != nil {
log.Printf("获取群成员信息失败: %s", err.Error())
return
@ -189,7 +190,7 @@ func getContactProfile(wxid string) (ent model.ContactProfile, err error) {
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&baseResp).
Post("http://10.0.0.73:19088/api/getContactProfile")
Post(config.Conf.Wechat.GetURL("/api/getContactProfile"))
if err != nil {
log.Printf("获取成员详情失败: %s", err.Error())
return

62
utils/callback.go Normal file
View File

@ -0,0 +1,62 @@
package utils
import (
"encoding/json"
"github.com/duke-git/lancet/v2/netutil"
"github.com/go-resty/resty/v2"
"go-wechat/config"
"log"
"net"
"strconv"
"strings"
)
// ClearCallback
// @description: 清理微信HOOK回调
func ClearCallback() {
res := resty.New()
resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
Post(config.Conf.Wechat.GetURL("/api/unhookSyncMsg"))
if err != nil {
log.Panicf("清理微信HOOK回调失败: %s", err.Error())
}
log.Printf("清理微信HOOK回调结果: %s", resp.String())
}
// SetCallback
// @description: 设置微信HOOK回调
// @param host
func SetCallback(userHost string) {
// 获取本机IP地址
host := net.ParseIP(netutil.GetInternalIp()).String()
port := 19099
if userHost != "" {
uh := strings.Split(strings.TrimSpace(userHost), ":")
host = uh[0]
if len(uh) == 2 {
port, _ = strconv.Atoi(uh[1])
}
}
// 组装参数
param := map[string]any{
"port": port, // socket端口
"ip": host, // socketIP
"url": "", // http接口地址
"timeout": 3000, // 超时毫秒数
"enableHttp": 0, // 是否使用http接口
}
pbs, _ := json.Marshal(param)
log.Printf("设置微信HOOK回调参数: %s", string(pbs))
res := resty.New()
resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
Post(config.Conf.Wechat.GetURL("/api/hookSyncMsg"))
if err != nil {
log.Panicf("设置微信HOOK回调失败: %s", err.Error())
}
log.Printf("设置微信HOOK回调结果: %s", resp.String())
}

View File

@ -3,6 +3,7 @@ package utils
import (
"encoding/json"
"github.com/go-resty/resty/v2"
"go-wechat/config"
"log"
"time"
)
@ -28,7 +29,7 @@ func SendMessage(toId, atId, msg string, retryCount int) {
resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
Post("http://10.0.0.73:19088/api/sendTextMsg")
Post(config.Conf.Wechat.GetURL("/api/sendTextMsg"))
if err != nil {
log.Printf("发送文本消息失败: %s", err.Error())
// 休眠五秒后重新发送