globallimit.go 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186
  1. package repository
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
  7. "github.com/go-nunu/nunu-layout-advanced/internal/model"
  8. "gorm.io/gorm"
  9. "time"
  10. )
  11. type GlobalLimitRepository interface {
  12. GetGlobalLimit(ctx context.Context, id int64) (*model.GlobalLimit, error)
  13. AddGlobalLimit(ctx context.Context, req *model.GlobalLimit) error
  14. UpdateGlobalLimitByHostId(ctx context.Context, req *model.GlobalLimit) error
  15. IsGlobalLimitExistByHostId(ctx context.Context, hostId int64) (bool, error)
  16. GetGlobalLimitByHostId(ctx context.Context, hostId int64) (*model.GlobalLimit, error)
  17. GetGlobalLimitAllExpired(ctx context.Context,ids []int) ([]v1.GlobalLimitExpiredByHost, error)
  18. GetGlobalLimitAllHostId(ctx context.Context) ([]v1.GlobalLimitExpired, error)
  19. GetGlobalLimitFirst(ctx context.Context,uid int64) (*model.GlobalLimit, error)
  20. GetUserInfo(ctx context.Context, uid int64) (v1.UserInfo, error)
  21. GetHostName(ctx context.Context,hostId int64) (string, error)
  22. GetNodeId(ctx context.Context, cndWebId int) (int64, error)
  23. // 获取套餐Id
  24. GetNodeArea(ctx context.Context, nodeAreaName string) (int64, error)
  25. // 修改套餐状态
  26. EditHostState(ctx context.Context, hostId int64, state bool) error
  27. // 获取指定到期时间
  28. GetGlobalLimitAlmostExpired(ctx context.Context, addTime int64) ([]model.GlobalLimit, error)
  29. }
  30. func NewGlobalLimitRepository(
  31. repository *Repository,
  32. ) GlobalLimitRepository {
  33. return &globalLimitRepository{
  34. Repository: repository,
  35. }
  36. }
  37. type globalLimitRepository struct {
  38. *Repository
  39. }
  40. func (r *globalLimitRepository) GetGlobalLimit(ctx context.Context, id int64) (*model.GlobalLimit, error) {
  41. var globalLimit model.GlobalLimit
  42. return &globalLimit, nil
  43. }
  44. func (r *globalLimitRepository) AddGlobalLimit(ctx context.Context, req *model.GlobalLimit) error {
  45. if err := r.DB(ctx).Create(&req).Error; err != nil {
  46. return err
  47. }
  48. return nil
  49. }
  50. func (r *globalLimitRepository) UpdateGlobalLimitByHostId(ctx context.Context, req *model.GlobalLimit) error {
  51. if err := r.DB(ctx).Where("host_id = ?", req.HostId).Updates(&req).Error; err != nil {
  52. return err
  53. }
  54. return nil
  55. }
  56. func (r *globalLimitRepository) IsGlobalLimitExistByHostId(ctx context.Context, hostId int64) (bool, error) {
  57. var count int64
  58. err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ? AND state = false", hostId).Count(&count).Error
  59. if err != nil {
  60. return false, err
  61. }
  62. return count > 0, nil
  63. }
  64. func (r *globalLimitRepository) GetGlobalLimitByHostId(ctx context.Context, hostId int64) (*model.GlobalLimit, error) {
  65. var globalLimit model.GlobalLimit
  66. if err := r.DB(ctx).Where("host_id = ?", hostId).First(&globalLimit).Error; err != nil {
  67. return nil, err
  68. }
  69. return &globalLimit, nil
  70. }
  71. func (r *globalLimitRepository) GetGlobalLimitAllExpired(ctx context.Context,ids []int) ([]v1.GlobalLimitExpiredByHost, error) {
  72. var res []v1.GlobalLimitExpiredByHost
  73. threeDaysDuration := 30 * 24 * time.Hour
  74. targetTime := time.Now().Add(threeDaysDuration)
  75. targetTimestamp := targetTime.Unix()
  76. if err := r.DB(ctx).Table("shd_host").
  77. Where("id IN (?)", ids).
  78. Where("nextduedate < ?", targetTimestamp).
  79. Select("id", "uid", "nextduedate").
  80. Find(&res).
  81. Error; err != nil {
  82. return nil, err
  83. }
  84. return res, nil
  85. }
  86. func (r *globalLimitRepository) GetGlobalLimitAllHostId(ctx context.Context) ([]v1.GlobalLimitExpired, error) {
  87. var res []v1.GlobalLimitExpired
  88. if err := r.DB(ctx).Model(&model.GlobalLimit{}).
  89. Select("host_id", "rule_id","comment").
  90. Find(&res).Error; err != nil {
  91. return nil, err
  92. }
  93. return res, nil
  94. }
  95. func (r *globalLimitRepository) GetGlobalLimitFirst(ctx context.Context,uid int64) (*model.GlobalLimit, error) {
  96. var req model.GlobalLimit
  97. if err := r.DB(ctx).Where("uid = ?", uid).First(&req).Error; err != nil {
  98. return nil, err
  99. }
  100. return &req, nil
  101. }
  102. func (r *globalLimitRepository) GetUserInfo(ctx context.Context, uid int64) (v1.UserInfo, error) {
  103. var res v1.UserInfo
  104. if err := r.DB(ctx).Table("shd_clients").
  105. Where("id = ?", uid).
  106. Select("username", "email", "phonenumber").
  107. Find(&res).Error; err != nil {
  108. return v1.UserInfo{}, err
  109. }
  110. return res, nil
  111. }
  112. func (r *globalLimitRepository) GetHostName(ctx context.Context,hostId int64) (string, error) {
  113. var projectName string
  114. err := r.db.WithContext(ctx).Table("shd_host").
  115. Select("shd_products.name").
  116. Joins("JOIN shd_products ON shd_host.productid = shd_products.id").
  117. Where("shd_host.id = ?", hostId).
  118. Find(&projectName).Error
  119. if err != nil {
  120. if errors.Is(err, gorm.ErrRecordNotFound) {
  121. return "", fmt.Errorf("未找到 hostId 为 %d 的项目名称", hostId)
  122. }
  123. return "", fmt.Errorf("查询 host 和 project 名称失败: %w", err)
  124. }
  125. // 如果查询成功,返回项目名称
  126. return projectName, nil
  127. }
  128. func (r *globalLimitRepository) GetNodeId(ctx context.Context, cndWebId int) (int64, error) {
  129. var nodeId int64
  130. if err := r.DBWithName(ctx,"cdn").WithContext(ctx).Table("cloud_servers").Where("id = ?", cndWebId).Select("clusterId").Scan(&nodeId).Error; err != nil {
  131. return 0, err
  132. }
  133. return nodeId, nil
  134. }
  135. // 获取cdn套餐ID
  136. func (r *globalLimitRepository) GetNodeArea(ctx context.Context, nodeAreaName string) (int64, error) {
  137. var nodeId int64
  138. if err := r.DBWithName(ctx,"cdn").WithContext(ctx).Table("cloud_plans").Where("name = ?", nodeAreaName).Select("id").Scan(&nodeId).Error; err != nil {
  139. return 0, err
  140. }
  141. return nodeId, nil
  142. }
  143. func (r *globalLimitRepository) EditHostState(ctx context.Context, hostId int64, state bool) error {
  144. if err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ?", hostId).Update("state", state).Error; err != nil {
  145. return err
  146. }
  147. return nil
  148. }
  149. // 获取指定到期时间
  150. func (r *globalLimitRepository) GetGlobalLimitAlmostExpired(ctx context.Context, addTime int64) ([]model.GlobalLimit, error) {
  151. var res []model.GlobalLimit
  152. expiredTime := time.Now().Unix() + addTime
  153. if err := r.DB(ctx).
  154. Where("expired_At < ?", expiredTime).
  155. Find(&res).Error; err != nil {
  156. return nil, err
  157. }
  158. return res, nil
  159. }