72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"embed"
|
|
"errors"
|
|
|
|
"go.unistack.org/unistack-org/pkgdash/models"
|
|
pb "go.unistack.org/unistack-org/pkgdash/proto"
|
|
"go.unistack.org/unistack-org/pkgdash/storage/postgres"
|
|
"go.unistack.org/unistack-org/pkgdash/storage/sqlite"
|
|
)
|
|
|
|
//go:embed migrations
|
|
var fs embed.FS
|
|
|
|
var (
|
|
storages = map[string]func(*sql.DB, embed.FS) interface{}{
|
|
"postgres": postgres.NewStorage(),
|
|
"sqlite": sqlite.NewStorage(),
|
|
}
|
|
)
|
|
|
|
type contextKey string
|
|
|
|
var (
|
|
storeIdent = contextKey("store")
|
|
)
|
|
|
|
type Migrate interface {
|
|
MigrateUp() error
|
|
MigrateDown() error
|
|
}
|
|
|
|
type Storage interface {
|
|
Migrate
|
|
|
|
ListPackage(ctx context.Context) (models.ListPackage, error)
|
|
UpdatePackage(ctx context.Context, req *pb.UpdatePackageReq) error
|
|
AddComment(ctx context.Context, req *pb.AddCommentReq) (uint64, error)
|
|
AddPackage(ctx context.Context, req *pb.AddPackageReq) error
|
|
InsertButchModules(ctx context.Context, req []models.Module) ([]uint64, error)
|
|
GetModule(ctx context.Context, req *pb.GetModuleReq) (models.ListModule, error)
|
|
GetComment(ctx context.Context, req *pb.GetCommentsReq) (models.ListComment, error)
|
|
}
|
|
|
|
func NewStorage(name string, db *sql.DB) (Storage, error) {
|
|
function, ok := storages[name]
|
|
if !ok {
|
|
return nil, errors.New("incorrect name store")
|
|
}
|
|
store := function(db, fs)
|
|
database, ok := store.(Storage)
|
|
if !ok {
|
|
return nil, errors.New("dont implements interface Storage")
|
|
}
|
|
return database, nil
|
|
}
|
|
|
|
func InContext(ctx context.Context, val Storage) context.Context {
|
|
return context.WithValue(ctx, storeIdent, val)
|
|
}
|
|
|
|
func FromContext(ctx context.Context) (Storage, error) {
|
|
if store, ok := ctx.Value(storeIdent).(Storage); !ok {
|
|
return nil, errors.New("empty store")
|
|
} else {
|
|
return store, nil
|
|
}
|
|
}
|