mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-11-12 21:44:22 +08:00
ArthurKoba
3c7b347d7b
* perf(nodes): paginate node lists and fetch details directly * fix(nodes): preserve Unicode search and page transition state * fix(nodes): validate search using the UTF-8 byte limit
55 lines
1.9 KiB
Go
55 lines
1.9 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
db "github.com/Rain-kl/Wavelet/internal/infra/persistence"
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ListOpenFlareNodePage returns a bounded page and the count for the same filters.
|
|
func ListOpenFlareNodePage(ctx context.Context, query model.OpenFlareNodeListQuery) ([]model.OpenFlareNode, int64, error) {
|
|
conn := db.DB(ctx)
|
|
if conn == nil {
|
|
return nil, 0, errors.New(errDatabaseNotInitialized)
|
|
}
|
|
filtered := filterOpenFlareNodes(conn.Model(&model.OpenFlareNode{}), query)
|
|
var total int64
|
|
if err := filtered.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
nodes := make([]model.OpenFlareNode, 0)
|
|
err := filtered.Order("id DESC").Limit(query.PageSize).Offset((query.Page - 1) * query.PageSize).Find(&nodes).Error
|
|
return nodes, total, err
|
|
}
|
|
|
|
func filterOpenFlareNodes(query *gorm.DB, filters model.OpenFlareNodeListQuery) *gorm.DB {
|
|
if filters.Search != "" {
|
|
escaped := strings.NewReplacer("!", "!!", "%", "!%", "_", "!_").Replace(filters.Search)
|
|
pattern := "%" + escaped + "%"
|
|
query = query.Where("(LOWER(name) LIKE LOWER(?) ESCAPE '!' OR LOWER(node_id) LIKE LOWER(?) ESCAPE '!' OR LOWER(ip) LIKE LOWER(?) ESCAPE '!')", pattern, pattern, pattern)
|
|
}
|
|
if filters.NodeType == "edge_node" {
|
|
query = query.Where("(node_type = ? OR node_type = '')", filters.NodeType)
|
|
} else if filters.NodeType != "" {
|
|
query = query.Where("node_type = ?", filters.NodeType)
|
|
}
|
|
cutoff := filters.Now.Add(-time.Minute)
|
|
switch filters.Status {
|
|
case "pending":
|
|
query = query.Where("(last_seen_at IS NULL OR last_seen_at = ?)", time.Time{})
|
|
case "online":
|
|
query = query.Where("last_seen_at >= ? AND last_seen_at <> ?", cutoff, time.Time{})
|
|
case "offline":
|
|
query = query.Where("last_seen_at < ? AND last_seen_at <> ?", cutoff, time.Time{})
|
|
}
|
|
return query
|
|
}
|