diff --git a/app/main/api/desc/admin/admin_feature.api b/app/main/api/desc/admin/admin_feature.api index 03a7212..93cd1ff 100644 --- a/app/main/api/desc/admin/admin_feature.api +++ b/app/main/api/desc/admin/admin_feature.api @@ -8,8 +8,8 @@ info ( // 功能管理接口 @server ( - prefix: /api/v1/admin/feature - group: admin_feature + prefix: /api/v1/admin/feature + group: admin_feature middleware: AdminAuthInterceptor ) service main { @@ -45,27 +45,31 @@ service main { type ( // 创建功能请求 AdminCreateFeatureReq { - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice *float64 `json:"whitelist_price,optional"` // 白名单屏蔽价格（单位：元） + CostPrice *float64 `json:"cost_price,optional"` // 天远API调用成本价（单位：元） } // 创建功能响应 - AdminCreateFeatureResp { - Id string `json:"id"` // 功能ID - } + AdminCreateFeatureResp { + Id string `json:"id"` // 功能ID + } // 更新功能请求 - AdminUpdateFeatureReq { - Id string `path:"id"` // 功能ID - ApiId *string `json:"api_id,optional"` // API标识 - Name *string `json:"name,optional"` // 描述 - } + AdminUpdateFeatureReq { + Id string `path:"id"` // 功能ID + ApiId *string `json:"api_id,optional"` // API标识 + Name *string `json:"name,optional"` // 描述 + WhitelistPrice *float64 `json:"whitelist_price,optional"` // 白名单屏蔽价格（单位：元） + CostPrice *float64 `json:"cost_price,optional"` // 天远API调用成本价（单位：元） + } // 更新功能响应 AdminUpdateFeatureResp { Success bool `json:"success"` // 是否成功 } // 删除功能请求 - AdminDeleteFeatureReq { - Id string `path:"id"` // 功能ID - } + AdminDeleteFeatureReq { + Id string `path:"id"` // 功能ID + } // 删除功能响应 AdminDeleteFeatureResp { Success bool `json:"success"` // 是否成功 @@ -78,12 +82,14 @@ type ( Name *string `form:"name,optional"` // 描述 } // 功能列表项 - FeatureListItem { - Id string `json:"id"` // 功能ID - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 - CreateTime string `json:"create_time"` // 创建时间 - UpdateTime string `json:"update_time"` // 更新时间 + FeatureListItem { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice float64 `json:"whitelist_price"` // 白名单屏蔽价格（单位：元） + CostPrice float64 `json:"cost_price"` // 天远API调用成本价（单位：元） + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 } // 获取功能列表响应 AdminGetFeatureListResp { @@ -91,38 +97,40 @@ type ( Items []FeatureListItem `json:"items"` // 列表数据 } // 获取功能详情请求 - AdminGetFeatureDetailReq { - Id string `path:"id"` // 功能ID - } + AdminGetFeatureDetailReq { + Id string `path:"id"` // 功能ID + } // 获取功能详情响应 - AdminGetFeatureDetailResp { - Id string `json:"id"` // 功能ID - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 - CreateTime string `json:"create_time"` // 创建时间 - UpdateTime string `json:"update_time"` // 更新时间 + AdminGetFeatureDetailResp { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice float64 `json:"whitelist_price"` // 白名单屏蔽价格（单位：元） + CostPrice float64 `json:"cost_price"` // 天远API调用成本价（单位：元） + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 } // 配置功能示例数据请求 - AdminConfigFeatureExampleReq { - FeatureId string `json:"feature_id"` // 功能ID - Data string `json:"data"` // 示例数据JSON - } + AdminConfigFeatureExampleReq { + FeatureId string `json:"feature_id"` // 功能ID + Data string `json:"data"` // 示例数据JSON + } // 配置功能示例数据响应 AdminConfigFeatureExampleResp { Success bool `json:"success"` // 是否成功 } // 查看功能示例数据请求 - AdminGetFeatureExampleReq { - FeatureId string `path:"feature_id"` // 功能ID - } + AdminGetFeatureExampleReq { + FeatureId string `path:"feature_id"` // 功能ID + } // 查看功能示例数据响应 - AdminGetFeatureExampleResp { - Id string `json:"id"` // 示例数据ID - FeatureId string `json:"feature_id"` // 功能ID - ApiId string `json:"api_id"` // API标识 - Data string `json:"data"` // 示例数据JSON - CreateTime string `json:"create_time"` // 创建时间 - UpdateTime string `json:"update_time"` // 更新时间 - } + AdminGetFeatureExampleResp { + Id string `json:"id"` // 示例数据ID + FeatureId string `json:"feature_id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Data string `json:"data"` // 示例数据JSON + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } ) diff --git a/app/main/api/desc/admin/dashboard.api b/app/main/api/desc/admin/dashboard.api new file mode 100644 index 0000000..e2a0bb3 --- /dev/null +++ b/app/main/api/desc/admin/dashboard.api @@ -0,0 +1,93 @@ +syntax = "v1" + +info ( + title: "后台统计面板服务" + desc: "后台统计面板相关接口" + author: "team" + version: "v1" +) + +// ============================================ +// 统计面板接口 +// ============================================ +@server ( + prefix: /api/v1/admin/dashboard + group: admin_dashboard + middleware: AdminAuthInterceptor +) +service main { + // 获取统计面板数据 + @handler AdminGetDashboardStatistics + get /statistics returns (AdminGetDashboardStatisticsResp) +} + +type ( + // 统计面板响应 + AdminGetDashboardStatisticsResp { + // 订单统计 + OrderStats AdminOrderStatistics `json:"order_stats"` + // 营收统计 + RevenueStats AdminRevenueStatistics `json:"revenue_stats"` + // 代理统计 + AgentStats AdminAgentStatistics `json:"agent_stats"` + // 利润统计 + ProfitStats AdminProfitStatistics `json:"profit_stats"` + // 订单趋势（最近7天） + OrderTrend []AdminTrendData `json:"order_trend"` + // 营收趋势（最近7天） + RevenueTrend []AdminTrendData `json:"revenue_trend"` + } + // 订单统计 + AdminOrderStatistics { + TodayCount int64 `json:"today_count"` // 今日订单数 + MonthCount int64 `json:"month_count"` // 当月订单数 + TotalCount int64 `json:"total_count"` // 总订单数 + YesterdayCount int64 `json:"yesterday_count"` // 昨日订单数 + ChangeRate float64 `json:"change_rate"` // 变化率（百分比） + } + // 营收统计 + AdminRevenueStatistics { + TodayAmount float64 `json:"today_amount"` // 今日营收 + MonthAmount float64 `json:"month_amount"` // 当月营收 + TotalAmount float64 `json:"total_amount"` // 总营收 + YesterdayAmount float64 `json:"yesterday_amount"` // 昨日营收 + ChangeRate float64 `json:"change_rate"` // 变化率（百分比） + } + // 代理统计 + AdminAgentStatistics { + TotalCount int64 `json:"total_count"` // 代理总数 + TodayNew int64 `json:"today_new"` // 今日新增 + MonthNew int64 `json:"month_new"` // 当月新增 + } + // 利润统计 + AdminProfitStatistics { + TodayProfit float64 `json:"today_profit"` // 今日利润 + MonthProfit float64 `json:"month_profit"` // 当月利润 + TotalProfit float64 `json:"total_profit"` // 总利润 + TodayProfitRate float64 `json:"today_profit_rate"` // 今日利润率 + MonthProfitRate float64 `json:"month_profit_rate"` // 当月利润率 + TotalProfitRate float64 `json:"total_profit_rate"` // 总利润率 + // 今日明细 + TodayDetail AdminProfitDetail `json:"today_detail"` + // 当月明细 + MonthDetail AdminProfitDetail `json:"month_detail"` + // 总计明细 + TotalDetail AdminProfitDetail `json:"total_detail"` + } + // 利润明细 + AdminProfitDetail { + Revenue float64 `json:"revenue"` // 营收 + Commission float64 `json:"commission"` // 佣金 + Rebate float64 `json:"rebate"` // 返利 + CompanyTax float64 `json:"company_tax"` // 税务成本 + ApiCost float64 `json:"api_cost"` // API调用成本 + TaxIncome float64 `json:"tax_income"` // 提现收税 + Profit float64 `json:"profit"` // 利润 + ProfitRate float64 `json:"profit_rate"` // 利润率 + } + // 趋势数据 + AdminTrendData { + Date string `json:"date"` // 日期（格式：MM-DD） + Value float64 `json:"value"` // 数值 + } +) diff --git a/app/main/api/desc/front/agent.api b/app/main/api/desc/front/agent.api index 63d04c2..6adcc3a 100644 --- a/app/main/api/desc/front/agent.api +++ b/app/main/api/desc/front/agent.api @@ -651,7 +651,7 @@ type ( ProductName string `json:"product_name"` // 产品名称 CreateTime string `json:"create_time"` // 创建时间 QueryState string `json:"query_state"` // 查询状态 - Params map[string]interface{} `json:"params"` // 查询参数（已脱敏） + Params map[string]interface{} `json:"params,optional"` // 查询参数（已脱敏） Price float64 `json:"price"` // 查询价格 } // ============================================ @@ -697,35 +697,29 @@ type ( StatusText string `json:"status_text"` // 状态文本 CreateTime string `json:"create_time"` // 创建时间 } - // 检查模块白名单状态请求 CheckFeatureWhitelistStatusReq { IdCard string `form:"id_card"` // 身份证号 FeatureApiId string `form:"feature_api_id"` // Feature的API标识 - QueryId string `form:"query_id,optional"` // 查询记录ID（可选，用于检查报告数据是否已删除） + QueryId string `form:"query_id,optional"` // 查询记录ID（可选） } - // 检查模块白名单状态响应 CheckFeatureWhitelistStatusResp { IsWhitelisted bool `json:"is_whitelisted"` // 是否在白名单中 - WhitelistPrice float64 `json:"whitelist_price"` // 屏蔽价格（单位：元），如果为0表示不支持下架 + WhitelistPrice float64 `json:"whitelist_price"` // 屏蔽价格（单位：元），0表示不支持下架 FeatureId string `json:"feature_id"` // Feature的UUID - DataDeleted bool `json:"data_deleted"` // 报告数据是否已删除（仅当提供了query_id时有效） + DataDeleted bool `json:"data_deleted"` // 报告数据是否已删除 } - // 下架单个模块请求 OfflineFeatureReq { FeatureApiId string `json:"feature_api_id"` // Feature的API标识 QueryId string `json:"query_id"` // 查询记录ID（Query表的ID，必选） } - // 下架单个模块响应 OfflineFeatureResp { Success bool `json:"success"` // 是否已完成下架 NeedPay bool `json:"need_pay"` // 是否需要发起支付 - Amount float64 `json:"amount"` // 需要支付的金额（单位：元），0表示无需支付 + Amount float64 `json:"amount"` // 需要支付的金额（元），0表示无需支付 } - // 检查订单是否属于当前代理请求 CheckOrderAgentReq { OrderId string `form:"order_id"` // 订单ID } - // 检查订单是否属于当前代理响应 CheckOrderAgentResp { IsAgentOrder bool `json:"is_agent_order"` // 是否是当前代理推广的订单 } diff --git a/app/main/api/desc/main.api b/app/main/api/desc/main.api index 855660b..2773fb0 100644 --- a/app/main/api/desc/main.api +++ b/app/main/api/desc/main.api @@ -24,6 +24,7 @@ import "./admin/platform_user.api" import "./admin/notification.api" import "./admin/admin_product.api" import "./admin/admin_feature.api" +import "./admin/dashboard.api" import "./admin/admin_query.api" import "./admin/admin_agent.api" import "./admin/admin_api.api" diff --git a/app/main/api/internal/handler/admin_dashboard/admingetdashboardstatisticshandler.go b/app/main/api/internal/handler/admin_dashboard/admingetdashboardstatisticshandler.go new file mode 100644 index 0000000..61b4be0 --- /dev/null +++ b/app/main/api/internal/handler/admin_dashboard/admingetdashboardstatisticshandler.go @@ -0,0 +1,17 @@ +package admin_dashboard + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_dashboard" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func AdminGetDashboardStatisticsHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := admin_dashboard.NewAdminGetDashboardStatisticsLogic(r.Context(), svcCtx) + resp, err := l.AdminGetDashboardStatistics() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/routes.go b/app/main/api/internal/handler/routes.go index 63bd362..4dd8855 100644 --- a/app/main/api/internal/handler/routes.go +++ b/app/main/api/internal/handler/routes.go @@ -7,6 +7,7 @@ import ( admin_agent "qnc-server/app/main/api/internal/handler/admin_agent" admin_api "qnc-server/app/main/api/internal/handler/admin_api" admin_auth "qnc-server/app/main/api/internal/handler/admin_auth" + admin_dashboard "qnc-server/app/main/api/internal/handler/admin_dashboard" admin_feature "qnc-server/app/main/api/internal/handler/admin_feature" admin_menu "qnc-server/app/main/api/internal/handler/admin_menu" admin_notification "qnc-server/app/main/api/internal/handler/admin_notification" @@ -178,6 +179,20 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) { rest.WithPrefix("/api/v1/admin/auth"), ) + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodGet, + Path: "/statistics", + Handler: admin_dashboard.AdminGetDashboardStatisticsHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/dashboard"), + ) + server.AddRoutes( rest.WithMiddlewares( []rest.Middleware{serverCtx.AdminAuthInterceptor}, diff --git a/app/main/api/internal/logic/admin_dashboard/admingetdashboardstatisticslogic.go b/app/main/api/internal/logic/admin_dashboard/admingetdashboardstatisticslogic.go new file mode 100644 index 0000000..6052feb --- /dev/null +++ b/app/main/api/internal/logic/admin_dashboard/admingetdashboardstatisticslogic.go @@ -0,0 +1,256 @@ +package admin_dashboard + +import ( + "context" + "time" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "qnc-server/app/main/api/internal/service" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" +) + +type AdminGetDashboardStatisticsLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetDashboardStatisticsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetDashboardStatisticsLogic { + return &AdminGetDashboardStatisticsLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetDashboardStatisticsLogic) AdminGetDashboardStatistics() (resp *types.AdminGetDashboardStatisticsResp, err error) { + loc, _ := time.LoadLocation("Asia/Shanghai") + now := time.Now().In(loc) + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, loc) + todayEnd := todayStart.AddDate(0, 0, 1) + yesterdayStart := todayStart.AddDate(0, 0, -1) + yesterdayEnd := todayStart + monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, loc) + monthEnd := monthStart.AddDate(0, 1, 0) + + orderStats, err := l.calculateOrderStatistics(todayStart, todayEnd, yesterdayStart, yesterdayEnd, monthStart, monthEnd) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算订单统计失败, %v", err) + } + revenueStats, err := l.calculateRevenueStatistics(todayStart, todayEnd, yesterdayStart, yesterdayEnd, monthStart, monthEnd) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算营收统计失败, %v", err) + } + agentStats, err := l.calculateAgentStatistics(todayStart, monthStart) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算代理统计失败, %v", err) + } + profitStats, err := l.calculateProfitStatistics(todayStart, todayEnd, monthStart, monthEnd, revenueStats) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算利润统计失败, %v", err) + } + orderTrend, err := l.calculateOrderTrend(now, loc) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算订单趋势失败, %v", err) + } + revenueTrend, err := l.calculateRevenueTrend(now, loc) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "计算营收趋势失败, %v", err) + } + return &types.AdminGetDashboardStatisticsResp{ + OrderStats: orderStats, + RevenueStats: revenueStats, + AgentStats: agentStats, + ProfitStats: profitStats, + OrderTrend: orderTrend, + RevenueTrend: revenueTrend, + }, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateOrderStatistics(todayStart, todayEnd, yesterdayStart, yesterdayEnd, monthStart, monthEnd time.Time) (types.AdminOrderStatistics, error) { + var stats types.AdminOrderStatistics + todayBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", todayStart, todayEnd) + todayCount, err := l.svcCtx.OrderModel.FindCount(l.ctx, todayBuilder, "id") + if err != nil { + return stats, err + } + stats.TodayCount = todayCount + yesterdayBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", yesterdayStart, yesterdayEnd) + yesterdayCount, _ := l.svcCtx.OrderModel.FindCount(l.ctx, yesterdayBuilder, "id") + stats.YesterdayCount = yesterdayCount + monthBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", monthStart, monthEnd) + monthCount, _ := l.svcCtx.OrderModel.FindCount(l.ctx, monthBuilder, "id") + stats.MonthCount = monthCount + totalBuilder := l.svcCtx.OrderModel.SelectBuilder().Where("status = ?", "paid") + totalCount, _ := l.svcCtx.OrderModel.FindCount(l.ctx, totalBuilder, "id") + stats.TotalCount = totalCount + if stats.YesterdayCount > 0 { + stats.ChangeRate = float64(stats.TodayCount-stats.YesterdayCount) / float64(stats.YesterdayCount) * 100 + } else if stats.TodayCount > 0 { + stats.ChangeRate = 100 + } + return stats, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateRevenueStatistics(todayStart, todayEnd, yesterdayStart, yesterdayEnd, monthStart, monthEnd time.Time) (types.AdminRevenueStatistics, error) { + var stats types.AdminRevenueStatistics + todayBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", todayStart, todayEnd) + todayAmount, _ := l.svcCtx.OrderModel.FindSum(l.ctx, todayBuilder, "amount") + stats.TodayAmount = todayAmount + yesterdayBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", yesterdayStart, yesterdayEnd) + yesterdayAmount, _ := l.svcCtx.OrderModel.FindSum(l.ctx, yesterdayBuilder, "amount") + stats.YesterdayAmount = yesterdayAmount + monthBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", monthStart, monthEnd) + monthAmount, _ := l.svcCtx.OrderModel.FindSum(l.ctx, monthBuilder, "amount") + stats.MonthAmount = monthAmount + totalBuilder := l.svcCtx.OrderModel.SelectBuilder().Where("status = ?", "paid") + totalAmount, _ := l.svcCtx.OrderModel.FindSum(l.ctx, totalBuilder, "amount") + stats.TotalAmount = totalAmount + if stats.YesterdayAmount > 0 { + stats.ChangeRate = (stats.TodayAmount - stats.YesterdayAmount) / stats.YesterdayAmount * 100 + } else if stats.TodayAmount > 0 { + stats.ChangeRate = 100 + } + return stats, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateAgentStatistics(todayStart, monthStart time.Time) (types.AdminAgentStatistics, error) { + var stats types.AdminAgentStatistics + totalCount, _ := l.svcCtx.AgentModel.FindCount(l.ctx, l.svcCtx.AgentModel.SelectBuilder(), "id") + stats.TotalCount = totalCount + todayBuilder := l.svcCtx.AgentModel.SelectBuilder().Where("create_time >= ?", todayStart) + todayNew, _ := l.svcCtx.AgentModel.FindCount(l.ctx, todayBuilder, "id") + stats.TodayNew = todayNew + monthBuilder := l.svcCtx.AgentModel.SelectBuilder().Where("create_time >= ?", monthStart) + monthNew, _ := l.svcCtx.AgentModel.FindCount(l.ctx, monthBuilder, "id") + stats.MonthNew = monthNew + return stats, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateProfitStatistics(todayStart, todayEnd, monthStart, monthEnd time.Time, revenueStats types.AdminRevenueStatistics) (types.AdminProfitStatistics, error) { + var stats types.AdminProfitStatistics + const companyTaxRate = 0.06 + todayRevenue := revenueStats.TodayAmount + todayCommissionBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("del_state = ? AND status != ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 3, todayStart, todayEnd) + todayCommission, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, todayCommissionBuilder, "amount") + todayRebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("del_state = ? AND status != ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 3, todayStart, todayEnd) + todayRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, todayRebateBuilder, "rebate_amount") + todayCompanyTax := todayRevenue * companyTaxRate + todayTaxIncomeBuilder := l.svcCtx.AgentWithdrawalTaxModel.SelectBuilder(). + Where("del_state = ? AND tax_status = ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 2, todayStart, todayEnd) + todayTaxIncome, _ := l.svcCtx.AgentWithdrawalTaxModel.FindSum(l.ctx, todayTaxIncomeBuilder, "tax_amount") + todayApiCost := 0.0 + if l.svcCtx.TianyuanapiCallLogService != nil { + todayApiStats, e := l.svcCtx.TianyuanapiCallLogService.GetStatistics(l.ctx, service.StatisticsFilter{StartDate: todayStart, EndDate: todayEnd}) + if e == nil { + todayApiCost = todayApiStats.TotalCost + } + } + stats.TodayProfit = todayRevenue - todayCommission - todayRebate - todayCompanyTax - todayApiCost + todayTaxIncome + if todayRevenue > 0 { + stats.TodayProfitRate = stats.TodayProfit / todayRevenue * 100 + } + stats.TodayDetail = types.AdminProfitDetail{ + Revenue: todayRevenue, Commission: todayCommission, Rebate: todayRebate, + CompanyTax: todayCompanyTax, ApiCost: todayApiCost, TaxIncome: todayTaxIncome, + Profit: stats.TodayProfit, ProfitRate: stats.TodayProfitRate, + } + monthRevenue := revenueStats.MonthAmount + monthCommissionBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("del_state = ? AND status != ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 3, monthStart, monthEnd) + monthCommission, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, monthCommissionBuilder, "amount") + monthRebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("del_state = ? AND status != ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 3, monthStart, monthEnd) + monthRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, monthRebateBuilder, "rebate_amount") + monthCompanyTax := monthRevenue * companyTaxRate + monthTaxIncomeBuilder := l.svcCtx.AgentWithdrawalTaxModel.SelectBuilder(). + Where("del_state = ? AND tax_status = ? AND create_time >= ? AND create_time < ?", globalkey.DelStateNo, 2, monthStart, monthEnd) + monthTaxIncome, _ := l.svcCtx.AgentWithdrawalTaxModel.FindSum(l.ctx, monthTaxIncomeBuilder, "tax_amount") + monthApiCost := 0.0 + if l.svcCtx.TianyuanapiCallLogService != nil { + monthApiStats, e := l.svcCtx.TianyuanapiCallLogService.GetStatistics(l.ctx, service.StatisticsFilter{StartDate: monthStart, EndDate: monthEnd}) + if e == nil { + monthApiCost = monthApiStats.TotalCost + } + } + stats.MonthProfit = monthRevenue - monthCommission - monthRebate - monthCompanyTax - monthApiCost + monthTaxIncome + if monthRevenue > 0 { + stats.MonthProfitRate = stats.MonthProfit / monthRevenue * 100 + } + stats.MonthDetail = types.AdminProfitDetail{ + Revenue: monthRevenue, Commission: monthCommission, Rebate: monthRebate, + CompanyTax: monthCompanyTax, ApiCost: monthApiCost, TaxIncome: monthTaxIncome, + Profit: stats.MonthProfit, ProfitRate: stats.MonthProfitRate, + } + totalRevenue := revenueStats.TotalAmount + totalCommissionBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder().Where("del_state = ? AND status != ?", globalkey.DelStateNo, 3) + totalCommission, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, totalCommissionBuilder, "amount") + totalRebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder().Where("status != ?", 3) + totalRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, totalRebateBuilder, "rebate_amount") + totalCompanyTax := totalRevenue * companyTaxRate + totalTaxIncomeBuilder := l.svcCtx.AgentWithdrawalTaxModel.SelectBuilder().Where("tax_status = ?", 2) + totalTaxIncome, _ := l.svcCtx.AgentWithdrawalTaxModel.FindSum(l.ctx, totalTaxIncomeBuilder, "tax_amount") + totalApiCost := 0.0 + if l.svcCtx.TianyuanapiCallLogService != nil { + totalApiStats, e := l.svcCtx.TianyuanapiCallLogService.GetStatistics(l.ctx, service.StatisticsFilter{}) + if e == nil { + totalApiCost = totalApiStats.TotalCost + } + } + stats.TotalProfit = totalRevenue - totalCommission - totalRebate - totalCompanyTax - totalApiCost + totalTaxIncome + if totalRevenue > 0 { + stats.TotalProfitRate = stats.TotalProfit / totalRevenue * 100 + } + stats.TotalDetail = types.AdminProfitDetail{ + Revenue: totalRevenue, Commission: totalCommission, Rebate: totalRebate, + CompanyTax: totalCompanyTax, ApiCost: totalApiCost, TaxIncome: totalTaxIncome, + Profit: stats.TotalProfit, ProfitRate: stats.TotalProfitRate, + } + return stats, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateOrderTrend(now time.Time, loc *time.Location) ([]types.AdminTrendData, error) { + var trend []types.AdminTrendData + for i := 6; i >= 0; i-- { + date := now.AddDate(0, 0, -i) + dateStart := time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, loc) + dateEnd := dateStart.AddDate(0, 0, 1) + builder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", dateStart, dateEnd) + count, err := l.svcCtx.OrderModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, err + } + trend = append(trend, types.AdminTrendData{Date: date.Format("01-02"), Value: float64(count)}) + } + return trend, nil +} + +func (l *AdminGetDashboardStatisticsLogic) calculateRevenueTrend(now time.Time, loc *time.Location) ([]types.AdminTrendData, error) { + var trend []types.AdminTrendData + for i := 6; i >= 0; i-- { + date := now.AddDate(0, 0, -i) + dateStart := time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, loc) + dateEnd := dateStart.AddDate(0, 0, 1) + builder := l.svcCtx.OrderModel.SelectBuilder(). + Where("status = ? AND create_time >= ? AND create_time < ?", "paid", dateStart, dateEnd) + amount, err := l.svcCtx.OrderModel.FindSum(l.ctx, builder, "amount") + if err != nil { + return nil, err + } + trend = append(trend, types.AdminTrendData{Date: date.Format("01-02"), Value: amount}) + } + return trend, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go b/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go index 3bfbcd7..9514c65 100644 --- a/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go +++ b/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go @@ -34,6 +34,12 @@ func (l *AdminCreateFeatureLogic) AdminCreateFeature(req *types.AdminCreateFeatu ApiId: req.ApiId, Name: req.Name, } + if req.WhitelistPrice != nil { + data.WhitelistPrice = *req.WhitelistPrice + } + if req.CostPrice != nil { + data.CostPrice = *req.CostPrice + } // 2. 数据库操作 result, err := l.svcCtx.FeatureModel.Insert(l.ctx, nil, data) diff --git a/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go b/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go index 0b56ade..cd65cc4 100644 --- a/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go +++ b/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go @@ -35,11 +35,13 @@ func (l *AdminGetFeatureDetailLogic) AdminGetFeatureDetail(req *types.AdminGetFe // 2. 构建响应 resp = &types.AdminGetFeatureDetailResp{ - Id: record.Id, - ApiId: record.ApiId, - Name: record.Name, - CreateTime: record.CreateTime.Format("2006-01-02 15:04:05"), - UpdateTime: record.UpdateTime.Format("2006-01-02 15:04:05"), + Id: record.Id, + ApiId: record.ApiId, + Name: record.Name, + WhitelistPrice: record.WhitelistPrice, + CostPrice: record.CostPrice, + CreateTime: record.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: record.UpdateTime.Format("2006-01-02 15:04:05"), } return resp, nil diff --git a/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go b/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go index e23dafc..487e1cb 100644 --- a/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go +++ b/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go @@ -49,11 +49,13 @@ func (l *AdminGetFeatureListLogic) AdminGetFeatureList(req *types.AdminGetFeatur items := make([]types.FeatureListItem, 0, len(list)) for _, item := range list { listItem := types.FeatureListItem{ - Id: item.Id, - ApiId: item.ApiId, - Name: item.Name, - CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), - UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), + Id: item.Id, + ApiId: item.ApiId, + Name: item.Name, + WhitelistPrice: item.WhitelistPrice, + CostPrice: item.CostPrice, + CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), } items = append(items, listItem) } diff --git a/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go b/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go index e8ff2e1..9158efa 100644 --- a/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go +++ b/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go @@ -46,6 +46,12 @@ func (l *AdminUpdateFeatureLogic) AdminUpdateFeature(req *types.AdminUpdateFeatu if req.Name != nil && *req.Name != "" { record.Name = *req.Name } + if req.WhitelistPrice != nil { + record.WhitelistPrice = *req.WhitelistPrice + } + if req.CostPrice != nil { + record.CostPrice = *req.CostPrice + } // 4. 执行更新操作 err = l.svcCtx.FeatureModel.UpdateWithVersion(l.ctx, nil, record) diff --git a/app/main/api/internal/logic/agent/checkfeaturewhiteliststatuslogic.go b/app/main/api/internal/logic/agent/checkfeaturewhiteliststatuslogic.go index 4ad3c98..10245cc 100644 --- a/app/main/api/internal/logic/agent/checkfeaturewhiteliststatuslogic.go +++ b/app/main/api/internal/logic/agent/checkfeaturewhiteliststatuslogic.go @@ -2,10 +2,14 @@ package agent import ( "context" + "strings" "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/logx" ) @@ -24,7 +28,46 @@ func NewCheckFeatureWhitelistStatusLogic(ctx context.Context, svcCtx *svc.Servic } func (l *CheckFeatureWhitelistStatusLogic) CheckFeatureWhitelistStatus(req *types.CheckFeatureWhitelistStatusReq) (resp *types.CheckFeatureWhitelistStatusResp, err error) { - // todo: add your logic here and delete this line - - return + if req.IdCard == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("身份证号不能为空"), "") + } + if req.FeatureApiId == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("模块API标识不能为空"), "") + } + mainApiId := req.FeatureApiId + if idx := strings.Index(req.FeatureApiId, "_"); idx > 0 { + mainApiId = req.FeatureApiId[:idx] + } + feature, err := l.svcCtx.FeatureModel.FindOneByApiId(l.ctx, mainApiId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("模块不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询模块信息失败, %v", err) + } + whitelistBuilder := l.svcCtx.UserFeatureWhitelistModel.SelectBuilder(). + Where("id_card = ? AND feature_api_id = ? AND status = ?", req.IdCard, mainApiId, 1) + whitelists, err := l.svcCtx.UserFeatureWhitelistModel.FindAll(l.ctx, whitelistBuilder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询白名单记录失败, %v", err) + } + isWhitelisted := len(whitelists) > 0 + dataDeleted := false + if req.QueryId != "" { + containsFeature, err := l.svcCtx.WhitelistService.CheckQueryDataContainsFeature(l.ctx, req.QueryId, req.FeatureApiId) + if err != nil { + logx.Errorf("检查报告数据是否包含模块失败：%v", err) + dataDeleted = true + } else { + dataDeleted = !containsFeature + } + } else { + dataDeleted = true + } + return &types.CheckFeatureWhitelistStatusResp{ + IsWhitelisted: isWhitelisted, + WhitelistPrice: feature.WhitelistPrice, + FeatureId: feature.Id, + DataDeleted: dataDeleted, + }, nil } diff --git a/app/main/api/internal/logic/agent/checkorderagentlogic.go b/app/main/api/internal/logic/agent/checkorderagentlogic.go index 9bc37de..8698168 100644 --- a/app/main/api/internal/logic/agent/checkorderagentlogic.go +++ b/app/main/api/internal/logic/agent/checkorderagentlogic.go @@ -5,7 +5,11 @@ import ( "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/logx" ) @@ -23,8 +27,28 @@ func NewCheckOrderAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *C } } +// CheckOrderAgent 判断订单是否为当前代理推广的订单（通过 agent_order 关联） func (l *CheckOrderAgentLogic) CheckOrderAgent(req *types.CheckOrderAgentReq) (resp *types.CheckOrderAgentResp, err error) { - // todo: add your logic here and delete this line - - return + if req.OrderId == "" { + return &types.CheckOrderAgentResp{IsAgentOrder: false}, nil + } + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return &types.CheckOrderAgentResp{IsAgentOrder: false}, nil + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + agentOrder, err := l.svcCtx.AgentOrderModel.FindOneByOrderId(l.ctx, req.OrderId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return &types.CheckOrderAgentResp{IsAgentOrder: false}, nil + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询订单关联失败, %v", err) + } + return &types.CheckOrderAgentResp{IsAgentOrder: agentOrder.AgentId == agent.Id}, nil } diff --git a/app/main/api/internal/logic/agent/createwhitelistorderlogic.go b/app/main/api/internal/logic/agent/createwhitelistorderlogic.go index 57182b9..c6aaec5 100644 --- a/app/main/api/internal/logic/agent/createwhitelistorderlogic.go +++ b/app/main/api/internal/logic/agent/createwhitelistorderlogic.go @@ -2,11 +2,18 @@ package agent import ( "context" + "fmt" "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "github.com/google/uuid" + "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" ) type CreateWhitelistOrderLogic struct { @@ -24,7 +31,98 @@ func NewCreateWhitelistOrderLogic(ctx context.Context, svcCtx *svc.ServiceContex } func (l *CreateWhitelistOrderLogic) CreateWhitelistOrder(req *types.CreateWhitelistOrderReq) (resp *types.CreateWhitelistOrderResp, err error) { - // todo: add your logic here and delete this line + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + _, err = l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + if req.IdCard == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("身份证号不能为空"), "") + } + if len(req.FeatureIds) == 0 { + return nil, errors.Wrapf(xerr.NewErrMsg("请至少选择一个模块"), "") + } - return + var totalAmount float64 + var orderItems []struct { + FeatureId string + FeatureApiId string + FeatureName string + Price float64 + } + + for _, featureId := range req.FeatureIds { + feature, err := l.svcCtx.FeatureModel.FindOne(l.ctx, featureId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("模块不存在: %s", featureId)), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询模块信息失败, %v", err) + } + whitelistPrice := feature.WhitelistPrice + if whitelistPrice <= 0 { + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("模块 %s 不支持白名单屏蔽", feature.Name)), "") + } + whitelistBuilder := l.svcCtx.UserFeatureWhitelistModel.SelectBuilder(). + Where("id_card = ? AND feature_id = ?", req.IdCard, featureId) + existing, err := l.svcCtx.UserFeatureWhitelistModel.FindAll(l.ctx, whitelistBuilder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询白名单记录失败, %v", err) + } + for _, item := range existing { + if item.Status == 1 { + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("身份证号 %s 的模块 %s 已经加入白名单", req.IdCard, feature.Name)), "") + } + } + totalAmount += whitelistPrice + orderItems = append(orderItems, struct { + FeatureId string + FeatureApiId string + FeatureName string + Price float64 + }{ + FeatureId: feature.Id, FeatureApiId: feature.ApiId, FeatureName: feature.Name, Price: whitelistPrice, + }) + } + + base := l.svcCtx.AlipayService.GenerateOutTradeNo() + orderNo := "W_" + base + if len(orderNo) > 32 { + orderNo = orderNo[:32] + } + + var orderId string + err = l.svcCtx.WhitelistOrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + order := &model.WhitelistOrder{ + Id: uuid.NewString(), OrderNo: orderNo, UserId: userID, IdCard: req.IdCard, + TotalAmount: totalAmount, Status: 1, + } + _, err := l.svcCtx.WhitelistOrderModel.Insert(ctx, session, order) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建订单失败, %v", err) + } + orderId = order.Id + for _, item := range orderItems { + orderItem := &model.WhitelistOrderItem{ + Id: uuid.NewString(), OrderId: orderId, FeatureId: item.FeatureId, + FeatureApiId: item.FeatureApiId, FeatureName: item.FeatureName, Price: item.Price, + } + if _, err := l.svcCtx.WhitelistOrderItemModel.Insert(ctx, session, orderItem); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建订单明细失败, %v", err) + } + } + return nil + }) + if err != nil { + return nil, err + } + return &types.CreateWhitelistOrderResp{ + OrderId: orderId, OrderNo: orderNo, TotalAmount: totalAmount, + }, nil } diff --git a/app/main/api/internal/logic/agent/getwhitelistfeatureslogic.go b/app/main/api/internal/logic/agent/getwhitelistfeatureslogic.go index 52b067b..d4f7298 100644 --- a/app/main/api/internal/logic/agent/getwhitelistfeatureslogic.go +++ b/app/main/api/internal/logic/agent/getwhitelistfeatureslogic.go @@ -3,10 +3,12 @@ package agent import ( "context" + "github.com/zeromicro/go-zero/core/logx" "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" - "github.com/zeromicro/go-zero/core/logx" + "github.com/pkg/errors" ) type GetWhitelistFeaturesLogic struct { @@ -23,8 +25,21 @@ func NewGetWhitelistFeaturesLogic(ctx context.Context, svcCtx *svc.ServiceContex } } +// GetWhitelistFeatures 返回支持白名单屏蔽的 feature 列表（whitelist_price > 0） func (l *GetWhitelistFeaturesLogic) GetWhitelistFeatures(req *types.GetWhitelistFeaturesReq) (resp *types.GetWhitelistFeaturesResp, err error) { - // todo: add your logic here and delete this line - - return + builder := l.svcCtx.FeatureModel.SelectBuilder().Where("whitelist_price > ?", 0) + list, err := l.svcCtx.FeatureModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询功能列表失败, %v", err) + } + items := make([]types.WhitelistFeatureItem, 0, len(list)) + for _, f := range list { + items = append(items, types.WhitelistFeatureItem{ + FeatureId: f.Id, + FeatureApiId: f.ApiId, + FeatureName: f.Name, + WhitelistPrice: f.WhitelistPrice, + }) + } + return &types.GetWhitelistFeaturesResp{List: items}, nil } diff --git a/app/main/api/internal/logic/agent/getwhitelistlistlogic.go b/app/main/api/internal/logic/agent/getwhitelistlistlogic.go index ddddbdb..cf7e0c4 100644 --- a/app/main/api/internal/logic/agent/getwhitelistlistlogic.go +++ b/app/main/api/internal/logic/agent/getwhitelistlistlogic.go @@ -5,7 +5,10 @@ import ( "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/logx" ) @@ -23,8 +26,53 @@ func NewGetWhitelistListLogic(ctx context.Context, svcCtx *svc.ServiceContext) * } } -func (l *GetWhitelistListLogic) GetWhitelistList(req *types.GetWhitelistListReq) (resp *types.GetWhitelistListResp, err error) { - // todo: add your logic here and delete this line - - return +func statusToText(status int64) string { + switch status { + case 1: + return "生效" + case 2: + return "已失效" + default: + return "" + } +} + +func (l *GetWhitelistListLogic) GetWhitelistList(req *types.GetWhitelistListReq) (resp *types.GetWhitelistListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + if req.Page <= 0 { + req.Page = 1 + } + if req.PageSize <= 0 || req.PageSize > 100 { + req.PageSize = 10 + } + builder := l.svcCtx.UserFeatureWhitelistModel.SelectBuilder().Where("user_id = ?", userID) + if req.IdCard != "" { + builder = builder.Where("id_card = ?", req.IdCard) + } + list, total, err := l.svcCtx.UserFeatureWhitelistModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询白名单列表失败, %v", err) + } + items := make([]types.WhitelistItem, 0, len(list)) + for _, r := range list { + featureName := r.FeatureApiId + if f, e := l.svcCtx.FeatureModel.FindOne(l.ctx, r.FeatureId); e == nil { + featureName = f.Name + } + items = append(items, types.WhitelistItem{ + Id: r.Id, + IdCard: r.IdCard, + FeatureId: r.FeatureId, + FeatureApiId: r.FeatureApiId, + FeatureName: featureName, + Amount: r.Amount, + Status: r.Status, + StatusText: statusToText(r.Status), + CreateTime: r.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + return &types.GetWhitelistListResp{Total: total, List: items}, nil } diff --git a/app/main/api/internal/logic/agent/offlinefeaturelogic.go b/app/main/api/internal/logic/agent/offlinefeaturelogic.go index 86d72f9..aa06ac8 100644 --- a/app/main/api/internal/logic/agent/offlinefeaturelogic.go +++ b/app/main/api/internal/logic/agent/offlinefeaturelogic.go @@ -2,10 +2,14 @@ package agent import ( "context" + "encoding/json" "qnc-server/app/main/api/internal/svc" "qnc-server/app/main/api/internal/types" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "github.com/pkg/errors" "github.com/zeromicro/go-zero/core/logx" ) @@ -24,7 +28,44 @@ func NewOfflineFeatureLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Of } func (l *OfflineFeatureLogic) OfflineFeature(req *types.OfflineFeatureReq) (resp *types.OfflineFeatureResp, err error) { - // todo: add your logic here and delete this line - - return + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + if req.FeatureApiId == "" || req.QueryId == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("模块标识和查询记录ID不能为空"), "") + } + // 从 Query 获取 id_card + query, err := l.svcCtx.QueryModel.FindOne(l.ctx, req.QueryId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询记录不存在或查询失败, %v", err) + } + idCard := "" + if query.QueryParams != "" { + var params map[string]interface{} + if e := json.Unmarshal([]byte(query.QueryParams), ¶ms); e == nil { + if v, ok := params["id_card"]; ok { + if s, ok := v.(string); ok { + idCard = s + } + } + } + } + if idCard == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("无法从查询记录中获取身份证号"), "") + } + needPay, amount, whitelistCreated, err := l.svcCtx.WhitelistService.ProcessOfflineFeature(l.ctx, nil, idCard, req.FeatureApiId, userID, req.QueryId) + if err != nil { + return nil, err + } + if whitelistCreated && !needPay { + if err := l.svcCtx.WhitelistService.DeleteFeatureFromQueryData(l.ctx, nil, req.QueryId, req.FeatureApiId); err != nil { + logx.Errorf("从报告数据中删除模块失败: %v", err) + } + return &types.OfflineFeatureResp{Success: true, NeedPay: false, Amount: 0}, nil + } + if needPay { + return &types.OfflineFeatureResp{Success: false, NeedPay: true, Amount: amount}, nil + } + return &types.OfflineFeatureResp{Success: true, NeedPay: false, Amount: 0}, nil } diff --git a/app/main/api/internal/logic/query/query_common.go b/app/main/api/internal/logic/query/query_common.go index 2f8350b..fe0102e 100644 --- a/app/main/api/internal/logic/query/query_common.go +++ b/app/main/api/internal/logic/query/query_common.go @@ -115,6 +115,8 @@ func BuildEncryptedQuery(ctx context.Context, svcCtx *svc.ServiceContext, queryM if err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结构体复制失败, %v", err) } + // 显式写入查询记录 ID，供前端报告页「模块下架」等白名单能力使用 + query.Id = queryModel.Id product, err := svcCtx.ProductModel.FindOne(ctx, queryModel.ProductId) if err != nil { return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取商品信息失败, %v", err) diff --git a/app/main/api/internal/service/apirequestService.go b/app/main/api/internal/service/apirequestService.go index e1b5a07..4b0c504 100644 --- a/app/main/api/internal/service/apirequestService.go +++ b/app/main/api/internal/service/apirequestService.go @@ -33,19 +33,33 @@ func generateAuthDateRange() string { } type ApiRequestService struct { - config config.Config - featureModel model.FeatureModel - productFeatureModel model.ProductFeatureModel - tianyuanapi *tianyuanapi.Client + config config.Config + featureModel model.FeatureModel + productFeatureModel model.ProductFeatureModel + userFeatureWhitelistModel model.UserFeatureWhitelistModel + whitelistService *WhitelistService + tianyuanapi *tianyuanapi.Client + tianyuanapiCallLogService *TianyuanapiCallLogService } -// NewApiRequestService 是一个构造函数，用于初始化 ApiRequestService -func NewApiRequestService(c config.Config, featureModel model.FeatureModel, productFeatureModel model.ProductFeatureModel, tianyuanapi *tianyuanapi.Client) *ApiRequestService { +// NewApiRequestService 构造函数 +func NewApiRequestService( + c config.Config, + featureModel model.FeatureModel, + productFeatureModel model.ProductFeatureModel, + userFeatureWhitelistModel model.UserFeatureWhitelistModel, + tianyuanapi *tianyuanapi.Client, + tianyuanapiCallLogService *TianyuanapiCallLogService, + whitelistService *WhitelistService, +) *ApiRequestService { return &ApiRequestService{ - config: c, - featureModel: featureModel, - productFeatureModel: productFeatureModel, - tianyuanapi: tianyuanapi, + config: c, + featureModel: featureModel, + productFeatureModel: productFeatureModel, + userFeatureWhitelistModel: userFeatureWhitelistModel, + tianyuanapi: tianyuanapi, + tianyuanapiCallLogService: tianyuanapiCallLogService, + whitelistService: whitelistService, } } @@ -61,23 +75,33 @@ type APIResponseData struct { func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]byte, error) { var ctx, cancel = context.WithCancel(context.Background()) defer cancel() - build := a.productFeatureModel.SelectBuilder().Where(squirrel.Eq{ - "product_id": productID, - }) + + // 白名单：已下架模块直接返回占位成功，不调用天远 API + idCard := gjson.GetBytes(params, "id_card").String() + var whitelistedFeatureApiIds map[string]bool + if a.whitelistService != nil { + whitelistedFeatureApiIds, _ = a.whitelistService.GetWhitelistedFeatureApisByIdCard(ctx, idCard) + } else { + whitelistedFeatureApiIds = make(map[string]bool) + } + + build := a.productFeatureModel.SelectBuilder().Where(squirrel.Eq{ + "product_id": productID, + }) productFeatureList, findProductFeatureErr := a.productFeatureModel.FindAll(ctx, build, "") if findProductFeatureErr != nil { return nil, findProductFeatureErr } - var featureIDs []string - isImportantMap := make(map[string]int64, len(productFeatureList)) + var featureIDs []string + isImportantMap := make(map[string]int64, len(productFeatureList)) for _, pf := range productFeatureList { - featureIDs = append(featureIDs, pf.FeatureId) - isImportantMap[pf.FeatureId] = pf.IsImportant + featureIDs = append(featureIDs, pf.FeatureId) + isImportantMap[pf.FeatureId] = pf.IsImportant } if len(featureIDs) == 0 { return nil, errors.New("featureIDs 是空的") } - builder := a.featureModel.SelectBuilder().Where(squirrel.Eq{"id": featureIDs}) + builder := a.featureModel.SelectBuilder().Where(squirrel.Eq{"id": featureIDs}) featureList, findFeatureErr := a.featureModel.FindAll(ctx, builder, "") if findFeatureErr != nil { return nil, findFeatureErr @@ -109,12 +133,19 @@ func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([] Success: false, } timestamp := time.Now().Format("2006-01-02 15:04:05") + // 白名单：已下架模块直接返回占位成功 + if whitelistedFeatureApiIds[feature.ApiId] { + result.Success = true + result.Timestamp = timestamp + result.Data = []byte("null") + resultsCh <- result + return + } var ( resp json.RawMessage preprocessErr error ) - // 若 isImportantMap[feature.ID] == 1，则表示需要在出错时重试 - isImportant := isImportantMap[feature.Id] == 1 + isImportant := isImportantMap[feature.Id] == 1 tryCount := 0 for { tryCount++ diff --git a/app/main/api/internal/service/tianyuanapiCallLogService.go b/app/main/api/internal/service/tianyuanapiCallLogService.go new file mode 100644 index 0000000..907531b --- /dev/null +++ b/app/main/api/internal/service/tianyuanapiCallLogService.go @@ -0,0 +1,190 @@ +package service + +import ( + "context" + "database/sql" + "encoding/json" + "fmt" + "time" + + "github.com/Masterminds/squirrel" + "github.com/zeromicro/go-zero/core/logx" + "qnc-server/app/main/model" +) + +// TianyuanapiCallLogService 天元API调用记录服务 +type TianyuanapiCallLogService struct { + tianyuanapiCallLogModel model.TianyuanapiCallLogModel + featureModel model.FeatureModel +} + +// NewTianyuanapiCallLogService 创建天元API调用记录服务 +func NewTianyuanapiCallLogService( + tianyuanapiCallLogModel model.TianyuanapiCallLogModel, + featureModel model.FeatureModel, +) *TianyuanapiCallLogService { + return &TianyuanapiCallLogService{ + tianyuanapiCallLogModel: tianyuanapiCallLogModel, + featureModel: featureModel, + } +} + +// CallLogOptions 调用记录选项 +type CallLogOptions struct { + FeatureID string // 功能ID + ApiID string // API标识（如：YYSYBE08） + OrderID string // 订单ID（可选） + QueryID string // 查询ID（可选） + CallStatus int64 // 调用状态：0=失败，1=成功 + ResponseTime int64 // 响应耗时（毫秒） + ErrorCode string // 错误码（失败时） + ErrorMessage string // 错误信息（失败时） + RequestParams interface{} // 请求参数（可选） + ResponseData interface{} // 响应数据（可选） + TransactionID string // 天元API流水号 +} + +// RecordCall 记录天元API调用 +func (s *TianyuanapiCallLogService) RecordCall(ctx context.Context, opts CallLogOptions) error { + // 1. 获取feature的成本价 + costPrice := 0.00 + if opts.CallStatus == 1 { // 只有成功才计算成本 + if opts.FeatureID == "" { + logx.Infof("记录API调用时feature_id为空，api_id=%s，无法获取成本价", opts.ApiID) + } else { + feature, err := s.featureModel.FindOne(ctx, opts.FeatureID) + if err == nil { + costPrice = feature.CostPrice + logx.Infof("记录API调用 - feature_id=%s, api_id=%s, cost_price=%f", opts.FeatureID, opts.ApiID, costPrice) + } else { + logx.Errorf("查询feature成本价失败，feature_id=%s, api_id=%s, err=%v", opts.FeatureID, opts.ApiID, err) + } + } + } + + // 2. 转换参数和响应为JSON字符串 + var requestParamsStr, responseDataStr *string + if opts.RequestParams != nil { + if bytes, err := json.Marshal(opts.RequestParams); err == nil { + jsonStr := string(bytes) + requestParamsStr = &jsonStr + } + } + if opts.ResponseData != nil { + if bytes, err := json.Marshal(opts.ResponseData); err == nil { + jsonStr := string(bytes) + if len(jsonStr) > 1000 { + jsonStr = jsonStr[:1000] + "...[truncated]" + } + responseDataStr = &jsonStr + } + } + + // 3. 构建调用记录 + callTime := time.Now() + deleteTime := sql.NullTime{} + callLog := &model.TianyuanapiCallLog{ + FeatureId: opts.FeatureID, + ApiId: opts.ApiID, + OrderId: sql.NullString{}, + QueryId: sql.NullString{}, + CallStatus: opts.CallStatus, + CallTime: callTime, + ResponseTime: sql.NullInt64{}, + CostPrice: costPrice, + ErrorCode: sql.NullString{}, + ErrorMessage: sql.NullString{}, + RequestParams: sql.NullString{}, + ResponseData: sql.NullString{}, + TransactionId: sql.NullString{}, + CreateTime: callTime, + UpdateTime: callTime, + DeleteTime: deleteTime, + DelState: 0, + Version: 0, + } + + if opts.OrderID != "" { + callLog.OrderId = sql.NullString{String: opts.OrderID, Valid: true} + } + if opts.QueryID != "" { + callLog.QueryId = sql.NullString{String: opts.QueryID, Valid: true} + } + if opts.ResponseTime > 0 { + callLog.ResponseTime = sql.NullInt64{Int64: opts.ResponseTime, Valid: true} + } + if opts.ErrorCode != "" { + callLog.ErrorCode = sql.NullString{String: opts.ErrorCode, Valid: true} + } + if opts.ErrorMessage != "" { + callLog.ErrorMessage = sql.NullString{String: opts.ErrorMessage, Valid: true} + } + if requestParamsStr != nil { + callLog.RequestParams = sql.NullString{String: *requestParamsStr, Valid: true} + } + if responseDataStr != nil { + callLog.ResponseData = sql.NullString{String: *responseDataStr, Valid: true} + } + if opts.TransactionID != "" { + callLog.TransactionId = sql.NullString{String: opts.TransactionID, Valid: true} + } + + _, err := s.tianyuanapiCallLogModel.Insert(ctx, nil, callLog) + if err != nil { + logx.Errorf("插入天元API调用记录失败，feature_id=%s, api_id=%s, err=%v", opts.FeatureID, opts.ApiID, err) + return fmt.Errorf("插入调用记录失败: %w", err) + } + return nil +} + +// StatisticsFilter 统计过滤条件 +type StatisticsFilter struct { + FeatureID string + ApiID string + StartDate time.Time + EndDate time.Time +} + +// Statistics 统计信息 +type Statistics struct { + TotalCalls int64 + SuccessCalls int64 + FailedCalls int64 + TotalCost float64 +} + +// GetStatistics 获取统计信息 +func (s *TianyuanapiCallLogService) GetStatistics(ctx context.Context, filter StatisticsFilter) (*Statistics, error) { + builder := s.tianyuanapiCallLogModel.SelectBuilder() + if filter.FeatureID != "" { + builder = builder.Where(squirrel.Eq{"feature_id": filter.FeatureID}) + } + if filter.ApiID != "" { + builder = builder.Where(squirrel.Eq{"api_id": filter.ApiID}) + } + if !filter.StartDate.IsZero() { + builder = builder.Where(squirrel.GtOrEq{"call_time": filter.StartDate}) + } + if !filter.EndDate.IsZero() { + builder = builder.Where(squirrel.Lt{"call_time": filter.EndDate}) + } + totalCalls, err := s.tianyuanapiCallLogModel.FindCount(ctx, builder, "id") + if err != nil { + return nil, fmt.Errorf("统计总调用次数失败: %w", err) + } + successBuilder := builder.Where(squirrel.Eq{"call_status": 1}) + successCalls, err := s.tianyuanapiCallLogModel.FindCount(ctx, successBuilder, "id") + if err != nil { + return nil, fmt.Errorf("统计成功次数失败: %w", err) + } + totalCost, err := s.tianyuanapiCallLogModel.FindSum(ctx, successBuilder, "cost_price") + if err != nil { + return nil, fmt.Errorf("统计总成本失败: %w", err) + } + return &Statistics{ + TotalCalls: totalCalls, + SuccessCalls: successCalls, + FailedCalls: totalCalls - successCalls, + TotalCost: totalCost, + }, nil +} diff --git a/app/main/api/internal/service/whitelistService.go b/app/main/api/internal/service/whitelistService.go new file mode 100644 index 0000000..78645fd --- /dev/null +++ b/app/main/api/internal/service/whitelistService.go @@ -0,0 +1,450 @@ +package service + +import ( + "context" + "database/sql" + "encoding/hex" + "encoding/json" + "strings" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + "qnc-server/pkg/lzkit/lzUtils" +) + +// WhitelistService 白名单领域服务 +type WhitelistService struct { + config config.Config + UserFeatureWhitelistModel model.UserFeatureWhitelistModel + WhitelistOrderModel model.WhitelistOrderModel + WhitelistOrderItemModel model.WhitelistOrderItemModel + QueryModel model.QueryModel + FeatureModel model.FeatureModel +} + +// NewWhitelistService 创建白名单服务 +func NewWhitelistService( + c config.Config, + userFeatureWhitelistModel model.UserFeatureWhitelistModel, + whitelistOrderModel model.WhitelistOrderModel, + whitelistOrderItemModel model.WhitelistOrderItemModel, + queryModel model.QueryModel, + featureModel model.FeatureModel, +) *WhitelistService { + return &WhitelistService{ + config: c, + UserFeatureWhitelistModel: userFeatureWhitelistModel, + WhitelistOrderModel: whitelistOrderModel, + WhitelistOrderItemModel: whitelistOrderItemModel, + QueryModel: queryModel, + FeatureModel: featureModel, + } +} + +// EnsureFreeWhitelist 免费下架：如果还没有生效白名单，则创建一条免费白名单记录 +func (s *WhitelistService) EnsureFreeWhitelist( + ctx context.Context, + session sqlx.Session, + idCard string, + feature *model.Feature, + userId string, + orderId string, +) error { + builder := s.UserFeatureWhitelistModel.SelectBuilder(). + Where("id_card = ? AND feature_id = ?", idCard, feature.Id) + records, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "") + if err != nil { + return errors.Wrap(err, "查询白名单记录失败") + } + for _, r := range records { + if r.Status == 1 { + return nil + } + } + wl := &model.UserFeatureWhitelist{ + Id: uuid.NewString(), + IdCard: idCard, + FeatureId: feature.Id, + FeatureApiId: feature.ApiId, + UserId: userId, + OrderId: lzUtils.StringToNullString(orderId), + WhitelistOrderId: lzUtils.StringToNullString(""), + Amount: 0, + Status: 1, + } + _, err = s.UserFeatureWhitelistModel.Insert(ctx, session, wl) + if err != nil { + return errors.Wrap(err, "创建免费白名单记录失败") + } + return nil +} + +// CreateWhitelistByPaidOrder 根据已支付的白名单订单，创建对应的白名单记录 +func (s *WhitelistService) CreateWhitelistByPaidOrder( + ctx context.Context, + session sqlx.Session, + order *model.Order, + whitelistOrder *model.WhitelistOrder, +) error { + if whitelistOrder.Status != 2 { + return nil + } + itemBuilder := s.WhitelistOrderItemModel.SelectBuilder(). + Where("order_id = ?", whitelistOrder.Id) + items, err := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "") + if err != nil { + return errors.Wrap(err, "查询白名单订单明细失败") + } + for _, item := range items { + wl := &model.UserFeatureWhitelist{ + Id: uuid.NewString(), + IdCard: whitelistOrder.IdCard, + FeatureId: item.FeatureId, + FeatureApiId: item.FeatureApiId, + UserId: whitelistOrder.UserId, + OrderId: lzUtils.StringToNullString(order.Id), + WhitelistOrderId: lzUtils.StringToNullString(whitelistOrder.Id), + Amount: item.Price, + Status: 1, + } + if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil { + return errors.Wrap(err, "创建白名单记录失败") + } + } + return nil +} + +// GetWhitelistedFeatureApisByIdCard 获取某个身份证号已下架的 feature_api_id 集合 +func (s *WhitelistService) GetWhitelistedFeatureApisByIdCard( + ctx context.Context, + idCard string, +) (map[string]bool, error) { + result := make(map[string]bool) + if s == nil || idCard == "" { + return result, nil + } + builder := s.UserFeatureWhitelistModel.SelectBuilder(). + Where("id_card = ? AND status = ?", idCard, 1) + list, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "") + if err != nil { + return nil, errors.Wrap(err, "查询白名单失败") + } + for _, wl := range list { + result[wl.FeatureApiId] = true + } + return result, nil +} + +// CheckWhitelistExists 检查指定身份证号和模块是否已有生效的白名单记录 +func (s *WhitelistService) CheckWhitelistExists( + ctx context.Context, + idCard string, + featureId string, +) (bool, error) { + if idCard == "" || featureId == "" { + return false, nil + } + builder := s.UserFeatureWhitelistModel.SelectBuilder(). + Where("id_card = ? AND feature_id = ? AND status = ?", idCard, featureId, 1) + list, err := s.UserFeatureWhitelistModel.FindAll(ctx, builder, "") + if err != nil { + return false, errors.Wrap(err, "查询白名单记录失败") + } + return len(list) > 0, nil +} + +// ProcessOfflineFeature 统一下架处理 +func (s *WhitelistService) ProcessOfflineFeature( + ctx context.Context, + session sqlx.Session, + idCard string, + featureApiId string, + userId string, + orderId string, +) (needPay bool, amount float64, whitelistCreated bool, err error) { + mainApiId := s.extractMainApiId(featureApiId) + feature, err := s.getFeatureByApiId(ctx, mainApiId) + if err != nil { + return false, 0, false, err + } + if feature.WhitelistPrice < 0 { + return false, 0, false, errors.Wrapf(xerr.NewErrMsg("该模块不支持下架"), "") + } + exists, err := s.CheckWhitelistExists(ctx, idCard, feature.Id) + if err != nil { + return false, 0, false, err + } + if exists { + return false, 0, true, nil + } + price := feature.WhitelistPrice + if price <= 0 { + if err := s.EnsureFreeWhitelist(ctx, session, idCard, feature, userId, orderId); err != nil { + return false, 0, false, err + } + return false, 0, true, nil + } + paidOrderId, err := s.findPaidWhitelistOrder(ctx, userId, idCard, feature.Id) + if err != nil { + return false, 0, false, err + } + if paidOrderId != "" { + if err := s.createWhitelistFromPaidOrder(ctx, session, idCard, feature, userId, orderId, paidOrderId, price); err != nil { + return false, 0, false, err + } + return false, price, true, nil + } + return true, price, false, nil +} + +func (s *WhitelistService) extractMainApiId(featureApiId string) string { + if idx := strings.Index(featureApiId, "_"); idx > 0 { + return featureApiId[:idx] + } + return featureApiId +} + +func (s *WhitelistService) getFeatureByApiId(ctx context.Context, apiId string) (*model.Feature, error) { + feature, err := s.FeatureModel.FindOneByApiId(ctx, apiId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrap(err, "模块不存在") + } + return nil, errors.Wrap(err, "查询模块信息失败") + } + return feature, nil +} + +func (s *WhitelistService) findPaidWhitelistOrder( + ctx context.Context, + userId string, + idCard string, + featureId string, +) (string, error) { + orderBuilder := s.WhitelistOrderModel.SelectBuilder(). + Where("user_id = ? AND id_card = ? AND status = ?", userId, idCard, 2) + orders, err := s.WhitelistOrderModel.FindAll(ctx, orderBuilder, "") + if err != nil { + return "", errors.Wrap(err, "查询白名单订单失败") + } + for _, order := range orders { + itemBuilder := s.WhitelistOrderItemModel.SelectBuilder(). + Where("order_id = ? AND feature_id = ?", order.Id, featureId) + items, itemErr := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "") + if itemErr != nil { + return "", errors.Wrap(itemErr, "查询白名单订单明细失败") + } + if len(items) > 0 { + return order.Id, nil + } + } + return "", nil +} + +func (s *WhitelistService) createWhitelistFromPaidOrder( + ctx context.Context, + session sqlx.Session, + idCard string, + feature *model.Feature, + userId string, + orderId string, + paidOrderId string, + price float64, +) error { + wl := &model.UserFeatureWhitelist{ + Id: uuid.NewString(), + IdCard: idCard, + FeatureId: feature.Id, + FeatureApiId: feature.ApiId, + UserId: userId, + OrderId: lzUtils.StringToNullString(orderId), + WhitelistOrderId: lzUtils.StringToNullString(paidOrderId), + Amount: price, + Status: 1, + } + if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil { + return errors.Wrap(err, "根据已支付订单创建白名单记录失败") + } + return nil +} + +// DeleteFeatureFromQueryData 从报告数据中删除指定模块的数据 +func (s *WhitelistService) DeleteFeatureFromQueryData( + ctx context.Context, + session sqlx.Session, + queryId string, + featureApiId string, +) error { + queryModel, err := s.getQueryModel(ctx, queryId) + if err != nil { + return err + } + if queryModel == nil { + return nil + } + mainApiId := s.extractMainApiId(featureApiId) + dataArray, key, err := s.decryptQueryData(queryModel) + if err != nil { + return err + } + modifiedArray, hasModified := s.clearFeatureData(dataArray, mainApiId) + if !hasModified { + logx.Infof("删除报告数据：查询记录 %s 中未找到模块 %s 的数据，跳过删除", queryId, featureApiId) + return nil + } + if err := s.updateQueryData(ctx, session, queryModel, modifiedArray, key); err != nil { + return err + } + return nil +} + +func (s *WhitelistService) getQueryModel(ctx context.Context, queryId string) (*model.Query, error) { + queryModel, err := s.QueryModel.FindOne(ctx, queryId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, nil + } + return nil, errors.Wrap(err, "查询报告记录失败") + } + if !queryModel.QueryData.Valid || queryModel.QueryData.String == "" { + return nil, nil + } + return queryModel, nil +} + +func (s *WhitelistService) decryptQueryData(queryModel *model.Query) ([]map[string]interface{}, []byte, error) { + secretKey := s.config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, nil, errors.Wrap(decodeErr, "获取AES密钥失败") + } + decryptedData, decryptErr := crypto.AesDecrypt(queryModel.QueryData.String, key) + if decryptErr != nil { + return nil, nil, errors.Wrap(decryptErr, "解密报告数据失败") + } + var dataArray []map[string]interface{} + if unmarshalErr := json.Unmarshal(decryptedData, &dataArray); unmarshalErr != nil { + return nil, nil, errors.Wrap(unmarshalErr, "解析报告数据失败") + } + return dataArray, key, nil +} + +func (s *WhitelistService) clearFeatureData(dataArray []map[string]interface{}, mainApiId string) ([]map[string]interface{}, bool) { + modifiedArray := make([]map[string]interface{}, 0, len(dataArray)) + hasModified := false + for _, item := range dataArray { + newItem := make(map[string]interface{}) + for k, v := range item { + newItem[k] = v + } + apiID, ok := item["apiID"].(string) + if !ok { + modifiedArray = append(modifiedArray, newItem) + continue + } + if s.extractMainApiId(apiID) == mainApiId { + newItem["data"] = nil + hasModified = true + } + modifiedArray = append(modifiedArray, newItem) + } + return modifiedArray, hasModified +} + +func (s *WhitelistService) updateQueryData( + ctx context.Context, + session sqlx.Session, + queryModel *model.Query, + filteredArray []map[string]interface{}, + key []byte, +) error { + filteredBytes, marshalErr := json.Marshal(filteredArray) + if marshalErr != nil { + return errors.Wrap(marshalErr, "序列化过滤后的报告数据失败") + } + encryptedData, encryptErr := crypto.AesEncrypt(filteredBytes, key) + if encryptErr != nil { + return errors.Wrap(encryptErr, "加密过滤后的报告数据失败") + } + queryModel.QueryData = sql.NullString{String: encryptedData, Valid: true} + if updateErr := s.QueryModel.UpdateWithVersion(ctx, session, queryModel); updateErr != nil { + return errors.Wrap(updateErr, "更新报告数据失败") + } + return nil +} + +// CheckQueryDataContainsFeature 检查报告数据中是否包含指定的模块 +func (s *WhitelistService) CheckQueryDataContainsFeature( + ctx context.Context, + queryId string, + featureApiId string, +) (bool, error) { + queryModel, err := s.getQueryModel(ctx, queryId) + if err != nil { + return false, err + } + if queryModel == nil { + return false, nil + } + mainApiId := s.extractMainApiId(featureApiId) + dataArray, _, err := s.decryptQueryData(queryModel) + if err != nil { + return false, err + } + for _, item := range dataArray { + apiID, ok := item["apiID"].(string) + if !ok { + continue + } + if s.extractMainApiId(apiID) == mainApiId { + dataValue, exists := item["data"] + if !exists || dataValue == nil { + return false, nil + } + return true, nil + } + } + return false, nil +} + +// ProcessPaidWhitelistOrder 处理已支付的白名单订单 +func (s *WhitelistService) ProcessPaidWhitelistOrder( + ctx context.Context, + session sqlx.Session, + order *model.Order, + whitelistOrder *model.WhitelistOrder, +) error { + if whitelistOrder.Status != 2 { + return nil + } + itemBuilder := s.WhitelistOrderItemModel.SelectBuilder(). + Where("order_id = ?", whitelistOrder.Id) + items, err := s.WhitelistOrderItemModel.FindAll(ctx, itemBuilder, "") + if err != nil { + return errors.Wrap(err, "查询白名单订单明细失败") + } + for _, item := range items { + wl := &model.UserFeatureWhitelist{ + Id: uuid.NewString(), + IdCard: whitelistOrder.IdCard, + FeatureId: item.FeatureId, + FeatureApiId: item.FeatureApiId, + UserId: whitelistOrder.UserId, + OrderId: lzUtils.StringToNullString(""), + WhitelistOrderId: lzUtils.StringToNullString(whitelistOrder.Id), + Amount: item.Price, + Status: 1, + } + if _, err := s.UserFeatureWhitelistModel.Insert(ctx, session, wl); err != nil { + return errors.Wrap(err, "创建白名单记录失败") + } + logx.Infof("白名单订单支付成功：订单 %s，模块 %s，已创建白名单记录", whitelistOrder.OrderNo, item.FeatureApiId) + } + return nil +} diff --git a/app/main/api/internal/svc/servicecontext.go b/app/main/api/internal/svc/servicecontext.go index 68bc542..436b6e0 100644 --- a/app/main/api/internal/svc/servicecontext.go +++ b/app/main/api/internal/svc/servicecontext.go @@ -34,6 +34,11 @@ type ServiceContext struct { FeatureModel model.FeatureModel ProductFeatureModel model.ProductFeatureModel + // 白名单相关模型 + UserFeatureWhitelistModel model.UserFeatureWhitelistModel + WhitelistOrderModel model.WhitelistOrderModel + WhitelistOrderItemModel model.WhitelistOrderItemModel + // 订单相关模型 OrderModel model.OrderModel OrderRefundModel model.OrderRefundModel @@ -77,11 +82,15 @@ type ServiceContext struct { GlobalNotificationsModel model.GlobalNotificationsModel AuthorizationDocumentModel model.AuthorizationDocumentModel + // 第三方服务 + TianyuanapiCallLogService *service.TianyuanapiCallLogService + // 服务 AlipayService *service.AliPayService WechatPayService *service.WechatPayService ApplePayService *service.ApplePayService ApiRequestService *service.ApiRequestService + WhitelistService *service.WhitelistService AsynqServer *asynq.Server AsynqService *service.AsynqService VerificationService *service.VerificationService @@ -152,10 +161,16 @@ func NewServiceContext(c config.Config) *ServiceContext { adminDictDataModel := model.NewAdminDictDataModel(db, cacheConf) adminDictTypeModel := model.NewAdminDictTypeModel(db, cacheConf) + // ============================== 白名单相关模型 ============================== + userFeatureWhitelistModel := model.NewUserFeatureWhitelistModel(db, cacheConf) + whitelistOrderModel := model.NewWhitelistOrderModel(db, cacheConf) + whitelistOrderItemModel := model.NewWhitelistOrderItemModel(db, cacheConf) + // ============================== 其他模型 ============================== exampleModel := model.NewExampleModel(db, cacheConf) globalNotificationsModel := model.NewGlobalNotificationsModel(db, cacheConf) authorizationDocumentModel := model.NewAuthorizationDocumentModel(db, cacheConf) + tianyuanapiCallLogModel := model.NewTianyuanapiCallLogModel(db, cacheConf) // ============================== 第三方服务初始化 ============================== tianyuanapi, err := tianyuanapi.NewClient(tianyuanapi.Config{ @@ -172,7 +187,9 @@ func NewServiceContext(c config.Config) *ServiceContext { alipayService := service.NewAliPayService(c) wechatPayService := service.NewWechatPayService(c, userAuthModel, service.InitTypeWxPayPubKey) applePayService := service.NewApplePayService(c) - apiRequestService := service.NewApiRequestService(c, featureModel, productFeatureModel, tianyuanapi) + tianyuanapiCallLogService := service.NewTianyuanapiCallLogService(tianyuanapiCallLogModel, featureModel) + whitelistService := service.NewWhitelistService(c, userFeatureWhitelistModel, whitelistOrderModel, whitelistOrderItemModel, queryModel, featureModel) + apiRequestService := service.NewApiRequestService(c, featureModel, productFeatureModel, userFeatureWhitelistModel, tianyuanapi, tianyuanapiCallLogService, whitelistService) verificationService := service.NewVerificationService(c, tianyuanapi, apiRequestService) asynqService := service.NewAsynqService(c) agentService := service.NewAgentService(c, orderModel, agentModel, agentWalletModel, @@ -214,6 +231,11 @@ func NewServiceContext(c config.Config) *ServiceContext { FeatureModel: featureModel, ProductFeatureModel: productFeatureModel, + // 白名单相关模型 + UserFeatureWhitelistModel: userFeatureWhitelistModel, + WhitelistOrderModel: whitelistOrderModel, + WhitelistOrderItemModel: whitelistOrderItemModel, + // 订单相关模型 OrderModel: orderModel, QueryModel: queryModel, @@ -257,11 +279,15 @@ func NewServiceContext(c config.Config) *ServiceContext { GlobalNotificationsModel: globalNotificationsModel, AuthorizationDocumentModel: authorizationDocumentModel, + // 第三方服务 + TianyuanapiCallLogService: tianyuanapiCallLogService, + // 服务 AlipayService: alipayService, WechatPayService: wechatPayService, ApplePayService: applePayService, ApiRequestService: apiRequestService, + WhitelistService: whitelistService, AsynqServer: asynqServer, AsynqService: asynqService, VerificationService: verificationService, diff --git a/app/main/api/internal/types/types.go b/app/main/api/internal/types/types.go index 68565a3..9b5ed05 100644 --- a/app/main/api/internal/types/types.go +++ b/app/main/api/internal/types/types.go @@ -1,6 +1,12 @@ // Code generated by goctl. DO NOT EDIT. package types +type AdminAgentStatistics struct { + TotalCount int64 `json:"total_count"` // 代理总数 + TodayNew int64 `json:"today_new"` // 今日新增 + MonthNew int64 `json:"month_new"` // 当月新增 +} + type AdminApiInfo struct { Id string `json:"id"` ApiName string `json:"api_name"` @@ -84,8 +90,10 @@ type AdminCreateApiResp struct { } type AdminCreateFeatureReq struct { - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice *float64 `json:"whitelist_price,optional"` // 白名单屏蔽价格（单位：元） + CostPrice *float64 `json:"cost_price,optional"` // 天远API调用成本价（单位：元） } type AdminCreateFeatureResp struct { @@ -383,16 +391,27 @@ type AdminGetApiListResp struct { Total int64 `json:"total"` } +type AdminGetDashboardStatisticsResp struct { + OrderStats AdminOrderStatistics `json:"order_stats"` + RevenueStats AdminRevenueStatistics `json:"revenue_stats"` + AgentStats AdminAgentStatistics `json:"agent_stats"` + ProfitStats AdminProfitStatistics `json:"profit_stats"` + OrderTrend []AdminTrendData `json:"order_trend"` + RevenueTrend []AdminTrendData `json:"revenue_trend"` +} + type AdminGetFeatureDetailReq struct { Id string `path:"id"` // 功能ID } type AdminGetFeatureDetailResp struct { - Id string `json:"id"` // 功能ID - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 - CreateTime string `json:"create_time"` // 创建时间 - UpdateTime string `json:"update_time"` // 更新时间 + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice float64 `json:"whitelist_price"` // 白名单屏蔽价格（单位：元） + CostPrice float64 `json:"cost_price"` // 天远API调用成本价（单位：元） + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 } type AdminGetFeatureExampleReq struct { @@ -684,6 +703,37 @@ type AdminLoginResp struct { Roles []string `json:"roles"` } +type AdminOrderStatistics struct { + TodayCount int64 `json:"today_count"` // 今日订单数 + MonthCount int64 `json:"month_count"` // 当月订单数 + TotalCount int64 `json:"total_count"` // 总订单数 + YesterdayCount int64 `json:"yesterday_count"` // 昨日订单数 + ChangeRate float64 `json:"change_rate"` // 变化率（百分比） +} + +type AdminProfitDetail struct { + Revenue float64 `json:"revenue"` // 营收 + Commission float64 `json:"commission"` // 佣金 + Rebate float64 `json:"rebate"` // 返利 + CompanyTax float64 `json:"company_tax"` // 税务成本 + ApiCost float64 `json:"api_cost"` // API调用成本 + TaxIncome float64 `json:"tax_income"` // 提现收税 + Profit float64 `json:"profit"` // 利润 + ProfitRate float64 `json:"profit_rate"` // 利润率 +} + +type AdminProfitStatistics struct { + TodayProfit float64 `json:"today_profit"` // 今日利润 + MonthProfit float64 `json:"month_profit"` // 当月利润 + TotalProfit float64 `json:"total_profit"` // 总利润 + TodayProfitRate float64 `json:"today_profit_rate"` // 今日利润率 + MonthProfitRate float64 `json:"month_profit_rate"` // 当月利润率 + TotalProfitRate float64 `json:"total_profit_rate"` // 总利润率 + TodayDetail AdminProfitDetail `json:"today_detail"` + MonthDetail AdminProfitDetail `json:"month_detail"` + TotalDetail AdminProfitDetail `json:"total_detail"` +} + type AdminQueryItem struct { Feature interface{} `json:"feature"` Data interface{} `json:"data"` // 这里可以是 map 或 具体的 struct @@ -729,6 +779,14 @@ type AdminRetryAgentProcessResp struct { ProcessedAt string `json:"processed_at"` // 处理时间 } +type AdminRevenueStatistics struct { + TodayAmount float64 `json:"today_amount"` // 今日营收 + MonthAmount float64 `json:"month_amount"` // 当月营收 + TotalAmount float64 `json:"total_amount"` // 总营收 + YesterdayAmount float64 `json:"yesterday_amount"` // 昨日营收 + ChangeRate float64 `json:"change_rate"` // 变化率（百分比） +} + type AdminRoleApiInfo struct { Id string `json:"id"` RoleId string `json:"role_id"` @@ -741,6 +799,11 @@ type AdminRoleApiInfo struct { Description string `json:"description"` } +type AdminTrendData struct { + Date string `json:"date"` // 日期（格式：MM-DD） + Value float64 `json:"value"` // 数值 +} + type AdminUpdateAgentConfigReq struct { LevelBonus *LevelBonusConfig `json:"level_bonus,optional"` // 等级加成配置 UpgradeFee *UpgradeFeeConfig `json:"upgrade_fee,optional"` // 升级费用配置 @@ -785,9 +848,11 @@ type AdminUpdateApiResp struct { } type AdminUpdateFeatureReq struct { - Id string `path:"id"` // 功能ID - ApiId *string `json:"api_id,optional"` // API标识 - Name *string `json:"name,optional"` // 描述 + Id string `path:"id"` // 功能ID + ApiId *string `json:"api_id,optional"` // API标识 + Name *string `json:"name,optional"` // 描述 + WhitelistPrice *float64 `json:"whitelist_price,optional"` // 白名单屏蔽价格（单位：元） + CostPrice *float64 `json:"cost_price,optional"` // 天远API调用成本价（单位：元） } type AdminUpdateFeatureResp struct { @@ -1148,14 +1213,14 @@ type BindMobileResp struct { type CheckFeatureWhitelistStatusReq struct { IdCard string `form:"id_card"` // 身份证号 FeatureApiId string `form:"feature_api_id"` // Feature的API标识 - QueryId string `form:"query_id,optional"` // 查询记录ID（可选，用于检查报告数据是否已删除） + QueryId string `form:"query_id,optional"` // 查询记录ID（可选） } type CheckFeatureWhitelistStatusResp struct { IsWhitelisted bool `json:"is_whitelisted"` // 是否在白名单中 - WhitelistPrice float64 `json:"whitelist_price"` // 屏蔽价格（单位：元），如果为0表示不支持下架 + WhitelistPrice float64 `json:"whitelist_price"` // 屏蔽价格（单位：元），0表示不支持下架 FeatureId string `json:"feature_id"` // Feature的UUID - DataDeleted bool `json:"data_deleted"` // 报告数据是否已删除（仅当提供了query_id时有效） + DataDeleted bool `json:"data_deleted"` // 报告数据是否已删除 } type CheckOrderAgentReq struct { @@ -1279,11 +1344,13 @@ type Feature struct { } type FeatureListItem struct { - Id string `json:"id"` // 功能ID - ApiId string `json:"api_id"` // API标识 - Name string `json:"name"` // 描述 - CreateTime string `json:"create_time"` // 创建时间 - UpdateTime string `json:"update_time"` // 更新时间 + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + WhitelistPrice float64 `json:"whitelist_price"` // 白名单屏蔽价格（单位：元） + CostPrice float64 `json:"cost_price"` // 天远API调用成本价（单位：元） + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 } type GenerateInviteCodeReq struct { @@ -1747,7 +1814,7 @@ type OfflineFeatureReq struct { type OfflineFeatureResp struct { Success bool `json:"success"` // 是否已完成下架 NeedPay bool `json:"need_pay"` // 是否需要发起支付 - Amount float64 `json:"amount"` // 需要支付的金额（单位：元），0表示无需支付 + Amount float64 `json:"amount"` // 需要支付的金额（元），0表示无需支付 } type OrderItem struct { @@ -1868,13 +1935,13 @@ type ProductResponse struct { } type PromotionQueryItem struct { - Id string `json:"id"` // 查询ID - OrderId string `json:"order_id"` // 订单ID - ProductName string `json:"product_name"` // 产品名称 - CreateTime string `json:"create_time"` // 创建时间 - QueryState string `json:"query_state"` // 查询状态 - Params map[string]interface{} `json:"params"` // 查询参数（已脱敏） - Price float64 `json:"price"` // 查询价格 + Id string `json:"id"` // 查询ID + OrderId string `json:"order_id"` // 订单ID + ProductName string `json:"product_name"` // 产品名称 + CreateTime string `json:"create_time"` // 创建时间 + QueryState string `json:"query_state"` // 查询状态 + Params map[string]interface{} `json:"params,optional"` // 查询参数（已脱敏） + Price float64 `json:"price"` // 查询价格 } type Query struct { diff --git a/app/main/model/featureModel_gen.go b/app/main/model/featureModel_gen.go index 2f1aed8..afce897 100644 --- a/app/main/model/featureModel_gen.go +++ b/app/main/model/featureModel_gen.go @@ -11,8 +11,6 @@ import ( "reflect" "time" - "qnc-server/common/globalkey" - "github.com/Masterminds/squirrel" "github.com/google/uuid" "github.com/pkg/errors" @@ -21,6 +19,7 @@ import ( "github.com/zeromicro/go-zero/core/stores/sqlc" "github.com/zeromicro/go-zero/core/stores/sqlx" "github.com/zeromicro/go-zero/core/stringx" + "qnc-server/common/globalkey" ) var ( @@ -59,14 +58,16 @@ type ( } Feature struct { - Id string `db:"id"` - CreateTime time.Time `db:"create_time"` // 创建时间 - UpdateTime time.Time `db:"update_time"` // 更新时间 - DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 - DelState int64 `db:"del_state"` // 删除状态 - Version int64 `db:"version"` // 版本号 - ApiId string `db:"api_id"` // API标识 - Name string `db:"name"` // 描述 + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + ApiId string `db:"api_id"` // API标识 + Name string `db:"name"` // 描述 + WhitelistPrice float64 `db:"whitelist_price"` // 白名单屏蔽价格（元） + CostPrice float64 `db:"cost_price"` // 天远API调用成本价（元） } ) @@ -83,11 +84,11 @@ func (m *defaultFeatureModel) Insert(ctx context.Context, session sqlx.Session, qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, data.ApiId) qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, data.Id) return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { - query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, featureRowsExpectAutoSet) + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?)", m.table, featureRowsExpectAutoSet) if session != nil { - return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name) + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name, data.WhitelistPrice, data.CostPrice) } - return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name) + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name, data.WhitelistPrice, data.CostPrice) }, qncFeatureApiIdKey, qncFeatureIdKey) } func (m *defaultFeatureModel) insertUUID(data *Feature) { @@ -154,9 +155,9 @@ func (m *defaultFeatureModel) Update(ctx context.Context, session sqlx.Session, return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, featureRowsWithPlaceHolder) if session != nil { - return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id) + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.WhitelistPrice, newData.CostPrice, newData.Id) } - return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id) + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.WhitelistPrice, newData.CostPrice, newData.Id) }, qncFeatureApiIdKey, qncFeatureIdKey) } @@ -177,9 +178,9 @@ func (m *defaultFeatureModel) UpdateWithVersion(ctx context.Context, session sql sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, featureRowsWithPlaceHolder) if session != nil { - return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id, oldVersion) + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.WhitelistPrice, newData.CostPrice, newData.Id, oldVersion) } - return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id, oldVersion) + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.WhitelistPrice, newData.CostPrice, newData.Id, oldVersion) }, qncFeatureApiIdKey, qncFeatureIdKey) if err != nil { return err diff --git a/app/main/model/tianyuanapiCallLogModel.go b/app/main/model/tianyuanapiCallLogModel.go new file mode 100644 index 0000000..8f103a4 --- /dev/null +++ b/app/main/model/tianyuanapiCallLogModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ TianyuanapiCallLogModel = (*customTianyuanapiCallLogModel)(nil) + +type ( + // TianyuanapiCallLogModel is an interface to be customized, add more methods here, + // and implement the added methods in customTianyuanapiCallLogModel. + TianyuanapiCallLogModel interface { + tianyuanapiCallLogModel + } + + customTianyuanapiCallLogModel struct { + *defaultTianyuanapiCallLogModel + } +) + +// NewTianyuanapiCallLogModel returns a model for the database table. +func NewTianyuanapiCallLogModel(conn sqlx.SqlConn, c cache.CacheConf) TianyuanapiCallLogModel { + return &customTianyuanapiCallLogModel{ + defaultTianyuanapiCallLogModel: newTianyuanapiCallLogModel(conn, c), + } +} diff --git a/app/main/model/tianyuanapiCallLogModel_gen.go b/app/main/model/tianyuanapiCallLogModel_gen.go new file mode 100644 index 0000000..d3a18d1 --- /dev/null +++ b/app/main/model/tianyuanapiCallLogModel_gen.go @@ -0,0 +1,398 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "qnc-server/common/globalkey" +) + +var ( + tianyuanapiCallLogFieldNames = builder.RawFieldNames(&TianyuanapiCallLog{}) + tianyuanapiCallLogRows = strings.Join(tianyuanapiCallLogFieldNames, ",") + tianyuanapiCallLogRowsExpectAutoSet = strings.Join(stringx.Remove(tianyuanapiCallLogFieldNames, "`id`", "`create_time`", "`update_time`"), ",") + tianyuanapiCallLogRowsWithPlaceHolder = strings.Join(stringx.Remove(tianyuanapiCallLogFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncTianyuanapiCallLogIdPrefix = "cache:qnc:tianyuanapiCallLog:id:" +) + +type ( + tianyuanapiCallLogModel interface { + Insert(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*TianyuanapiCallLog, error) + Update(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*TianyuanapiCallLog, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*TianyuanapiCallLog, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*TianyuanapiCallLog, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*TianyuanapiCallLog, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*TianyuanapiCallLog, error) + Delete(ctx context.Context, session sqlx.Session, id int64) error + } + + defaultTianyuanapiCallLogModel struct { + sqlc.CachedConn + table string + } + + TianyuanapiCallLog struct { + Id int64 `db:"id"` + FeatureId string `db:"feature_id"` // 功能ID（关联feature表） + ApiId string `db:"api_id"` // API标识 + OrderId sql.NullString `db:"order_id"` // 订单ID + QueryId sql.NullString `db:"query_id"` // 查询ID + CallStatus int64 `db:"call_status"` // 调用状态：0=失败，1=成功 + CallTime time.Time `db:"call_time"` // 调用时间 + ResponseTime sql.NullInt64 `db:"response_time"` // 响应耗时（毫秒） + CostPrice float64 `db:"cost_price"` // 本次调用成本（元） + ErrorCode sql.NullString `db:"error_code"` // 错误码 + ErrorMessage sql.NullString `db:"error_message"` // 错误信息 + RequestParams sql.NullString `db:"request_params"` // 请求参数（JSON） + ResponseData sql.NullString `db:"response_data"` // 响应数据（JSON） + TransactionId sql.NullString `db:"transaction_id"` // 天远API流水号 + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` + DelState int64 `db:"del_state"` // 0=未删除，1=已删除 + Version int64 `db:"version"` + } +) + +func newTianyuanapiCallLogModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultTianyuanapiCallLogModel { + return &defaultTianyuanapiCallLogModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`tianyuanapi_call_log`", + } +} + +func (m *defaultTianyuanapiCallLogModel) Insert(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncTianyuanapiCallLogIdKey := fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, tianyuanapiCallLogRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version) + }, qncTianyuanapiCallLogIdKey) +} +func (m *defaultTianyuanapiCallLogModel) insertUUID(data *TianyuanapiCallLog) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultTianyuanapiCallLogModel) FindOne(ctx context.Context, id int64) (*TianyuanapiCallLog, error) { + qncTianyuanapiCallLogIdKey := fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, id) + var resp TianyuanapiCallLog + err := m.QueryRowCtx(ctx, &resp, qncTianyuanapiCallLogIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", tianyuanapiCallLogRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultTianyuanapiCallLogModel) Update(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) (sql.Result, error) { + qncTianyuanapiCallLogIdKey := fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, tianyuanapiCallLogRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncTianyuanapiCallLogIdKey) +} + +func (m *defaultTianyuanapiCallLogModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncTianyuanapiCallLogIdKey := fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, tianyuanapiCallLogRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.FeatureId, data.ApiId, data.OrderId, data.QueryId, data.CallStatus, data.CallTime, data.ResponseTime, data.CostPrice, data.ErrorCode, data.ErrorMessage, data.RequestParams, data.ResponseData, data.TransactionId, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncTianyuanapiCallLogIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultTianyuanapiCallLogModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *TianyuanapiCallLog) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "TianyuanapiCallLogModel delete err : %+v", err) + } + return nil +} + +func (m *defaultTianyuanapiCallLogModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*TianyuanapiCallLog, error) { + + builder = builder.Columns(tianyuanapiCallLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*TianyuanapiCallLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*TianyuanapiCallLog, error) { + + builder = builder.Columns(tianyuanapiCallLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*TianyuanapiCallLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*TianyuanapiCallLog, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(tianyuanapiCallLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*TianyuanapiCallLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*TianyuanapiCallLog, error) { + + builder = builder.Columns(tianyuanapiCallLogRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*TianyuanapiCallLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultTianyuanapiCallLogModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*TianyuanapiCallLog, error) { + + builder = builder.Columns(tianyuanapiCallLogRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*TianyuanapiCallLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultTianyuanapiCallLogModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultTianyuanapiCallLogModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultTianyuanapiCallLogModel) Delete(ctx context.Context, session sqlx.Session, id int64) error { + qncTianyuanapiCallLogIdKey := fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncTianyuanapiCallLogIdKey) + return err +} +func (m *defaultTianyuanapiCallLogModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncTianyuanapiCallLogIdPrefix, primary) +} +func (m *defaultTianyuanapiCallLogModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", tianyuanapiCallLogRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultTianyuanapiCallLogModel) tableName() string { + return m.table +} diff --git a/app/main/model/userFeatureWhitelistModel.go b/app/main/model/userFeatureWhitelistModel.go new file mode 100644 index 0000000..e601ff3 --- /dev/null +++ b/app/main/model/userFeatureWhitelistModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ UserFeatureWhitelistModel = (*customUserFeatureWhitelistModel)(nil) + +type ( + // UserFeatureWhitelistModel is an interface to be customized, add more methods here, + // and implement the added methods in customUserFeatureWhitelistModel. + UserFeatureWhitelistModel interface { + userFeatureWhitelistModel + } + + customUserFeatureWhitelistModel struct { + *defaultUserFeatureWhitelistModel + } +) + +// NewUserFeatureWhitelistModel returns a model for the database table. +func NewUserFeatureWhitelistModel(conn sqlx.SqlConn, c cache.CacheConf) UserFeatureWhitelistModel { + return &customUserFeatureWhitelistModel{ + defaultUserFeatureWhitelistModel: newUserFeatureWhitelistModel(conn, c), + } +} diff --git a/app/main/model/userFeatureWhitelistModel_gen.go b/app/main/model/userFeatureWhitelistModel_gen.go new file mode 100644 index 0000000..79c5040 --- /dev/null +++ b/app/main/model/userFeatureWhitelistModel_gen.go @@ -0,0 +1,432 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "qnc-server/common/globalkey" +) + +var ( + userFeatureWhitelistFieldNames = builder.RawFieldNames(&UserFeatureWhitelist{}) + userFeatureWhitelistRows = strings.Join(userFeatureWhitelistFieldNames, ",") + userFeatureWhitelistRowsExpectAutoSet = strings.Join(stringx.Remove(userFeatureWhitelistFieldNames, "`create_time`", "`update_time`"), ",") + userFeatureWhitelistRowsWithPlaceHolder = strings.Join(stringx.Remove(userFeatureWhitelistFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncUserFeatureWhitelistIdPrefix = "cache:qnc:userFeatureWhitelist:id:" + cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix = "cache:qnc:userFeatureWhitelist:idCard:featureApiId:delState:" +) + +type ( + userFeatureWhitelistModel interface { + Insert(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) (sql.Result, error) + FindOne(ctx context.Context, id string) (*UserFeatureWhitelist, error) + FindOneByIdCardFeatureApiIdDelState(ctx context.Context, idCard string, featureApiId string, delState int64) (*UserFeatureWhitelist, error) + Update(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*UserFeatureWhitelist, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserFeatureWhitelist, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserFeatureWhitelist, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserFeatureWhitelist, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserFeatureWhitelist, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultUserFeatureWhitelistModel struct { + sqlc.CachedConn + table string + } + + UserFeatureWhitelist struct { + Id string `db:"id"` // UUID主键 + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` + DelState int64 `db:"del_state"` // 0=未删除，1=已删除 + Version int64 `db:"version"` + IdCard string `db:"id_card"` // 身份证号（查询对象标识） + FeatureId string `db:"feature_id"` // Feature的UUID + FeatureApiId string `db:"feature_api_id"` // Feature的API标识 + UserId string `db:"user_id"` // 操作用户ID + OrderId sql.NullString `db:"order_id"` // 关联的查询订单ID + WhitelistOrderId sql.NullString `db:"whitelist_order_id"` // 关联的白名单订单ID + Amount float64 `db:"amount"` // 费用（元） + Status int64 `db:"status"` // 状态：1=生效，2=已失效 + } +) + +func newUserFeatureWhitelistModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultUserFeatureWhitelistModel { + return &defaultUserFeatureWhitelistModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`user_feature_whitelist`", + } +} + +func (m *defaultUserFeatureWhitelistModel) Insert(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix, data.IdCard, data.FeatureApiId, data.DelState) + qncUserFeatureWhitelistIdKey := fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, userFeatureWhitelistRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.IdCard, data.FeatureId, data.FeatureApiId, data.UserId, data.OrderId, data.WhitelistOrderId, data.Amount, data.Status) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.IdCard, data.FeatureId, data.FeatureApiId, data.UserId, data.OrderId, data.WhitelistOrderId, data.Amount, data.Status) + }, qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey, qncUserFeatureWhitelistIdKey) +} +func (m *defaultUserFeatureWhitelistModel) insertUUID(data *UserFeatureWhitelist) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultUserFeatureWhitelistModel) FindOne(ctx context.Context, id string) (*UserFeatureWhitelist, error) { + qncUserFeatureWhitelistIdKey := fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, id) + var resp UserFeatureWhitelist + err := m.QueryRowCtx(ctx, &resp, qncUserFeatureWhitelistIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userFeatureWhitelistRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindOneByIdCardFeatureApiIdDelState(ctx context.Context, idCard string, featureApiId string, delState int64) (*UserFeatureWhitelist, error) { + qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix, idCard, featureApiId, delState) + var resp UserFeatureWhitelist + err := m.QueryRowIndexCtx(ctx, &resp, qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `id_card` = ? and `feature_api_id` = ? and `del_state` = ? and del_state = ? limit 1", userFeatureWhitelistRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, idCard, featureApiId, delState, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) Update(ctx context.Context, session sqlx.Session, newData *UserFeatureWhitelist) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix, data.IdCard, data.FeatureApiId, data.DelState) + qncUserFeatureWhitelistIdKey := fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, userFeatureWhitelistRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.IdCard, newData.FeatureId, newData.FeatureApiId, newData.UserId, newData.OrderId, newData.WhitelistOrderId, newData.Amount, newData.Status, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.IdCard, newData.FeatureId, newData.FeatureApiId, newData.UserId, newData.OrderId, newData.WhitelistOrderId, newData.Amount, newData.Status, newData.Id) + }, qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey, qncUserFeatureWhitelistIdKey) +} + +func (m *defaultUserFeatureWhitelistModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *UserFeatureWhitelist) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix, data.IdCard, data.FeatureApiId, data.DelState) + qncUserFeatureWhitelistIdKey := fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, userFeatureWhitelistRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.IdCard, newData.FeatureId, newData.FeatureApiId, newData.UserId, newData.OrderId, newData.WhitelistOrderId, newData.Amount, newData.Status, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.IdCard, newData.FeatureId, newData.FeatureApiId, newData.UserId, newData.OrderId, newData.WhitelistOrderId, newData.Amount, newData.Status, newData.Id, oldVersion) + }, qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey, qncUserFeatureWhitelistIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultUserFeatureWhitelistModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *UserFeatureWhitelist) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "UserFeatureWhitelistModel delete err : %+v", err) + } + return nil +} + +func (m *defaultUserFeatureWhitelistModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*UserFeatureWhitelist, error) { + + builder = builder.Columns(userFeatureWhitelistRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserFeatureWhitelist + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserFeatureWhitelist, error) { + + builder = builder.Columns(userFeatureWhitelistRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserFeatureWhitelist + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserFeatureWhitelist, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(userFeatureWhitelistRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*UserFeatureWhitelist + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserFeatureWhitelist, error) { + + builder = builder.Columns(userFeatureWhitelistRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserFeatureWhitelist + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserFeatureWhitelist, error) { + + builder = builder.Columns(userFeatureWhitelistRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserFeatureWhitelist + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserFeatureWhitelistModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultUserFeatureWhitelistModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultUserFeatureWhitelistModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncUserFeatureWhitelistIdCardFeatureApiIdDelStatePrefix, data.IdCard, data.FeatureApiId, data.DelState) + qncUserFeatureWhitelistIdKey := fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncUserFeatureWhitelistIdCardFeatureApiIdDelStateKey, qncUserFeatureWhitelistIdKey) + return err +} +func (m *defaultUserFeatureWhitelistModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncUserFeatureWhitelistIdPrefix, primary) +} +func (m *defaultUserFeatureWhitelistModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userFeatureWhitelistRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultUserFeatureWhitelistModel) tableName() string { + return m.table +} diff --git a/app/main/model/whitelistOrderItemModel.go b/app/main/model/whitelistOrderItemModel.go new file mode 100644 index 0000000..77bc118 --- /dev/null +++ b/app/main/model/whitelistOrderItemModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ WhitelistOrderItemModel = (*customWhitelistOrderItemModel)(nil) + +type ( + // WhitelistOrderItemModel is an interface to be customized, add more methods here, + // and implement the added methods in customWhitelistOrderItemModel. + WhitelistOrderItemModel interface { + whitelistOrderItemModel + } + + customWhitelistOrderItemModel struct { + *defaultWhitelistOrderItemModel + } +) + +// NewWhitelistOrderItemModel returns a model for the database table. +func NewWhitelistOrderItemModel(conn sqlx.SqlConn, c cache.CacheConf) WhitelistOrderItemModel { + return &customWhitelistOrderItemModel{ + defaultWhitelistOrderItemModel: newWhitelistOrderItemModel(conn, c), + } +} diff --git a/app/main/model/whitelistOrderItemModel_gen.go b/app/main/model/whitelistOrderItemModel_gen.go new file mode 100644 index 0000000..7d646a2 --- /dev/null +++ b/app/main/model/whitelistOrderItemModel_gen.go @@ -0,0 +1,390 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "qnc-server/common/globalkey" +) + +var ( + whitelistOrderItemFieldNames = builder.RawFieldNames(&WhitelistOrderItem{}) + whitelistOrderItemRows = strings.Join(whitelistOrderItemFieldNames, ",") + whitelistOrderItemRowsExpectAutoSet = strings.Join(stringx.Remove(whitelistOrderItemFieldNames, "`create_time`", "`update_time`"), ",") + whitelistOrderItemRowsWithPlaceHolder = strings.Join(stringx.Remove(whitelistOrderItemFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncWhitelistOrderItemIdPrefix = "cache:qnc:whitelistOrderItem:id:" +) + +type ( + whitelistOrderItemModel interface { + Insert(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) (sql.Result, error) + FindOne(ctx context.Context, id string) (*WhitelistOrderItem, error) + Update(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*WhitelistOrderItem, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrderItem, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrderItem, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*WhitelistOrderItem, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*WhitelistOrderItem, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultWhitelistOrderItemModel struct { + sqlc.CachedConn + table string + } + + WhitelistOrderItem struct { + Id string `db:"id"` // UUID主键 + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` + DelState int64 `db:"del_state"` // 0=未删除，1=已删除 + Version int64 `db:"version"` + OrderId string `db:"order_id"` // 订单ID（关联whitelist_order.id） + FeatureId string `db:"feature_id"` // Feature的UUID + FeatureApiId string `db:"feature_api_id"` // Feature的API标识 + FeatureName string `db:"feature_name"` // Feature名称 + Price float64 `db:"price"` // 单价（元） + } +) + +func newWhitelistOrderItemModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultWhitelistOrderItemModel { + return &defaultWhitelistOrderItemModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`whitelist_order_item`", + } +} + +func (m *defaultWhitelistOrderItemModel) Insert(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncWhitelistOrderItemIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, whitelistOrderItemRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price) + }, qncWhitelistOrderItemIdKey) +} +func (m *defaultWhitelistOrderItemModel) insertUUID(data *WhitelistOrderItem) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultWhitelistOrderItemModel) FindOne(ctx context.Context, id string) (*WhitelistOrderItem, error) { + qncWhitelistOrderItemIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, id) + var resp WhitelistOrderItem + err := m.QueryRowCtx(ctx, &resp, qncWhitelistOrderItemIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", whitelistOrderItemRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderItemModel) Update(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) (sql.Result, error) { + qncWhitelistOrderItemIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, whitelistOrderItemRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price, data.Id) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price, data.Id) + }, qncWhitelistOrderItemIdKey) +} + +func (m *defaultWhitelistOrderItemModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncWhitelistOrderItemIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, whitelistOrderItemRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.OrderId, data.FeatureId, data.FeatureApiId, data.FeatureName, data.Price, data.Id, oldVersion) + }, qncWhitelistOrderItemIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultWhitelistOrderItemModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *WhitelistOrderItem) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "WhitelistOrderItemModel delete err : %+v", err) + } + return nil +} + +func (m *defaultWhitelistOrderItemModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*WhitelistOrderItem, error) { + + builder = builder.Columns(whitelistOrderItemRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrderItem + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrderItem, error) { + + builder = builder.Columns(whitelistOrderItemRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrderItem + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrderItem, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(whitelistOrderItemRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*WhitelistOrderItem + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*WhitelistOrderItem, error) { + + builder = builder.Columns(whitelistOrderItemRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrderItem + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderItemModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*WhitelistOrderItem, error) { + + builder = builder.Columns(whitelistOrderItemRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrderItem + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderItemModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultWhitelistOrderItemModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultWhitelistOrderItemModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncWhitelistOrderItemIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncWhitelistOrderItemIdKey) + return err +} +func (m *defaultWhitelistOrderItemModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncWhitelistOrderItemIdPrefix, primary) +} +func (m *defaultWhitelistOrderItemModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", whitelistOrderItemRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultWhitelistOrderItemModel) tableName() string { + return m.table +} diff --git a/app/main/model/whitelistOrderModel.go b/app/main/model/whitelistOrderModel.go new file mode 100644 index 0000000..eae263d --- /dev/null +++ b/app/main/model/whitelistOrderModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ WhitelistOrderModel = (*customWhitelistOrderModel)(nil) + +type ( + // WhitelistOrderModel is an interface to be customized, add more methods here, + // and implement the added methods in customWhitelistOrderModel. + WhitelistOrderModel interface { + whitelistOrderModel + } + + customWhitelistOrderModel struct { + *defaultWhitelistOrderModel + } +) + +// NewWhitelistOrderModel returns a model for the database table. +func NewWhitelistOrderModel(conn sqlx.SqlConn, c cache.CacheConf) WhitelistOrderModel { + return &customWhitelistOrderModel{ + defaultWhitelistOrderModel: newWhitelistOrderModel(conn, c), + } +} diff --git a/app/main/model/whitelistOrderModel_gen.go b/app/main/model/whitelistOrderModel_gen.go new file mode 100644 index 0000000..98680e0 --- /dev/null +++ b/app/main/model/whitelistOrderModel_gen.go @@ -0,0 +1,433 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "qnc-server/common/globalkey" +) + +var ( + whitelistOrderFieldNames = builder.RawFieldNames(&WhitelistOrder{}) + whitelistOrderRows = strings.Join(whitelistOrderFieldNames, ",") + whitelistOrderRowsExpectAutoSet = strings.Join(stringx.Remove(whitelistOrderFieldNames, "`create_time`", "`update_time`"), ",") + whitelistOrderRowsWithPlaceHolder = strings.Join(stringx.Remove(whitelistOrderFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncWhitelistOrderIdPrefix = "cache:qnc:whitelistOrder:id:" + cacheQncWhitelistOrderOrderNoPrefix = "cache:qnc:whitelistOrder:orderNo:" +) + +type ( + whitelistOrderModel interface { + Insert(ctx context.Context, session sqlx.Session, data *WhitelistOrder) (sql.Result, error) + FindOne(ctx context.Context, id string) (*WhitelistOrder, error) + FindOneByOrderNo(ctx context.Context, orderNo string) (*WhitelistOrder, error) + Update(ctx context.Context, session sqlx.Session, data *WhitelistOrder) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *WhitelistOrder) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *WhitelistOrder) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*WhitelistOrder, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrder, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrder, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*WhitelistOrder, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*WhitelistOrder, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultWhitelistOrderModel struct { + sqlc.CachedConn + table string + } + + WhitelistOrder struct { + Id string `db:"id"` // UUID主键 + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` + DelState int64 `db:"del_state"` // 0=未删除，1=已删除 + Version int64 `db:"version"` + OrderNo string `db:"order_no"` // 订单号（唯一） + UserId string `db:"user_id"` // 用户ID（代理的user_id） + IdCard string `db:"id_card"` // 身份证号 + TotalAmount float64 `db:"total_amount"` // 总金额（元） + Status int64 `db:"status"` // 1=待支付，2=已支付，3=已取消 + PaymentMethod sql.NullString `db:"payment_method"` // 支付方式：wechat, alipay, appleiap等 + PaymentPlatform sql.NullString `db:"payment_platform"` // 支付平台 + PlatformOrderId sql.NullString `db:"platform_order_id"` // 支付平台订单号 + PayTime sql.NullTime `db:"pay_time"` // 支付时间 + } +) + +func newWhitelistOrderModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultWhitelistOrderModel { + return &defaultWhitelistOrderModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`whitelist_order`", + } +} + +func (m *defaultWhitelistOrderModel) Insert(ctx context.Context, session sqlx.Session, data *WhitelistOrder) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncWhitelistOrderIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, data.Id) + qncWhitelistOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderOrderNoPrefix, data.OrderNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, whitelistOrderRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.OrderNo, data.UserId, data.IdCard, data.TotalAmount, data.Status, data.PaymentMethod, data.PaymentPlatform, data.PlatformOrderId, data.PayTime) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.OrderNo, data.UserId, data.IdCard, data.TotalAmount, data.Status, data.PaymentMethod, data.PaymentPlatform, data.PlatformOrderId, data.PayTime) + }, qncWhitelistOrderIdKey, qncWhitelistOrderOrderNoKey) +} +func (m *defaultWhitelistOrderModel) insertUUID(data *WhitelistOrder) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultWhitelistOrderModel) FindOne(ctx context.Context, id string) (*WhitelistOrder, error) { + qncWhitelistOrderIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, id) + var resp WhitelistOrder + err := m.QueryRowCtx(ctx, &resp, qncWhitelistOrderIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", whitelistOrderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) FindOneByOrderNo(ctx context.Context, orderNo string) (*WhitelistOrder, error) { + qncWhitelistOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderOrderNoPrefix, orderNo) + var resp WhitelistOrder + err := m.QueryRowIndexCtx(ctx, &resp, qncWhitelistOrderOrderNoKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `order_no` = ? and del_state = ? limit 1", whitelistOrderRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, orderNo, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) Update(ctx context.Context, session sqlx.Session, newData *WhitelistOrder) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncWhitelistOrderIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, data.Id) + qncWhitelistOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderOrderNoPrefix, data.OrderNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, whitelistOrderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.OrderNo, newData.UserId, newData.IdCard, newData.TotalAmount, newData.Status, newData.PaymentMethod, newData.PaymentPlatform, newData.PlatformOrderId, newData.PayTime, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.OrderNo, newData.UserId, newData.IdCard, newData.TotalAmount, newData.Status, newData.PaymentMethod, newData.PaymentPlatform, newData.PlatformOrderId, newData.PayTime, newData.Id) + }, qncWhitelistOrderIdKey, qncWhitelistOrderOrderNoKey) +} + +func (m *defaultWhitelistOrderModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *WhitelistOrder) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncWhitelistOrderIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, data.Id) + qncWhitelistOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderOrderNoPrefix, data.OrderNo) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, whitelistOrderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.OrderNo, newData.UserId, newData.IdCard, newData.TotalAmount, newData.Status, newData.PaymentMethod, newData.PaymentPlatform, newData.PlatformOrderId, newData.PayTime, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.OrderNo, newData.UserId, newData.IdCard, newData.TotalAmount, newData.Status, newData.PaymentMethod, newData.PaymentPlatform, newData.PlatformOrderId, newData.PayTime, newData.Id, oldVersion) + }, qncWhitelistOrderIdKey, qncWhitelistOrderOrderNoKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultWhitelistOrderModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *WhitelistOrder) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "WhitelistOrderModel delete err : %+v", err) + } + return nil +} + +func (m *defaultWhitelistOrderModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultWhitelistOrderModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultWhitelistOrderModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*WhitelistOrder, error) { + + builder = builder.Columns(whitelistOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrder, error) { + + builder = builder.Columns(whitelistOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*WhitelistOrder, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(whitelistOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*WhitelistOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultWhitelistOrderModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*WhitelistOrder, error) { + + builder = builder.Columns(whitelistOrderRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*WhitelistOrder, error) { + + builder = builder.Columns(whitelistOrderRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*WhitelistOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultWhitelistOrderModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultWhitelistOrderModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultWhitelistOrderModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncWhitelistOrderIdKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, id) + qncWhitelistOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncWhitelistOrderOrderNoPrefix, data.OrderNo) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncWhitelistOrderIdKey, qncWhitelistOrderOrderNoKey) + return err +} +func (m *defaultWhitelistOrderModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncWhitelistOrderIdPrefix, primary) +} +func (m *defaultWhitelistOrderModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", whitelistOrderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultWhitelistOrderModel) tableName() string { + return m.table +} diff --git a/deploy/script/gen_models.ps1 b/deploy/script/gen_models.ps1 index b950a48..3aab2d4 100644 --- a/deploy/script/gen_models.ps1 +++ b/deploy/script/gen_models.ps1 @@ -10,6 +10,14 @@ $HOME_DIR = Join-Path $PSScriptRoot "..\template" # 表名列表 $tables = @( + # ============================================ + # 统计/白名单/成本价 同步用（执行 sync_whitelist_cost_tables.sql 后再生成） + # ============================================ + "feature", + "tianyuanapi_call_log", + "user_feature_whitelist", + "whitelist_order", + "whitelist_order_item" # ============================================ # 新代理系统表 # ============================================ @@ -28,7 +36,7 @@ $tables = @( # "admin_user_role", # "agent", # "agent_commission", - # "agent_config" + # # "agent_config" # "agent_freeze_task", # "agent_invite_code", # "agent_invite_code_usage", @@ -62,7 +70,7 @@ $tables = @( # 为每个表生成模型 foreach ($table in $tables) { Write-Host "正在生成表: $table" -ForegroundColor Green - # goctl model mysql datasource -url="qnc:5vg67b3UNHu8@tcp(127.0.0.1:21001)/qnc" -table="$table" -dir="./model" --home="$HOME_DIR" -cache=true --style=goZero + # goctl model mysql datasource -url="qnc:5vg67b3UNHu8@tcp(127.0.0.1:21201)/qnc" -table="$table" -dir="./model" --home="$HOME_DIR" -cache=true --style=goZero goctl model mysql datasource -url="qnc:5vg67b3UNHu8@tcp(127.0.0.1:21201)/qnc" -table="$table" -dir="./model" --home="$HOME_DIR" -cache=true --style=goZero # 移动生成的文件到目标目录