227 lines
8.4 KiB
Go
227 lines
8.4 KiB
Go
|
package mongo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"finclip-app-manager/domain/entity"
|
||
|
|
"finclip-app-manager/domain/repository"
|
||
|
|
"finclip-app-manager/infrastructure/utility"
|
||
|
|
"finclip-app-manager/infrastructure/utils"
|
||
|
|
mgo "gitlab.finogeeks.club/finclip-backend-v2/finclip-mgo"
|
||
|
|
"gitlab.finogeeks.club/finclip-backend-v2/finclip-mgo/bson"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ repository.IQrCodeInfoRepo = new(QrCodeInfoByMongoRepo)
|
||
|
|
|
||
|
|
type QrCodeInfoByMongoRepo struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
type QrCodeInfoMongo struct {
|
||
|
|
Type string `json:"type" bson:"type"` //二维码的类型
|
||
|
|
Uuid string `json:"uuid" bson:"uuid"` //标识该二维码
|
||
|
|
AppId string `json:"appId" bson:"app_id"` //小程序Id
|
||
|
|
Sequence int `json:"sequence" bson:"sequence"` //小程序序列号
|
||
|
|
ApiServer string `json:"apiServer" bson:"api_server"` //小程序apiServer
|
||
|
|
CodeId string `json:"codeId" bson:"code_id"` //标识某个编译版本的id
|
||
|
|
PathAndQuery string `json:"pathAndQuery" bson:"path_and_query"` //小程序启动参数
|
||
|
|
ExpireTime int64 `json:"expireTime" bson:"expire_time"` //过期时间
|
||
|
|
CreateTime int64 `json:"createTime" bson:"create_time"` //创建时间
|
||
|
|
UpdateTime int64 `json:"updateTime" bson:"Update_time"` //更新时间
|
||
|
|
DeleteTime int64 `json:"deleteTime" bson:"Delete_time"` //删除时间-暂不用
|
||
|
|
DebugInfo map[string]interface{} `json:"debugInfo" bson:"debug_info"` //拓展数据
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) Insert(ctx context.Context, info *entity.QrCodeInfo) error {
|
||
|
|
return qrCodeInfoTable.Insert(ctx, convertQrCodeInfoToQrCodeInfoMongo(info))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GenInfo(ctx context.Context, info *entity.QrCodeInfo) error {
|
||
|
|
switch info.Type {
|
||
|
|
case entity.QrCodeTypeReview:
|
||
|
|
count, err := qrCodeInfoTable.Count(ctx, bson.M{"type": info.Type, "app_id": info.AppId, "sequence": info.Sequence})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
//如果存在就不再进行插入
|
||
|
|
if count > 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
case entity.QrCodeTypeRelease:
|
||
|
|
count, err := qrCodeInfoTable.Count(ctx, bson.M{"type": info.Type, "app_id": info.AppId})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
//如果已经存在，更新就好
|
||
|
|
if count > 0 {
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"type": info.Type, "app_id": info.AppId}, bson.M{"$set": bson.M{"update_time": time.Now().UnixNano() / 1e6}})
|
||
|
|
}
|
||
|
|
case entity.QrCodeTypeTrial:
|
||
|
|
count, err := qrCodeInfoTable.Count(ctx, bson.M{"type": info.Type, "app_id": info.AppId})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if count > 0 {
|
||
|
|
upInfo := bson.M{"$set": bson.M{"path_and_query": "", "code_id": info.CodeId, "update_time": utils.GetNowMs()}}
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"type": info.Type, "app_id": info.AppId}, upInfo)
|
||
|
|
}
|
||
|
|
case entity.QrCodeTypeTemporary:
|
||
|
|
//直接插入
|
||
|
|
default:
|
||
|
|
return errors.New("info type err")
|
||
|
|
}
|
||
|
|
//不存在就插入
|
||
|
|
return qrCodeInfoTable.Insert(ctx, convertQrCodeInfoToQrCodeInfoMongo(info))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetInfoByUuid(ctx context.Context, uuid string) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"uuid": uuid}, &qrCodeInfoMongo)
|
||
|
|
log.Infof(utility.InterfaceToJsonString(qrCodeInfoMongo))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetInfoByCodeId(ctx context.Context, codeId string) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"code_id": codeId}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetReviewInfo(ctx context.Context, appId string, seq int) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"type": entity.QrCodeTypeReview, "app_id": appId, "sequence": seq}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetReleaseInfo(ctx context.Context, appId string) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"type": entity.QrCodeTypeRelease, "app_id": appId}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetTrialInfo(ctx context.Context, appId string) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"type": entity.QrCodeTypeTrial, "app_id": appId}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetTemporaryInfo(ctx context.Context, appId string, seq int) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"type": entity.QrCodeTypeTemporary, "app_id": appId}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GetRemoteDebugInfo(ctx context.Context, appId string, seq int) (*entity.QrCodeInfo, error) {
|
||
|
|
qrCodeInfoMongo := QrCodeInfoMongo{}
|
||
|
|
err := qrCodeInfoTable.GetOne(ctx, bson.M{"type": entity.QrCodeTypeRomoteDebug, "app_id": appId, "sequence": seq}, &qrCodeInfoMongo)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo)
|
||
|
|
|
||
|
|
return &result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) UpdateTrialStartParams(ctx context.Context, codeId string, p entity.AppStartParams) error {
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"type": entity.QrCodeTypeTrial, "code_id": codeId}, bson.M{"$set": bson.M{"path_and_query": p.PathAndQuery}})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) UpdateStartParamsByUuid(ctx context.Context, uuid string, p entity.AppStartParams) error {
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"uuid": uuid}, bson.M{"$set": bson.M{"path_and_query": p.PathAndQuery}})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) UpdateApiServer(ctx context.Context, uuid string, apiServer string) error {
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"uuid": uuid}, bson.M{"$set": bson.M{"api_server": apiServer}})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) UpdateInfo(ctx context.Context, uuid string, upInfo map[string]interface{}) error {
|
||
|
|
params := make(map[string]interface{})
|
||
|
|
if t, ok := upInfo["expire_time"]; ok {
|
||
|
|
params["expire_time"] = t
|
||
|
|
}
|
||
|
|
return qrCodeInfoTable.UpdateData(bson.M{"uuid": uuid}, bson.M{"$set": params})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) GenReviewQrCodeInfo(ctx context.Context, info *entity.QrCodeInfo) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QrCodeInfoByMongoRepo) NotFound(err error) bool {
|
||
|
|
return err == mgo.ErrNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
func convertQrCodeInfoToQrCodeInfoMongo(qrCodeInfo *entity.QrCodeInfo) QrCodeInfoMongo {
|
||
|
|
result := QrCodeInfoMongo{}
|
||
|
|
|
||
|
|
result.Type = qrCodeInfo.Type
|
||
|
|
result.Uuid = qrCodeInfo.Uuid
|
||
|
|
result.AppId = qrCodeInfo.AppId
|
||
|
|
result.Sequence = qrCodeInfo.Sequence
|
||
|
|
result.ApiServer = qrCodeInfo.ApiServer
|
||
|
|
result.CodeId = qrCodeInfo.CodeId
|
||
|
|
result.PathAndQuery = qrCodeInfo.StartParams.PathAndQuery
|
||
|
|
result.ExpireTime = qrCodeInfo.ExpireTime
|
||
|
|
result.CreateTime = qrCodeInfo.CreateTime
|
||
|
|
result.UpdateTime = qrCodeInfo.UpdateTime
|
||
|
|
result.DeleteTime = qrCodeInfo.DeleteTime
|
||
|
|
result.DebugInfo = qrCodeInfo.DebugInfo
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func convertQrCodeInfoMongoToQrCodeInfo(qrCodeInfoMongo QrCodeInfoMongo) entity.QrCodeInfo {
|
||
|
|
result := entity.QrCodeInfo{}
|
||
|
|
|
||
|
|
appStartParams := entity.AppStartParams{
|
||
|
|
PathAndQuery: qrCodeInfoMongo.PathAndQuery,
|
||
|
|
}
|
||
|
|
|
||
|
|
result.Type = qrCodeInfoMongo.Type
|
||
|
|
result.Uuid = qrCodeInfoMongo.Uuid
|
||
|
|
result.AppId = qrCodeInfoMongo.AppId
|
||
|
|
result.Sequence = qrCodeInfoMongo.Sequence
|
||
|
|
result.ApiServer = qrCodeInfoMongo.ApiServer
|
||
|
|
result.CodeId = qrCodeInfoMongo.CodeId
|
||
|
|
result.StartParams = appStartParams
|
||
|
|
result.ExpireTime = qrCodeInfoMongo.ExpireTime
|
||
|
|
result.CreateTime = qrCodeInfoMongo.CreateTime
|
||
|
|
result.UpdateTime = qrCodeInfoMongo.UpdateTime
|
||
|
|
result.DeleteTime = qrCodeInfoMongo.DeleteTime
|
||
|
|
result.DebugInfo = qrCodeInfoMongo.DebugInfo
|
||
|
|
return result
|
||
|
|
}
|