60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"users_management/m/model/entity"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type FishboneRepo interface {
|
|
Post(fishbone entity.Fishbone) error
|
|
GetAll() ([]entity.Fishbone, error)
|
|
Update(id uuid.UUID,updates map[string]interface{}) error
|
|
|
|
GetByID(id uuid.UUID) (entity.Fishbone, error)
|
|
}
|
|
|
|
type fishboneRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewFishboneRepo(db *gorm.DB) FishboneRepo {
|
|
return &fishboneRepo{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *fishboneRepo) Post(fishbone entity.Fishbone) error {
|
|
err := r.db.Create(&fishbone).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *fishboneRepo) GetAll() ([]entity.Fishbone, error) {
|
|
var fishbones []entity.Fishbone
|
|
err := r.db.Preload("DeviceStart").Preload("DeviceEnd").Preload("Backbone").Preload("Backbone.DeviceStart").Preload("Backbone.DeviceEnd").Find(&fishbones).Error
|
|
if err != nil {
|
|
return fishbones, err
|
|
}
|
|
return fishbones, nil
|
|
}
|
|
|
|
func (r *fishboneRepo) Update(id uuid.UUID,updates map[string]interface{}) error {
|
|
err := r.db.Model(&entity.Fishbone{}).Where("id = ?", id).Updates(updates).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *fishboneRepo) GetByID(id uuid.UUID) (entity.Fishbone, error) {
|
|
var fishbone entity.Fishbone
|
|
err := r.db.Where("id = ?", id).Preload("DeviceStart").Preload("DeviceEnd").Preload("Backbone.DeviceStart").Preload("Backbone.DeviceEnd").First(&fishbone).Error
|
|
if err != nil {
|
|
return fishbone, err
|
|
}
|
|
return fishbone, nil
|
|
} |