cialloo
ecf835068a
All checks were successful
CI - Build and Push / Build and Push Docker Image (push) Successful in 1m27s
- 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.
60 lines
1.5 KiB
Go
60 lines
1.5 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 TotalChatMessageCountLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Get total chat message count within a specified time range
|
|
func NewTotalChatMessageCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) *TotalChatMessageCountLogic {
|
|
return &TotalChatMessageCountLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *TotalChatMessageCountLogic) TotalChatMessageCount(req *types.TotalChatMessageCountReq) (resp *types.TotalChatMessageCountResp, err error) {
|
|
query := `
|
|
SELECT COUNT(*)
|
|
FROM steam_union.chat_log
|
|
WHERE message_time >= TO_TIMESTAMP($1 / 1000.0)
|
|
AND message_time <= TO_TIMESTAMP($2 / 1000.0)
|
|
`
|
|
|
|
args := []interface{}{req.TimeRangeStart, req.TimeRangeEnd}
|
|
|
|
// Add player filter if provided
|
|
if len(req.PlayerFilter) > 0 {
|
|
query += ` AND steamid64 = ANY($3)`
|
|
// Convert []string to []int64 for steamid64
|
|
steamIDs := make([]int64, len(req.PlayerFilter))
|
|
for i, id := range req.PlayerFilter {
|
|
fmt.Sscanf(id, "%d", &steamIDs[i])
|
|
}
|
|
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 chat message count: %v", err)
|
|
return nil, err
|
|
}
|
|
|
|
return &types.TotalChatMessageCountResp{
|
|
TotalChatMessageCount: count,
|
|
}, nil
|
|
}
|