mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
134 lines
3.9 KiB
Go
134 lines
3.9 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
)
|
|
|
|
// Origin OpenFlare 源站实体。
|
|
type Origin struct {
|
|
ID uint `json:"id" gorm:"primaryKey;autoIncrement"`
|
|
Name string `json:"name" gorm:"size:255;not null"`
|
|
Address string `json:"address" gorm:"uniqueIndex;size:255;not null"`
|
|
Remark string `json:"remark" gorm:"size:255"`
|
|
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
|
|
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
|
|
}
|
|
|
|
// TableName 表名。
|
|
func (Origin) TableName() string {
|
|
return "of_origins"
|
|
}
|
|
|
|
// OriginRouteCount 源站关联的代理规则数量。
|
|
type OriginRouteCount struct {
|
|
OriginID uint `json:"origin_id"`
|
|
RouteCount int64 `json:"route_count"`
|
|
}
|
|
|
|
// OriginProxyRoute 源站模块查询代理规则时使用的最小字段集。
|
|
type OriginProxyRoute struct {
|
|
ID uint `gorm:"column:id;primaryKey"`
|
|
OriginID *uint `gorm:"column:origin_id"`
|
|
Domain string `gorm:"column:domain"`
|
|
OriginURL string `gorm:"column:origin_url"`
|
|
Upstreams string `gorm:"column:upstreams"`
|
|
Enabled bool `gorm:"column:enabled"`
|
|
UpdatedAt time.Time `gorm:"column:updated_at"`
|
|
}
|
|
|
|
// TableName 表名。
|
|
func (OriginProxyRoute) TableName() string {
|
|
return tableOfProxyRoutes
|
|
}
|
|
|
|
// HasProxyRoutesTable 判断代理规则表是否已迁移。
|
|
func HasProxyRoutesTable(ctx context.Context) bool {
|
|
return db.DB(ctx).Migrator().HasTable(&OriginProxyRoute{})
|
|
}
|
|
|
|
// ListOrigins 列出全部源站。
|
|
func ListOrigins(ctx context.Context) ([]Origin, error) {
|
|
var origins []Origin
|
|
if err := db.DB(ctx).Order("id desc").Find(&origins).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return origins, nil
|
|
}
|
|
|
|
// GetOriginByID 按 ID 查询源站。
|
|
func GetOriginByID(ctx context.Context, id uint) (*Origin, error) {
|
|
var origin Origin
|
|
if err := db.DB(ctx).First(&origin, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &origin, nil
|
|
}
|
|
|
|
// GetOriginByAddress 按地址查询源站。
|
|
func GetOriginByAddress(ctx context.Context, address string) (*Origin, error) {
|
|
var origin Origin
|
|
if err := db.DB(ctx).Where("address = ?", address).First(&origin).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &origin, nil
|
|
}
|
|
|
|
// CreateOriginRecord 创建源站。
|
|
func CreateOriginRecord(ctx context.Context, origin *Origin) error {
|
|
return db.DB(ctx).Create(origin).Error
|
|
}
|
|
|
|
// SaveOrigin 保存源站。
|
|
func SaveOrigin(ctx context.Context, origin *Origin) error {
|
|
return db.DB(ctx).Save(origin).Error
|
|
}
|
|
|
|
// DeleteOriginRecord 删除源站。
|
|
func DeleteOriginRecord(ctx context.Context, id uint) error {
|
|
return db.DB(ctx).Delete(&Origin{}, id).Error
|
|
}
|
|
|
|
// ListOriginRouteCounts 统计各源站关联的代理规则数量。
|
|
func ListOriginRouteCounts(ctx context.Context) ([]OriginRouteCount, error) {
|
|
if !HasProxyRoutesTable(ctx) {
|
|
return nil, nil
|
|
}
|
|
result := make([]OriginRouteCount, 0)
|
|
err := db.DB(ctx).Model(&OriginProxyRoute{}).
|
|
Select("origin_id, COUNT(*) AS route_count").
|
|
Where("origin_id IS NOT NULL").
|
|
Group("origin_id").
|
|
Scan(&result).Error
|
|
return result, err
|
|
}
|
|
|
|
// ListProxyRoutesByOriginID 列出源站关联的代理规则。
|
|
func ListProxyRoutesByOriginID(ctx context.Context, originID uint) ([]OriginProxyRoute, error) {
|
|
if !HasProxyRoutesTable(ctx) {
|
|
return nil, nil
|
|
}
|
|
var routes []OriginProxyRoute
|
|
if err := db.DB(ctx).Where("origin_id = ?", originID).Order("id desc").Find(&routes).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return routes, nil
|
|
}
|
|
|
|
// CountProxyRoutesByOriginID 统计源站关联的代理规则数量。
|
|
func CountProxyRoutesByOriginID(ctx context.Context, originID uint) (int64, error) {
|
|
if !HasProxyRoutesTable(ctx) {
|
|
return 0, nil
|
|
}
|
|
var count int64
|
|
if err := db.DB(ctx).Model(&OriginProxyRoute{}).Where("origin_id = ?", originID).Count(&count).Error; err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|