65 lines
1.8 KiB
Go
65 lines
1.8 KiB
Go
package databases
|
|
|
|
// go generate: mockery --name DeviceDatabase
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/SowinskiBraeden/DeviceBookingAPI/models"
|
|
)
|
|
|
|
const deviceDBO = "devices"
|
|
|
|
// DeviceDatabase contains the methods to use with the cow database
|
|
type DeviceDatabase interface {
|
|
FindOne(ctx context.Context, filter interface{}) (*models.Device, error)
|
|
Find(ctx context.Context, filter interface{}) ([]models.Device, error)
|
|
InsertOne(ctx context.Context, document interface{}) (*mongoInsertOneResult, error)
|
|
UpdateOne(ctx context.Context, filter, document interface{}) (*mongoUpdateResult, error)
|
|
}
|
|
|
|
type deviceDatabase struct {
|
|
db DatabaseHelper
|
|
}
|
|
|
|
// NewDeviceDatabase initializes a new instance of a device database with the provided db connection
|
|
func NewDeviceDatabase(db DatabaseHelper) DeviceDatabase {
|
|
return &deviceDatabase{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (d *deviceDatabase) FindOne(ctx context.Context, filter interface{}) (*models.Device, error) {
|
|
device := &models.Device{}
|
|
err := d.db.Collection(deviceDBO).FindOne(ctx, filter).Decode(&device)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return device, nil
|
|
}
|
|
|
|
func (d *deviceDatabase) Find(ctx context.Context, filter interface{}) ([]models.Device, error) {
|
|
var devices []models.Device
|
|
err := d.db.Collection(deviceDBO).FindOne(ctx, filter).Decode(&devices)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return devices, nil
|
|
}
|
|
|
|
func (d *deviceDatabase) InsertOne(ctx context.Context, document interface{}) (*mongoInsertOneResult, error) {
|
|
result, err := d.db.Collection(deviceDBO).InsertOne(ctx, document)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
func (d *deviceDatabase) UpdateOne(ctx context.Context, filter, update interface{}) (*mongoUpdateResult, error) {
|
|
result, err := d.db.Collection(deviceDBO).UpdateOne(ctx, filter, update)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|