Skip to content

Commit 43f1ed7

Browse files
PetrusZPetrusZ
committed
add rollback stat api
Signed-off-by: Patrick Zhao <zhaoyu@koderover.com>
1 parent 1000b0d commit 43f1ed7

File tree

5 files changed

+873
-0
lines changed

5 files changed

+873
-0
lines changed

‎pkg/microservice/aslan/core/common/repository/mongodb/env_info.go‎

Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package mongodb
22

33
import (
44
"context"
5+
"fmt"
56
"time"
67

78
"github.com/pkg/errors"
@@ -102,6 +103,7 @@ type ListEnvInfoOption struct {
102103
StartTime int64
103104
EndTime int64
104105
Operation config.EnvOperation
106+
Production *bool
105107
}
106108

107109
func (c *EnvInfoColl) List(ctx context.Context, opt *ListEnvInfoOption) ([]*models.EnvInfo, int64, error) {
@@ -125,6 +127,10 @@ func (c *EnvInfoColl) List(ctx context.Context, opt *ListEnvInfoOption) ([]*mode
125127
if opt.Operation != "" {
126128
findOption["operation"] = opt.Operation
127129
}
130+
// 根据 production 参数过滤环境信息
131+
if opt.Production != nil {
132+
findOption["production"] = *opt.Production
133+
}
128134
findOption["create_time"] = bson.M{
129135
"$gte": opt.StartTime,
130136
"$lte": opt.EndTime,
@@ -205,3 +211,88 @@ func (c *EnvInfoColl) List(ctx context.Context, opt *ListEnvInfoOption) ([]*mode
205211

206212
return res, total, nil
207213
}
214+
215+
type RollbackServiceCount struct {
216+
Production bool `bson:"production" json:"production"`
217+
ProjectName string `bson:"project_name,omitempty" json:"project_name,omitempty"`
218+
ServiceName string `bson:"service_name,omitempty" json:"service_name,omitempty"`
219+
Count int `bson:"count" json:"count"`
220+
}
221+
222+
func (c *EnvInfoColl) GetTopRollbackedService(ctx context.Context, startTime, endTime int64, productionType config.ProductionType, projects []string, top int) ([]*RollbackServiceCount, error) {
223+
// 参数验证
224+
if startTime > endTime {
225+
return nil, fmt.Errorf("invalid time range: startTime (%d) should be less than or equal to endTime (%d)", startTime, endTime)
226+
}
227+
if top <= 0 {
228+
return nil, fmt.Errorf("invalid top parameter: top (%d) should be greater than 0", top)
229+
}
230+
231+
// 构建查询条件：只查询 rollback 操作
232+
query := bson.M{
233+
"operation": config.EnvOperationRollback,
234+
"create_time": bson.M{"$gte": startTime, "$lte": endTime},
235+
}
236+
237+
// 根据生产环境类型过滤
238+
switch productionType {
239+
case config.Production:
240+
query["production"] = true
241+
case config.Testing:
242+
query["production"] = false
243+
case config.Both:
244+
break
245+
default:
246+
return nil, fmt.Errorf("invalid production type: %s", productionType)
247+
}
248+
249+
if len(projects) > 0 {
250+
query["project_name"] = bson.M{"$in": projects}
251+
}
252+
253+
// 构建聚合管道
254+
pipeline := []bson.M{
255+
// 匹配 rollback 操作记录
256+
{"$match": query},
257+
// 按项目、服务和生产环境类型分组，统计 rollback 次数
258+
{
259+
"$group": bson.M{
260+
"_id": bson.M{
261+
"production": "$production",
262+
"project_name": "$project_name",
263+
"service_name": "$service_name",
264+
},
265+
"count": bson.M{"$sum": 1},
266+
},
267+
},
268+
// 按 rollback 次数降序排序
269+
{"$sort": bson.M{"count": -1}},
270+
// 限制返回数量
271+
{"$limit": top},
272+
// 投影字段，映射到返回结构
273+
{
274+
"$project": bson.M{
275+
"_id": 0,
276+
"production": "$_id.production",
277+
"project_name": "$_id.project_name",
278+
"service_name": "$_id.service_name",
279+
"count": 1,
280+
},
281+
},
282+
}
283+
284+
// 执行聚合查询
285+
cursor, err := c.Aggregate(mongotool.SessionContext(ctx, c.Session), pipeline)
286+
if err != nil {
287+
return nil, errors.Wrap(err, "failed to aggregate rollback service statistics")
288+
}
289+
defer cursor.Close(mongotool.SessionContext(ctx, c.Session))
290+
291+
// 解析结果
292+
result := make([]*RollbackServiceCount, 0)
293+
if err := cursor.All(mongotool.SessionContext(ctx, c.Session), &result); err != nil {
294+
return nil, errors.Wrap(err, "failed to decode rollback service statistics")
295+
}
296+
297+
return result, nil
298+
}

‎pkg/microservice/aslan/core/stat/handler/rollback_v2.go‎

Lines changed: 136 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,136 @@
1+
/*
2+
Copyright 2024 The KodeRover Authors.
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package handler
18+
19+
import (
20+
"github.com/gin-gonic/gin"
21+
e "github.com/koderover/zadig/v2/pkg/tool/errors"
22+
23+
"github.com/koderover/zadig/v2/pkg/microservice/aslan/core/stat/service"
24+
internalhandler "github.com/koderover/zadig/v2/pkg/shared/handler"
25+
)
26+
27+
// func CreateWeeklyDeployStat(c *gin.Context) {
28+
// ctx := internalhandler.NewContext(c)
29+
// defer func() { internalhandler.JSONResponse(c, ctx) }()
30+
31+
// ctx.RespErr = service.CreateWeeklyDeployStat(ctx.Logger)
32+
// }
33+
34+
// func CreateMonthlyDeployStat(c *gin.Context) {
35+
// ctx := internalhandler.NewContext(c)
36+
// defer func() { internalhandler.JSONResponse(c, ctx) }()
37+
38+
// ctx.RespErr = service.CreateMonthlyDeployStat(ctx.Logger)
39+
// }
40+
41+
// @Summary 获取回滚统计总数
42+
// @Description
43+
// @Tags stat
44+
// @Accept json
45+
// @Produce json
46+
// @Param startDate query int true "开始时间，格式为时间戳"
47+
// @Param endDate query int true "结束时间，格式为时间戳"
48+
// @Param projects query []string true "项目列表"
49+
// @Success 200 {object} service.RollbackTotalStat
50+
// @Router /api/aslan/stat/v2/quality/rollback/total [get]
51+
func GetRollbackTotalStat(c *gin.Context) {
52+
ctx := internalhandler.NewContext(c)
53+
defer func() { internalhandler.JSONResponse(c, ctx) }()
54+
55+
args := new(getStatGeneralReq)
56+
if err := c.ShouldBindQuery(args); err != nil {
57+
ctx.RespErr = e.ErrInvalidParam.AddErr(err)
58+
return
59+
}
60+
61+
ctx.Resp, ctx.RespErr = service.GetRollbackTotalStat(args.StartTime, args.EndTime, args.Projects, ctx.Logger)
62+
}
63+
64+
// @Summary 获取回滚次数TopN的服务
65+
// @Description
66+
// @Tags stat
67+
// @Accept json
68+
// @Produce json
69+
// @Param startDate query int true "开始时间，格式为时间戳"
70+
// @Param endDate query int true "结束时间，格式为时间戳"
71+
// @Param top query int true "TopN"
72+
// @Param production query string true "环境类型，可选值为 production、testing、both"
73+
// @Success 200 {array} mongodb.RollbackServiceCount
74+
// @Router /api/aslan/stat/v2/quality/rollback/service/top [get]
75+
func GetTopRollbackedService(c *gin.Context) {
76+
ctx := internalhandler.NewContext(c)
77+
defer func() { internalhandler.JSONResponse(c, ctx) }()
78+
79+
args := new(getStateReqWithTop)
80+
if err := c.ShouldBindQuery(args); err != nil {
81+
ctx.RespErr = e.ErrInvalidParam.AddErr(err)
82+
return
83+
}
84+
85+
ctx.Resp, ctx.RespErr = service.GetTopRollbackedProject(args.StartTime, args.EndTime, args.Top, args.ProductionType, args.Projects, ctx.Logger)
86+
}
87+
88+
func GetRollbackWeeklyTrend(c *gin.Context) {
89+
ctx := internalhandler.NewContext(c)
90+
defer func() { internalhandler.JSONResponse(c, ctx) }()
91+
92+
args := new(getStatGeneralReq)
93+
if err := c.ShouldBindQuery(args); err != nil {
94+
ctx.RespErr = e.ErrInvalidParam.AddErr(err)
95+
return
96+
}
97+
98+
ctx.Resp, ctx.RespErr = service.GetDeployWeeklyTrend(args.StartTime, args.EndTime, args.Projects, args.ProductionType, ctx.Logger)
99+
}
100+
101+
func GetRollbackMonthlyTrend(c *gin.Context) {
102+
ctx := internalhandler.NewContext(c)
103+
defer func() { internalhandler.JSONResponse(c, ctx) }()
104+
105+
args := new(getStatGeneralReq)
106+
if err := c.ShouldBindQuery(args); err != nil {
107+
ctx.RespErr = e.ErrInvalidParam.AddErr(err)
108+
return
109+
}
110+
111+
// ctx.Resp, ctx.RespErr = service.GetRollbackMonthlyTrend(args.StartTime, args.EndTime, args.Projects, args.ProductionType, ctx.Logger)
112+
}
113+
114+
// @Summary 获取回滚数据详情
115+
// @Description
116+
// @Tags stat
117+
// @Accept json
118+
// @Produce json
119+
// @Param startDate query int true "开始时间，格式为时间戳"
120+
// @Param endDate query int true "结束时间，格式为时间戳"
121+
// @Param projects query []string true "项目列表"
122+
// @Param production query string true "环境类型，可选值为 production、testing、both"
123+
// @Success 200 {object} service.GetRollbackStatResponse
124+
// @Router /api/aslan/stat/v2/quality/rollback/stat [get]
125+
func GetRollbackStat(c *gin.Context) {
126+
ctx := internalhandler.NewContext(c)
127+
defer func() { internalhandler.JSONResponse(c, ctx) }()
128+
129+
args := new(getStatGeneralReq)
130+
if err := c.ShouldBindQuery(args); err != nil {
131+
ctx.RespErr = e.ErrInvalidParam.AddErr(err)
132+
return
133+
}
134+
135+
ctx.Resp, ctx.RespErr = service.GetRollbackStat(args.StartTime, args.EndTime, args.ProductionType, args.Projects, ctx.Logger)
136+
}

‎pkg/microservice/aslan/core/stat/handler/router.go‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,18 @@ func (*Router) Inject(router *gin.RouterGroup) {
105105
deployV2.GET("/service/failure", GetTopDeployFailuresByService)
106106
}
107107

108+
rollbackV2 := qualityV2.Group("rollback")
109+
{
110+
rollbackV2.POST("/weekly", CreateWeeklyDeployStat)
111+
// rollbackV2.POST("/monthly", CreateMonthlyDeployStat)
112+
rollbackV2.GET("/total", GetRollbackTotalStat)
113+
rollbackV2.GET("/service/top", GetTopRollbackedService)
114+
rollbackV2.GET("/trend/weekly", GetRollbackWeeklyTrend)
115+
rollbackV2.GET("/stat", GetRollbackStat)
116+
// rollbackV2.GET("/trend/monthly", GetDeployMonthlyTrend)
117+
// rollbackV2.GET("/service/failure", GetTopDeployFailuresByService)
118+
}
119+
108120
}
109121

110122
type OpenAPIRouter struct{}

0 commit comments

Comments
 (0)