Files
ServerStatistics/app/internal/logic/totaldamagecountlogic.go
cialloo ecf835068a
All checks were successful
CI - Build and Push / Build and Push Docker Image (push) Successful in 1m27s
feat: add total player count and total playtime logic
- Implemented TotalPlayerCountLogic to retrieve the total number of unique players within a specified time range.
- Implemented TotalPlayTimeLogic to calculate the total playtime of players within a specified time range.
- Created ServiceContext to manage database connections and initialize necessary tables.
- Added types for total player count and total playtime requests and responses.
- Set up the main server file to start the application with the necessary configurations.
- Updated go.mod and go.sum for new dependencies.
2025-10-12 11:20:34 +08:00

69 lines
1.8 KiB
Go

package logic
import (
"context"
"fmt"
"git.cialloo.com/CiallooWeb/ServerStatistics/app/internal/svc"
"git.cialloo.com/CiallooWeb/ServerStatistics/app/internal/types"
"github.com/zeromicro/go-zero/core/logx"
)
type TotalDamageCountLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
// Get total damage count within a specified time range
func NewTotalDamageCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) *TotalDamageCountLogic {
return &TotalDamageCountLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}
func (l *TotalDamageCountLogic) TotalDamageCount(req *types.TotalDamageCountReq) (resp *types.TotalDamageCountResp, err error) {
query := `
SELECT COALESCE(SUM(dmg_health), 0)
FROM steam_union.event_player_hurt_log
WHERE event_player_hurt_time >= TO_TIMESTAMP($1 / 1000.0)
AND event_player_hurt_time <= TO_TIMESTAMP($2 / 1000.0)
AND attacker_steamid64 > 1
`
args := []interface{}{req.TimeRangeStart, req.TimeRangeEnd}
argIndex := 3
// Add weapon filter if provided
if len(req.WeaponFilter) > 0 {
query += fmt.Sprintf(` AND weapon = ANY($%d)`, argIndex)
args = append(args, req.WeaponFilter)
argIndex++
}
// Add player filter if provided
if len(req.PlayerFilter) > 0 {
// Convert []string to []int64 for steamid64
steamIDs := make([]int64, len(req.PlayerFilter))
for i, id := range req.PlayerFilter {
fmt.Sscanf(id, "%d", &steamIDs[i])
}
query += fmt.Sprintf(` AND attacker_steamid64 = ANY($%d)`, argIndex)
args = append(args, steamIDs)
}
var count int64
err = l.svcCtx.DB.QueryRowContext(l.ctx, query, args...).Scan(&count)
if err != nil {
l.Errorf("Failed to query total damage count: %v", err)
return nil, err
}
return &types.TotalDamageCountResp{
TotalDamageCount: count,
}, nil
}