Skip to content

Commit 6971611

Browse files
silverwindclaude
andcommitted
Remove unnecessary heatmap JSON helper functions from model layer
Move the [][2]int64 conversion inline into the route handlers instead of adding dedicated model functions for it. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 3623f5e commit 6971611

3 files changed

Lines changed: 14 additions & 38 deletions

File tree

models/activities/user_heatmap.go

Lines changed: 0 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -80,35 +80,3 @@ func GetTotalContributionsInHeatmap(hdata []*UserHeatmapData) int64 {
8080
}
8181
return total
8282
}
83-
84-
// heatmapDataJSON converts UserHeatmapData to a [][2]int64 format and computes
85-
// total contributions.
86-
func heatmapDataJSON(hdata []*UserHeatmapData) ([][2]int64, int64) {
87-
result := make([][2]int64, len(hdata))
88-
var total int64
89-
for i, v := range hdata {
90-
result[i] = [2]int64{int64(v.Timestamp), v.Contributions}
91-
total += v.Contributions
92-
}
93-
return result, total
94-
}
95-
96-
// GetUserHeatmapDataByUserJSON returns heatmap data as [][2]int64 and total contributions
97-
func GetUserHeatmapDataByUserJSON(ctx context.Context, user, doer *user_model.User) ([][2]int64, int64, error) {
98-
hdata, err := getUserHeatmapData(ctx, user, nil, doer)
99-
if err != nil {
100-
return nil, 0, err
101-
}
102-
data, total := heatmapDataJSON(hdata)
103-
return data, total, nil
104-
}
105-
106-
// GetUserHeatmapDataByUserTeamJSON returns heatmap data as [][2]int64 and total contributions
107-
func GetUserHeatmapDataByUserTeamJSON(ctx context.Context, user *user_model.User, team *organization.Team, doer *user_model.User) ([][2]int64, int64, error) {
108-
hdata, err := getUserHeatmapData(ctx, user, team, doer)
109-
if err != nil {
110-
return nil, 0, err
111-
}
112-
data, total := heatmapDataJSON(hdata)
113-
return data, total, nil
114-
}

routers/web/user/home.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -162,14 +162,18 @@ func DashboardHeatmap(ctx *context.Context) {
162162
return
163163
}
164164

165-
data, total, err := activities_model.GetUserHeatmapDataByUserTeamJSON(ctx, ctxUser, ctx.Org.Team, ctx.Doer)
165+
hdata, err := activities_model.GetUserHeatmapDataByUserTeam(ctx, ctxUser, ctx.Org.Team, ctx.Doer)
166166
if err != nil {
167-
ctx.ServerError("GetUserHeatmapDataByUserTeamJSON", err)
167+
ctx.ServerError("GetUserHeatmapDataByUserTeam", err)
168168
return
169169
}
170+
data := make([][2]int64, len(hdata))
171+
for i, v := range hdata {
172+
data[i] = [2]int64{int64(v.Timestamp), v.Contributions}
173+
}
170174
ctx.JSON(http.StatusOK, map[string]any{
171175
"heatmapData": data,
172-
"totalContributions": total,
176+
"totalContributions": activities_model.GetTotalContributionsInHeatmap(hdata),
173177
})
174178
}
175179

routers/web/user/profile.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -321,14 +321,18 @@ func Heatmap(ctx *context.Context) {
321321
return
322322
}
323323

324-
data, total, err := activities_model.GetUserHeatmapDataByUserJSON(ctx, ctx.ContextUser, ctx.Doer)
324+
hdata, err := activities_model.GetUserHeatmapDataByUser(ctx, ctx.ContextUser, ctx.Doer)
325325
if err != nil {
326-
ctx.ServerError("GetUserHeatmapDataByUserJSON", err)
326+
ctx.ServerError("GetUserHeatmapDataByUser", err)
327327
return
328328
}
329+
data := make([][2]int64, len(hdata))
330+
for i, v := range hdata {
331+
data[i] = [2]int64{int64(v.Timestamp), v.Contributions}
332+
}
329333
ctx.JSON(http.StatusOK, map[string]any{
330334
"heatmapData": data,
331-
"totalContributions": total,
335+
"totalContributions": activities_model.GetTotalContributionsInHeatmap(hdata),
332336
})
333337
}
334338

0 commit comments

Comments
 (0)