84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"acc-server-manager/local/model"
|
|
"context"
|
|
"errors"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ServerRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewServerRepository(db *gorm.DB) *ServerRepository {
|
|
return &ServerRepository{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// GetFirst
|
|
// Gets first row from Server table.
|
|
//
|
|
// Args:
|
|
// context.Context: Application context
|
|
// Returns:
|
|
// model.ServerModel: Server object from database.
|
|
func (as ServerRepository) GetFirst(ctx context.Context, serverId int) *model.Server {
|
|
db := as.db.WithContext(ctx)
|
|
ServerModel := new(model.Server)
|
|
db.Where("id=?", serverId).First(&ServerModel)
|
|
return ServerModel
|
|
}
|
|
|
|
// GetFirstByServiceName
|
|
// Gets first row from Server table.
|
|
//
|
|
// Args:
|
|
// context.Context: Application context
|
|
// Returns:
|
|
// model.ServerModel: Server object from database.
|
|
func (as ServerRepository) GetFirstByServiceName(ctx context.Context, serviceName string) *model.Server {
|
|
db := as.db.WithContext(ctx)
|
|
ServerModel := new(model.Server)
|
|
result := db.Where("service_name=?", serviceName).First(&ServerModel)
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
return nil
|
|
}
|
|
return ServerModel
|
|
}
|
|
|
|
// GetAll
|
|
// Gets All rows from Server table.
|
|
//
|
|
// Args:
|
|
// context.Context: Application context
|
|
// Returns:
|
|
// model.ServerModel: Server object from database.
|
|
func (as ServerRepository) GetAll(ctx context.Context) *[]model.Server {
|
|
db := as.db.WithContext(ctx)
|
|
ServerModel := new([]model.Server)
|
|
db.Find(&ServerModel)
|
|
return ServerModel
|
|
}
|
|
|
|
// UpdateServer
|
|
// Updates Server row from Server table.
|
|
//
|
|
// Args:
|
|
// context.Context: Application context
|
|
// Returns:
|
|
// model.Server: Server object from database.
|
|
func (as ServerRepository) UpdateServer(ctx context.Context, body *model.Server) *model.Server {
|
|
db := as.db.WithContext(ctx)
|
|
|
|
existingServer := new(model.Server)
|
|
result := db.Where("id=?", body.ID).First(existingServer)
|
|
if !errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
body.ID = existingServer.ID
|
|
}
|
|
db.Save(body)
|
|
return body
|
|
}
|