50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/cloudprint/cloudprint-backend/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type AdminRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewAdminRepository(db *gorm.DB) *AdminRepository {
|
|
return &AdminRepository{db: db}
|
|
}
|
|
|
|
func (r *AdminRepository) Create(ctx context.Context, admin *model.Admin) error {
|
|
return r.db.WithContext(ctx).Create(admin).Error
|
|
}
|
|
|
|
func (r *AdminRepository) GetByID(ctx context.Context, id int32) (*model.Admin, error) {
|
|
var admin model.Admin
|
|
err := r.db.WithContext(ctx).First(&admin, id).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
func (r *AdminRepository) GetByUsername(ctx context.Context, username string) (*model.Admin, error) {
|
|
var admin model.Admin
|
|
err := r.db.WithContext(ctx).Where("username = ?", username).First(&admin).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
func (r *AdminRepository) Update(ctx context.Context, admin *model.Admin) error {
|
|
return r.db.WithContext(ctx).Save(admin).Error
|
|
}
|