430 lines
16 KiB
Go
430 lines
16 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"xiawan/wx/api/req"
|
|
"xiawan/wx/api/vo"
|
|
"xiawan/wx/clientsdk"
|
|
"xiawan/wx/clientsdk/baseinfo"
|
|
"xiawan/wx/clientsdk/proxynet"
|
|
"xiawan/wx/db"
|
|
"xiawan/wx/db/table"
|
|
"xiawan/wx/protobuf/wechat"
|
|
"xiawan/wx/srv"
|
|
"xiawan/wx/srv/srvconfig"
|
|
"xiawan/wx/srv/websrv"
|
|
"xiawan/wx/srv/wxcore"
|
|
"xiawan/wx/srv/wxface"
|
|
"xiawan/wx/srv/wxrouter"
|
|
|
|
"github.com/lunny/log"
|
|
)
|
|
|
|
// 基础服务功能
|
|
var WXServer wxface.IWXServer
|
|
|
|
type BusinessFunc = func(connect wxface.IWXConnect, newIWXConnect bool) vo.DTO
|
|
type USerInfoCallFunc = func(account *srv.WXAccount, invoker wxface.IWXReqInvoker, state uint32) vo.DTO
|
|
|
|
// 初始化微信响应路由
|
|
func InitWXServerRouter() {
|
|
// 开启服务器
|
|
wxServer := wxcore.NewWXServer()
|
|
// 注册微信响应路由
|
|
// wxServer.AddWXRouter(baseinfo.MMRequestTypeGetLoginQRCode, new(wxrouter.WXGetLoginQrcodeRouter)) // 获取登陆二维码响应
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeCheckLoginQRCode, new(wxrouter.WXCheckQrcodeRouter)) // 检测二维码状态请求
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypePushQrLogin, new(wxrouter.WXPushQrCodeLoginRouter)) // 唤醒登录
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeManualAuth, new(wxrouter.WXManualAuthRouter))
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeHybridManualAuth, new(wxrouter.WXManualAuthRouter)) // 登录
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeLogout, new(wxrouter.WXLogoutRouter)) // 退出登录
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeAutoAuth, new(wxrouter.WXAutoAuthRouter)) // token登陆
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeNewSync, new(wxrouter.WXNewSyncRouter)) // 同步消息，联系人
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeNewInit, new(wxrouter.WXNewInitRouter)) // 首次登录初始化
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeHeartBeat, new(wxrouter.WXHeartBeatRouter)) // 心跳包
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeGetProfile, new(wxrouter.WXGetProfileRouter)) // 获取帐号信息
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeGetCdnDNS, new(wxrouter.WXGetCDNDnsRouter)) // 获取CDNDns信息
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeInitContact, new(wxrouter.WXInitContactRouter)) // 初始化联系人
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeBatchGetContactBriefInfo, new(wxrouter.WXBatchGetContactBriefInfoReqRouter)) // 初始化联系人
|
|
// wxServer.AddWXRouter(baseinfo.MMRequestTypeGetContact, new(wxrouter.WXGetContactRouter)) // 批量获取联系人信息
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeReceiveWxHB, new(wxrouter.WXReceiveHBRouter)) // 接收红包
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeOpenWxHB, new(wxrouter.WXOpenHBRouter)) // 打开红包
|
|
// wxServer.AddWXRouter(baseinfo.MMRequestTypeOplog, new(wxrouter.WXOplogRouter)) // Oplog请求
|
|
// wxServer.AddWXRouter(baseinfo.MMRequestTypeNewSendMsg, new(wxrouter.WXNewSendMsgRouter)) // 发送文本消息
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeFavSync, new(wxrouter.WXFavSyncRouter)) // 同步收藏
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeGetFavInfo, new(wxrouter.WXGetFavInfoRouter)) // 获取收藏信息
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeBatchGetFavItem, new(wxrouter.WXBatchGetFavItemRouter)) // 获取单条收藏（收藏转发）
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeMMSnsPost, new(wxrouter.WXSnsPostRouter)) // 发送朋友圈
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeMMSnsSync, new(wxrouter.WXSnsSyncRouter)) // 同步朋友圈
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeMMSnsUserPage, new(wxrouter.WXSnsUserPageRouter)) // 朋友圈同步转发
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeMMSnsTimeLine, new(wxrouter.WXSnsTimeLineRouter)) // 同步朋友圈
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeMMSnsComment, new(wxrouter.WXSnsCommentRouter)) // 评论/点赞朋友圈
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeGetContactLabelList, new(wxrouter.WXGetContactLabelListRouter)) // 获取标签列表
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeAddContactLabel, new(wxrouter.WXAddContactLabelRouter)) // 新增标签列表
|
|
wxServer.AddWXRouter(baseinfo.MMRequestTypeGetQrCode, new(wxrouter.WXGetQrcodeRouter)) // 获取二维码
|
|
// wxServer.AddWXRouter(baseinfo.MMRequestTypeBindQueryNew, new(wxrouter.WXBindQueryNewRouter)) // 获取钱包信息
|
|
wxServer.Start()
|
|
WXServer = wxServer
|
|
}
|
|
|
|
func CreateWXConnectByQueryKey(queryKey, proxy string, wxAccount *srv.WXAccount) wxface.IWXConnect {
|
|
var proxyInfo *proxynet.WXProxyInfo
|
|
if proxy != "" {
|
|
proxyInfo = proxynet.ParseWXProxyInfo(proxy)
|
|
}
|
|
|
|
if wxAccount == nil {
|
|
wxAccount = srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, proxyInfo, uint32(0), nil)
|
|
|
|
}
|
|
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
return wxConnect
|
|
}
|
|
|
|
// checkExIdReturn 检测实例Id并返回实例
|
|
func checkExIdReturn(queryKey string, ClientVersion uint32) wxface.IWXConnect {
|
|
// 判断 ClientVersion == 0
|
|
if !(int(ClientVersion) > 0) {
|
|
ClientVersion = baseinfo.ClientVersion
|
|
}
|
|
//获取实例管理器
|
|
connectMgr := WXServer.GetWXConnectMgr()
|
|
//查询的queryKey为空创建一个链接实例
|
|
if queryKey == "" {
|
|
return nil
|
|
}
|
|
|
|
//查询该链接是否存在
|
|
iwxConnect := connectMgr.GetWXConnectByUserInfoUUID(queryKey)
|
|
if iwxConnect != nil {
|
|
//执行回调方法
|
|
return iwxConnect
|
|
} else {
|
|
//如果链接管理器不存在该链接查询数据库是否存在
|
|
dbUserInfo := db.GetUSerInfoByUUID(queryKey)
|
|
//数据库存在该链接数据 重新实例化一个链接对象
|
|
if dbUserInfo != nil {
|
|
if !(int(dbUserInfo.ClientVersion) > 0) {
|
|
dbUserInfo.ClientVersion = ClientVersion
|
|
}
|
|
//创建一个用户信息
|
|
wxAccount := srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, nil, dbUserInfo.ClientVersion, dbUserInfo)
|
|
//创建一个新链接
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
//设置用户信息
|
|
wxAccount.SetUserInfo(dbUserInfo)
|
|
return wxConnect
|
|
} else {
|
|
//创建新一个用户信息
|
|
wxAccount := srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, nil, ClientVersion, dbUserInfo)
|
|
//创建一个新链接
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
return wxConnect
|
|
}
|
|
}
|
|
}
|
|
|
|
// 检查实例Id是否存在 不存在创建新的链接
|
|
func checkExIdPerform(queryKey string, ClientVersion uint32, businessFunc BusinessFunc) vo.DTO {
|
|
// 判断 ClientVersion == 0
|
|
if !(int(ClientVersion) > 0) {
|
|
ClientVersion = baseinfo.ClientVersion
|
|
}
|
|
//获取实例管理器
|
|
connectMgr := WXServer.GetWXConnectMgr()
|
|
//查询的queryKey为空创建一个链接实例
|
|
if queryKey == "" {
|
|
return businessFunc(nil, true)
|
|
}
|
|
//查询该链接是否存在
|
|
iwxConnect := connectMgr.GetWXConnectByUserInfoUUID(queryKey)
|
|
if iwxConnect != nil {
|
|
//执行回调方法
|
|
return businessFunc(iwxConnect, false)
|
|
} else {
|
|
//如果链接管理器不存在该链接查询数据库是否存在
|
|
dbUserInfo := db.GetUSerInfoByUUID(queryKey)
|
|
//数据库存在该链接数据 重新实例化一个链接对象
|
|
if dbUserInfo != nil {
|
|
if !(int(dbUserInfo.ClientVersion) > 0) {
|
|
dbUserInfo.ClientVersion = ClientVersion
|
|
}
|
|
// 设置代理信息
|
|
key := fmt.Sprintf("%s%s", "wechat:Proxy:", queryKey)
|
|
exists, _ := db.Exists(key)
|
|
newModel := &req.GetLoginQrCodeModel{
|
|
Proxy: "",
|
|
Check: false,
|
|
}
|
|
if exists {
|
|
db.GETObj(key, &newModel)
|
|
}
|
|
var proxyInfo *proxynet.WXProxyInfo
|
|
if newModel.Proxy != "" {
|
|
proxyInfo = proxynet.ParseWXProxyInfo(newModel.Proxy)
|
|
} else {
|
|
proxyInfo = nil
|
|
}
|
|
//创建一个用户信息
|
|
wxAccount := srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, proxyInfo, dbUserInfo.ClientVersion, dbUserInfo)
|
|
//创建一个新链接
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
//设置用户信息
|
|
wxAccount.SetUserInfo(dbUserInfo)
|
|
return businessFunc(wxConnect, false)
|
|
} else {
|
|
// 设置代理信息
|
|
key := fmt.Sprintf("%s%s", "wechat:Proxy:", queryKey)
|
|
exists, _ := db.Exists(key)
|
|
newModel := &req.GetLoginQrCodeModel{
|
|
Proxy: "",
|
|
Check: false,
|
|
}
|
|
if exists {
|
|
db.GETObj(key, &newModel)
|
|
}
|
|
var proxyInfo *proxynet.WXProxyInfo
|
|
if newModel.Proxy != "" {
|
|
proxyInfo = proxynet.ParseWXProxyInfo(newModel.Proxy)
|
|
} else {
|
|
proxyInfo = nil
|
|
}
|
|
//创建新一个用户信息
|
|
wxAccount := srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, proxyInfo, ClientVersion, dbUserInfo)
|
|
//创建一个新链接
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
return businessFunc(wxConnect, true)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 检查实例Id是否存在 链接不存在返回错误不创建新链接
|
|
func checkExIdPerformNoCreateConnect(queryKey string, businessFunc BusinessFunc) vo.DTO {
|
|
ClientVersion := baseinfo.ClientVersion
|
|
//获取实例管理器
|
|
connectMgr := WXServer.GetWXConnectMgr()
|
|
//查询的queryKey为空创建一个链接实例
|
|
if queryKey == "" {
|
|
return businessFunc(nil, true)
|
|
}
|
|
|
|
//查询该链接是否存在
|
|
iwxConnect := connectMgr.GetWXConnectByUserInfoUUID(queryKey)
|
|
if iwxConnect != nil {
|
|
//执行回调方法
|
|
return businessFunc(iwxConnect, false)
|
|
} else {
|
|
//如果链接管理器不存在该链接查询数据库是否存在
|
|
dbUserInfo := db.GetUSerInfoByUUID(queryKey)
|
|
//数据库存在该链接数据 重新实例化一个链接对象
|
|
if dbUserInfo != nil {
|
|
if !(int(dbUserInfo.ClientVersion) > 0) {
|
|
dbUserInfo.ClientVersion = ClientVersion
|
|
}
|
|
// 设置代理信息
|
|
key := fmt.Sprintf("%s%s", "wechat:Proxy:", dbUserInfo.UUID)
|
|
exists, _ := db.Exists(key)
|
|
newModel := &req.GetLoginQrCodeModel{
|
|
Proxy: "",
|
|
Check: false,
|
|
}
|
|
if exists {
|
|
db.GETObj(key, &newModel)
|
|
}
|
|
var proxyInfo *proxynet.WXProxyInfo
|
|
if newModel.Proxy != "" {
|
|
proxyInfo = proxynet.ParseWXProxyInfo(newModel.Proxy)
|
|
} else {
|
|
proxyInfo = nil
|
|
}
|
|
|
|
//创建一个用户信息
|
|
wxAccount := srv.NewWXAccount(&websrv.TaskInfo{
|
|
UUID: queryKey,
|
|
}, proxyInfo, dbUserInfo.ClientVersion, dbUserInfo)
|
|
|
|
//创建一个新链接
|
|
wxConnect := wxcore.NewWXConnect(WXServer, wxAccount)
|
|
|
|
//设置用户信息
|
|
wxAccount.SetUserInfo(dbUserInfo)
|
|
return businessFunc(wxConnect, false)
|
|
} else {
|
|
// 诊断逻辑：为什么查不到？
|
|
reason := "未知原因"
|
|
var uEntity table.UserInfoEntity
|
|
err := db.MysqlDB.Where("uuid = ?", queryKey).First(&uEntity).Error
|
|
if err != nil {
|
|
reason = fmt.Sprintf("数据库UserInfo表未找到该UUID记录: %v", err)
|
|
} else {
|
|
// UUID 存在，检查 TargetIp
|
|
if srvconfig.GlobalSetting.TargetIp != "" && uEntity.TargetIp != srvconfig.GlobalSetting.TargetIp {
|
|
reason = fmt.Sprintf("TargetIp不匹配(DB:%s, Config:%s)", uEntity.TargetIp, srvconfig.GlobalSetting.TargetIp)
|
|
} else {
|
|
// 检查 DeviceInfo
|
|
// 判断是62还是A16
|
|
key := fmt.Sprintf("%s%s", "wechat:a16DeviceInfo:", uEntity.WxId)
|
|
exists, _ := db.Exists(key)
|
|
if !exists {
|
|
var dEntity table.DeviceInfoEntity
|
|
if err := db.MysqlDB.Where("wxId = ?", uEntity.WxId).First(&dEntity).Error; err != nil {
|
|
reason = fmt.Sprintf("缺少设备信息(DeviceInfo表无记录且Redis无A16缓存), WxId: %s, Err: %v", uEntity.WxId, err)
|
|
} else {
|
|
reason = "UserInfo和DeviceInfo都存在，但GetUSerInfoByUUID返回nil，可能是其他数据完整性问题"
|
|
}
|
|
} else {
|
|
reason = "UserInfo存在且有A16缓存，但GetUSerInfoByUUID返回nil，可能是反序列化或其他问题"
|
|
}
|
|
}
|
|
}
|
|
|
|
return vo.DTO{
|
|
Code: vo.FAIL_UUID,
|
|
Data: nil,
|
|
Text: fmt.Sprintf("%s 该链接不存在！调试信息: %s", queryKey, reason),
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 获取缓存中已存在的 wx 链接
|
|
func getExistWxConnect(queryKey string, businessFunc BusinessFunc) vo.DTO {
|
|
//获取实例管理器
|
|
connectMgr := WXServer.GetWXConnectMgr()
|
|
//查询的queryKey为空创建一个链接实例
|
|
if queryKey == "" {
|
|
return businessFunc(nil, true)
|
|
}
|
|
|
|
//查询该链接是否存在
|
|
iwxConnect := connectMgr.GetWXConnectByUserInfoUUID(queryKey)
|
|
if iwxConnect != nil {
|
|
//执行回调方法
|
|
userInfo := iwxConnect.GetWXAccount().GetUserInfo()
|
|
iwxConnect.GetWXAccount().SetUserInfo(userInfo)
|
|
return businessFunc(iwxConnect, false)
|
|
}
|
|
return businessFunc(nil, true)
|
|
}
|
|
|
|
func DeepFields(ifaceType reflect.Type) []reflect.StructField {
|
|
var fields []reflect.StructField
|
|
|
|
for i := 0; i < ifaceType.NumField(); i++ {
|
|
v := ifaceType.Field(i)
|
|
if v.Anonymous && v.Type.Kind() == reflect.Struct {
|
|
fields = append(fields, DeepFields(v.Type)...)
|
|
} else {
|
|
fields = append(fields, v)
|
|
}
|
|
}
|
|
|
|
return fields
|
|
}
|
|
|
|
func StructCopy(DstStructPtr interface{}, SrcStructPtr interface{}) {
|
|
srcv := reflect.ValueOf(SrcStructPtr)
|
|
dstv := reflect.ValueOf(DstStructPtr)
|
|
srct := reflect.TypeOf(SrcStructPtr)
|
|
dstt := reflect.TypeOf(DstStructPtr)
|
|
if srct.Kind() != reflect.Ptr || dstt.Kind() != reflect.Ptr ||
|
|
srct.Elem().Kind() == reflect.Ptr || dstt.Elem().Kind() == reflect.Ptr {
|
|
panic("Fatal error:type of parameters must be Ptr of value")
|
|
}
|
|
if srcv.IsNil() || dstv.IsNil() {
|
|
panic("Fatal error:value of parameters should not be nil")
|
|
}
|
|
srcV := srcv.Elem()
|
|
dstV := dstv.Elem()
|
|
srcfields := DeepFields(reflect.ValueOf(SrcStructPtr).Elem().Type())
|
|
for _, v := range srcfields {
|
|
if v.Anonymous {
|
|
continue
|
|
}
|
|
dst := dstV.FieldByName(v.Name)
|
|
src := srcV.FieldByName(v.Name)
|
|
if !dst.IsValid() {
|
|
continue
|
|
}
|
|
if src.Type() == dst.Type() && dst.CanSet() {
|
|
dst.Set(src)
|
|
continue
|
|
}
|
|
if src.Kind() == reflect.Ptr && !src.IsNil() && src.Type().Elem() == dst.Type() {
|
|
dst.Set(src.Elem())
|
|
continue
|
|
}
|
|
if dst.Kind() == reflect.Ptr && dst.Type().Elem() == src.Type() {
|
|
dst.Set(reflect.New(src.Type()))
|
|
dst.Elem().Set(src)
|
|
continue
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
/*
|
|
*
|
|
获取DeviceToken
|
|
*/
|
|
func checkDeviceToken(tmpUserInfo *baseinfo.UserInfo) {
|
|
//如果是android
|
|
if strings.HasPrefix(tmpUserInfo.LoginDataInfo.LoginData, "A") {
|
|
key := fmt.Sprintf("%s%s", "wechat:deviceTokenA16:", tmpUserInfo.LoginDataInfo.UserName)
|
|
exists, _ := db.Exists(key)
|
|
if exists {
|
|
//A16存redis
|
|
trustRes := &wechat.TrustResp{}
|
|
error := db.GETObj(key, &trustRes)
|
|
if error != nil {
|
|
log.Error("redis deviceToken is error=" + error.Error())
|
|
}
|
|
tmpUserInfo.DeviceInfoA16.DeviceToken = trustRes
|
|
} else {
|
|
tmpUserInfo.DeviceInfoA16.DeviceId = []byte(tmpUserInfo.LoginDataInfo.LoginData[:15])
|
|
tmpUserInfo.DeviceInfoA16.DeviceIdStr = tmpUserInfo.LoginDataInfo.LoginData
|
|
deviceTokenRsp, err := clientsdk.SendAndroIdDeviceTokenRequest(tmpUserInfo)
|
|
if err != nil {
|
|
log.Error("android 请求 deviceTokenRequest error!")
|
|
}
|
|
//保存5天
|
|
db.SETExpirationObj(key, &deviceTokenRsp, 60*60*24*5)
|
|
tmpUserInfo.DeviceInfoA16.DeviceToken = deviceTokenRsp
|
|
}
|
|
} else if strings.HasPrefix(tmpUserInfo.LoginDataInfo.LoginData, "62") {
|
|
key := fmt.Sprintf("%s%s", "wechat:deviceTokenIos:", tmpUserInfo.LoginDataInfo.UserName)
|
|
exists, _ := db.Exists(key)
|
|
if exists {
|
|
//ios存redis
|
|
trustRes := &wechat.TrustResp{}
|
|
error := db.GETObj(key, &trustRes)
|
|
if error != nil {
|
|
log.Error("ios redis deviceTokenIos is error=" + error.Error())
|
|
}
|
|
|
|
tmpUserInfo.DeviceInfo.DeviceToken = trustRes
|
|
} else {
|
|
deviceTokenRsp, err := clientsdk.SendIosDeviceTokenRequest(tmpUserInfo)
|
|
if err != nil {
|
|
log.Error("ios 请求 deviceTokenRequest error!")
|
|
}
|
|
//保存5天
|
|
db.SETExpirationObj(key, &deviceTokenRsp, 60*60*24*5)
|
|
tmpUserInfo.DeviceInfo.DeviceToken = deviceTokenRsp
|
|
fmt.Println(deviceTokenRsp, 333)
|
|
}
|
|
}
|
|
}
|