init: 1.0.0

2025-04-25 12:08:29 +09:00
commit 8addcc584b
1031 changed files with 76472 additions and 0 deletions
@@ -0,0 +1,69 @@
+package traffic
+
+import (
+	"context"
+	"errors"
+
+	"gorm.io/gorm"
+)
+
+var _ Model = (*customTrafficModel)(nil)
+
+type (
+	Model interface {
+		trafficModel
+		customTrafficLogicModel
+	}
+	trafficModel interface {
+		Insert(ctx context.Context, data *TrafficLog) error
+		FindOne(ctx context.Context, id int64) (*TrafficLog, error)
+		Update(ctx context.Context, data *TrafficLog) error
+		Delete(ctx context.Context, id int64) error
+		Transaction(ctx context.Context, fn func(db *gorm.DB) error) error
+	}
+
+	customTrafficModel struct {
+		*defaultTrafficModel
+	}
+	defaultTrafficModel struct {
+		Conn  *gorm.DB
+		table string
+	}
+)
+
+func newTrafficModel(db *gorm.DB) *defaultTrafficModel {
+	return &defaultTrafficModel{
+		Conn:  db,
+		table: "`traffic`",
+	}
+}
+
+func (m *defaultTrafficModel) Insert(ctx context.Context, data *TrafficLog) error {
+	return m.Conn.WithContext(ctx).Create(&data).Error
+}
+
+func (m *defaultTrafficModel) FindOne(ctx context.Context, id int64) (*TrafficLog, error) {
+	var data TrafficLog
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).Where("`id` = ?", id).First(&data).Error
+	return &data, err
+}
+
+func (m *defaultTrafficModel) Update(ctx context.Context, data *TrafficLog) error {
+	return m.Conn.WithContext(ctx).Save(data).Error
+}
+
+func (m *defaultTrafficModel) Delete(ctx context.Context, id int64) error {
+	_, err := m.FindOne(ctx, id)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil
+		}
+		return err
+	}
+
+	return m.Conn.WithContext(ctx).Delete(&TrafficLog{}, id).Error
+}
+
+func (m *defaultTrafficModel) Transaction(ctx context.Context, fn func(db *gorm.DB) error) error {
+	return m.Conn.WithContext(ctx).Transaction(fn)
+}
@@ -0,0 +1,123 @@
+package traffic
+
+import (
+	"context"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type customTrafficLogicModel interface {
+	QueryServerTrafficByDay(ctx context.Context, serverId int64, date time.Time) (*TotalTraffic, error)
+	QueryTrafficByDay(ctx context.Context, date time.Time) (*TotalTraffic, error)
+	QueryTrafficByMonthly(ctx context.Context, date time.Time) (*TotalTraffic, error)
+	TopServersTrafficByDay(ctx context.Context, date time.Time, limit int) ([]ServerTrafficRanking, error)
+	TopServersTrafficByMonthly(ctx context.Context, date time.Time, limit int) ([]ServerTrafficRanking, error)
+	TopUsersTrafficByDay(ctx context.Context, date time.Time, limit int) ([]UserTrafficRanking, error)
+	TopUsersTrafficByMonthly(ctx context.Context, date time.Time, limit int) ([]UserTrafficRanking, error)
+	QueryTrafficLogPageList(ctx context.Context, userId, subscribeId int64, page, size int) ([]*TrafficLog, int64, error)
+}
+
+// NewModel returns a model for the database table.
+func NewModel(conn *gorm.DB) Model {
+	return &customTrafficModel{
+		defaultTrafficModel: newTrafficModel(conn),
+	}
+}
+
+func (m *customTrafficModel) QueryServerTrafficByDay(ctx context.Context, serverId int64, date time.Time) (*TotalTraffic, error) {
+	var data TotalTraffic
+	start := date.Truncate(24 * time.Hour)
+	end := start.Add(24 * time.Hour).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("sum(download) as download, sum(upload) as upload").
+		Where("server_id = ? AND timestamp BETWEEN ? AND ?", serverId, start, end).
+		Scan(&data).Error
+	return &data, err
+}
+
+func (m *customTrafficModel) QueryTrafficByDay(ctx context.Context, date time.Time) (*TotalTraffic, error) {
+	var data TotalTraffic
+	start := time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, time.Local)
+	end := start.Add(24 * time.Hour).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("sum(download) as download, sum(upload) as upload").
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Scan(&data).Error
+	return &data, err
+}
+
+func (m *customTrafficModel) QueryTrafficByMonthly(ctx context.Context, date time.Time) (*TotalTraffic, error) {
+	var data TotalTraffic
+	start := time.Date(date.Year(), date.Month(), 1, 0, 0, 0, 0, time.Local)
+	end := start.AddDate(0, 1, 0).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("sum(download) as download, sum(upload) as upload").
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Scan(&data).Error
+	return &data, err
+}
+
+func (m *customTrafficModel) TopServersTrafficByDay(ctx context.Context, date time.Time, limit int) ([]ServerTrafficRanking, error) {
+	var summaries []ServerTrafficRanking
+	start := time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, time.Local)
+	end := start.Add(24 * time.Hour).Add(-time.Nanosecond)
+	err := m.Conn.Debug().WithContext(ctx).Model(&TrafficLog{}).
+		Select("server_id, SUM(download + upload) AS total, SUM(download) AS download, SUM(upload) AS upload").
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Group("server_id").
+		Order("total DESC").
+		Limit(limit).
+		Scan(&summaries).Error
+	return summaries, err
+}
+
+func (m *customTrafficModel) TopServersTrafficByMonthly(ctx context.Context, date time.Time, limit int) ([]ServerTrafficRanking, error) {
+	var summaries []ServerTrafficRanking
+	start := time.Date(date.Year(), date.Month(), 1, 0, 0, 0, 0, time.Local)
+	end := start.AddDate(0, 1, 0).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("server_id, SUM(download + upload) AS total, SUM(download) AS download, SUM(upload) AS upload").
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Group("server_id").
+		Order("total DESC").
+		Limit(limit).
+		Scan(&summaries).Error
+	return summaries, err
+}
+
+func (m *customTrafficModel) TopUsersTrafficByDay(ctx context.Context, date time.Time, limit int) ([]UserTrafficRanking, error) {
+	var summaries []UserTrafficRanking
+	start := time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, time.Local)
+	end := start.Add(24 * time.Hour).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("user_id, subscribe_id, SUM(download + upload) AS total, SUM(download) AS download, SUM(upload) AS upload").
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Group("user_id, subscribe_id"). // 修改这里，添加 subscribe_id 到 GROUP BY 子句
+		Order("total DESC").
+		Limit(limit).
+		Scan(&summaries).Error
+	return summaries, err
+}
+
+func (m *customTrafficModel) TopUsersTrafficByMonthly(ctx context.Context, date time.Time, limit int) ([]UserTrafficRanking, error) {
+	var summaries []UserTrafficRanking
+	start := time.Date(date.Year(), date.Month(), 1, 0, 0, 0, 0, time.Local)
+	end := start.AddDate(0, 1, 0).Add(-time.Nanosecond)
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).
+		Select("user_id, subscribe_id, SUM(download + upload) AS total, SUM(download) AS download, SUM(upload) AS upload"). // 添加 subscribe_id 到 SELECT 列表
+		Where("timestamp BETWEEN ? AND ?", start, end).
+		Group("user_id, subscribe_id"). // 修改这里，添加 subscribe_id 到 GROUP BY 子句
+		Order("total DESC").
+		Limit(limit).
+		Scan(&summaries).Error
+	return summaries, err
+}
+
+// QueryTrafficLogPageList returns a list of records that meet the conditions.
+func (m *customTrafficModel) QueryTrafficLogPageList(ctx context.Context, userId, subscribeId int64, page, size int) ([]*TrafficLog, int64, error) {
+	var list []*TrafficLog
+	var total int64
+	err := m.Conn.WithContext(ctx).Model(&TrafficLog{}).Where("user_id = ? and subscribe_id= ?", userId, subscribeId).Count(&total).Limit(size).Offset((page - 1) * size).Find(&list).Error
+	return list, total, err
+}
@@ -0,0 +1,38 @@
+package traffic
+
+import "time"
+
+//goland:noinspection GoNameStartsWithPackageName
+type TrafficLog struct {
+	Id          int64     `gorm:"primaryKey"`
+	ServerId    int64     `gorm:"index:idx_server_id;not null;comment:Server ID"`
+	UserId      int64     `gorm:"index:idx_user_id;not null;comment:User ID"`
+	SubscribeId int64     `gorm:"index:idx_subscribe_id;not null;comment:Subscription ID"`
+	Download    int64     `gorm:"default:0;comment:Download Traffic"`
+	Upload      int64     `gorm:"default:0;comment:Upload Traffic"`
+	Timestamp   time.Time `gorm:"default:CURRENT_TIMESTAMP(3);not null;comment:Traffic Log Time"`
+}
+
+type TotalTraffic struct {
+	Download int64
+	Upload   int64
+}
+
+type ServerTrafficRanking struct {
+	ServerId int64
+	Download int64
+	Upload   int64
+	Total    int64
+}
+
+type UserTrafficRanking struct {
+	UserId      int64
+	SubscribeId int64
+	Download    int64
+	Upload      int64
+	Total       int64
+}
+
+func (TrafficLog) TableName() string {
+	return "traffic_log"
+}