fleets/storage/storage.go

package storage

import (
	"fmt"

	"github.com/spf13/viper"
)

type Storage interface {
	//Vehicles management
	CreateVehicle(Vehicle) error
	GetVehicle(id string) (*Vehicle, error)
	GetVehicles(namespaces []string) ([]Vehicle, error)
	UpdateVehicle(Vehicle) error

	//Bookings management
	CreateBooking(Booking) error
	UpdateBooking(Booking) error
	GetBooking(id string) (*Booking, error)
	GetBookings() ([]Booking, error)
	GetBookingsForVehicle(vehicleid string) ([]Booking, error)
	GetBookingsForDriver(driver string) ([]Booking, error)
	DeleteBooking(id string) error
}

func NewStorage(cfg *viper.Viper) (Storage, error) {
	var (
		storage_type = cfg.GetString("storage.db.type")
	)

	switch storage_type {
	case "mongodb":
		s, err := NewMongoDBStorage(cfg)
		return s, err
	case "psql":
		s, err := NewPostgresqlStorage(cfg)
		return s, err
	default:
		return nil, fmt.Errorf("storage type %v is not supported", storage_type)
	}

}