engigu
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package services
|
|
|
|
import (
|
|
"baihu/internal/database"
|
|
"baihu/internal/models"
|
|
)
|
|
|
|
type EnvService struct{}
|
|
|
|
func NewEnvService() *EnvService {
|
|
return &EnvService{}
|
|
}
|
|
|
|
func (es *EnvService) CreateEnvVar(name, value, remark string, userID int) *models.EnvironmentVariable {
|
|
env := &models.EnvironmentVariable{
|
|
Name: name,
|
|
Value: value,
|
|
Remark: remark,
|
|
UserID: uint(userID),
|
|
}
|
|
database.DB.Create(env)
|
|
return env
|
|
}
|
|
|
|
func (es *EnvService) GetEnvVarsByUserID(userID int) []models.EnvironmentVariable {
|
|
var envs []models.EnvironmentVariable
|
|
database.DB.Where("user_id = ?", userID).Find(&envs)
|
|
return envs
|
|
}
|
|
|
|
func (es *EnvService) GetEnvVarsWithPagination(userID int, name string, page, pageSize int) ([]models.EnvironmentVariable, int64) {
|
|
var envs []models.EnvironmentVariable
|
|
var total int64
|
|
|
|
query := database.DB.Model(&models.EnvironmentVariable{}).Where("user_id = ?", userID)
|
|
if name != "" {
|
|
query = query.Where("name LIKE ?", "%"+name+"%")
|
|
}
|
|
|
|
query.Count(&total)
|
|
query.Order("id DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&envs)
|
|
return envs, total
|
|
}
|
|
|
|
func (es *EnvService) GetEnvVarByID(id int) *models.EnvironmentVariable {
|
|
var env models.EnvironmentVariable
|
|
if err := database.DB.First(&env, id).Error; err != nil {
|
|
return nil
|
|
}
|
|
return &env
|
|
}
|
|
|
|
func (es *EnvService) UpdateEnvVar(id int, name, value, remark string) *models.EnvironmentVariable {
|
|
var env models.EnvironmentVariable
|
|
if err := database.DB.First(&env, id).Error; err != nil {
|
|
return nil
|
|
}
|
|
env.Name = name
|
|
env.Value = value
|
|
env.Remark = remark
|
|
database.DB.Save(&env)
|
|
return &env
|
|
}
|
|
|
|
func (es *EnvService) DeleteEnvVar(id int) bool {
|
|
result := database.DB.Delete(&models.EnvironmentVariable{}, id)
|
|
return result.RowsAffected > 0
|
|
}
|