add filtering and base repository
This commit is contained in:
@@ -9,28 +9,15 @@ import (
|
||||
)
|
||||
|
||||
type ServerRepository struct {
|
||||
db *gorm.DB
|
||||
*BaseRepository[model.Server, model.ServerFilter]
|
||||
}
|
||||
|
||||
func NewServerRepository(db *gorm.DB) *ServerRepository {
|
||||
return &ServerRepository{
|
||||
db: db,
|
||||
BaseRepository: NewBaseRepository[model.Server, model.ServerFilter](db, model.Server{}),
|
||||
}
|
||||
}
|
||||
|
||||
// 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.
|
||||
@@ -39,45 +26,13 @@ func (as ServerRepository) GetFirst(ctx context.Context, serverId int) *model.Se
|
||||
// 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
|
||||
func (r *ServerRepository) GetFirstByServiceName(ctx context.Context, serviceName string) (*model.Server, error) {
|
||||
result := new(model.Server)
|
||||
if err := r.db.WithContext(ctx).Where("service_name = ?", serviceName).First(result).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, err
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
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
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
Reference in New Issue
Block a user