OpenFlare/atsf_server/service/access_log.go

package service

import (
	"atsflare/model"
	"strings"
	"time"
)

const (
	defaultAccessLogPageSize = 50
	maxAccessLogPageSize     = 200
)

type AccessLogView struct {
	ID         uint      `json:"id"`
	NodeID     string    `json:"node_id"`
	NodeName   string    `json:"node_name"`
	LoggedAt   time.Time `json:"logged_at"`
	RemoteAddr string    `json:"remote_addr"`
	Host       string    `json:"host"`
	Path       string    `json:"path"`
	StatusCode int       `json:"status_code"`
}

type AccessLogList struct {
	Items    []AccessLogView `json:"items"`
	Page     int             `json:"page"`
	PageSize int             `json:"page_size"`
	HasMore  bool            `json:"has_more"`
}

func ListAccessLogs(nodeID string, page int, pageSize int) (*AccessLogList, error) {
	normalizedPage := normalizeAccessLogPage(page)
	normalizedPageSize := normalizeAccessLogPageSize(pageSize)
	offset := normalizedPage * normalizedPageSize
	logs, err := model.ListNodeAccessLogs(
		strings.TrimSpace(nodeID),
		time.Now().Add(-nodeAccessLogRetentionWindow),
		offset,
		normalizedPageSize+1,
	)
	if err != nil {
		return nil, err
	}
	nodes, err := model.ListNodes()
	if err != nil {
		return nil, err
	}
	nodeNames := make(map[string]string, len(nodes))
	for _, node := range nodes {
		if node == nil {
			continue
		}
		nodeNames[node.NodeID] = node.Name
	}
	hasMore := len(logs) > normalizedPageSize
	if hasMore {
		logs = logs[:normalizedPageSize]
	}
	views := make([]AccessLogView, 0, len(logs))
	for _, item := range logs {
		if item == nil {
			continue
		}
		views = append(views, AccessLogView{
			ID:         item.ID,
			NodeID:     item.NodeID,
			NodeName:   nodeNames[item.NodeID],
			LoggedAt:   item.LoggedAt,
			RemoteAddr: item.RemoteAddr,
			Host:       item.Host,
			Path:       item.Path,
			StatusCode: item.StatusCode,
		})
	}
	return &AccessLogList{
		Items:    views,
		Page:     normalizedPage,
		PageSize: normalizedPageSize,
		HasMore:  hasMore,
	}, nil
}

func normalizeAccessLogPage(page int) int {
	if page < 0 {
		return 0
	}
	return page
}

func normalizeAccessLogPageSize(pageSize int) int {
	if pageSize <= 0 {
		return defaultAccessLogPageSize
	}
	if pageSize > maxAccessLogPageSize {
		return maxAccessLogPageSize
	}
	return pageSize
}