133 lines
3.6 KiB
Go
133 lines
3.6 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package satellitedb
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/skyrings/skyring-common/tools/uuid"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/pkg/satellite"
|
|
"storj.io/storj/pkg/satellite/satellitedb/dbx"
|
|
"storj.io/storj/pkg/utils"
|
|
)
|
|
|
|
// implementation of Projects interface repository using spacemonkeygo/dbx orm
|
|
type projects struct {
|
|
db dbx.Methods
|
|
}
|
|
|
|
// GetAll is a method for querying all projects from the database.
|
|
func (projects *projects) GetAll(ctx context.Context) ([]satellite.Project, error) {
|
|
projectsDbx, err := projects.db.All_Project(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return projectsFromDbxSlice(projectsDbx)
|
|
}
|
|
|
|
// GetByUserID is a method for querying all projects from the database by userID.
|
|
func (projects *projects) GetByUserID(ctx context.Context, userID uuid.UUID) ([]satellite.Project, error) {
|
|
projectsDbx, err := projects.db.All_Project_By_ProjectMember_MemberId(ctx, dbx.ProjectMember_MemberId(userID[:]))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return projectsFromDbxSlice(projectsDbx)
|
|
}
|
|
|
|
// Get is a method for querying project from the database by id.
|
|
func (projects *projects) Get(ctx context.Context, id uuid.UUID) (*satellite.Project, error) {
|
|
project, err := projects.db.Get_Project_By_Id(ctx, dbx.Project_Id(id[:]))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return projectFromDBX(project)
|
|
}
|
|
|
|
// Insert is a method for inserting project into the database.
|
|
func (projects *projects) Insert(ctx context.Context, project *satellite.Project) (*satellite.Project, error) {
|
|
projectID, err := uuid.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
createdProject, err := projects.db.Create_Project(ctx,
|
|
dbx.Project_Id(projectID[:]),
|
|
dbx.Project_Name(project.Name),
|
|
dbx.Project_Description(project.Description),
|
|
dbx.Project_TermsAccepted(project.TermsAccepted))
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return projectFromDBX(createdProject)
|
|
}
|
|
|
|
// Delete is a method for deleting project by Id from the database.
|
|
func (projects *projects) Delete(ctx context.Context, id uuid.UUID) error {
|
|
_, err := projects.db.Delete_Project_By_Id(ctx, dbx.Project_Id(id[:]))
|
|
|
|
return err
|
|
}
|
|
|
|
// Update is a method for updating project entity
|
|
func (projects *projects) Update(ctx context.Context, project *satellite.Project) error {
|
|
updateFields := dbx.Project_Update_Fields{
|
|
Description: dbx.Project_Description(project.Description),
|
|
TermsAccepted: dbx.Project_TermsAccepted(project.TermsAccepted),
|
|
}
|
|
|
|
_, err := projects.db.Update_Project_By_Id(ctx,
|
|
dbx.Project_Id(project.ID[:]),
|
|
updateFields)
|
|
|
|
return err
|
|
}
|
|
|
|
// projectFromDBX is used for creating Project entity from autogenerated dbx.Project struct
|
|
func projectFromDBX(project *dbx.Project) (*satellite.Project, error) {
|
|
if project == nil {
|
|
return nil, errs.New("project parameter is nil")
|
|
}
|
|
|
|
id, err := bytesToUUID(project.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
u := &satellite.Project{
|
|
ID: id,
|
|
Name: project.Name,
|
|
Description: project.Description,
|
|
TermsAccepted: project.TermsAccepted,
|
|
CreatedAt: project.CreatedAt,
|
|
}
|
|
|
|
return u, nil
|
|
}
|
|
|
|
// projectsFromDbxSlice is used for creating []Project entities from autogenerated []*dbx.Project struct
|
|
func projectsFromDbxSlice(projectsDbx []*dbx.Project) ([]satellite.Project, error) {
|
|
var projects []satellite.Project
|
|
var errors []error
|
|
|
|
// Generating []dbo from []dbx and collecting all errors
|
|
for _, projectDbx := range projectsDbx {
|
|
project, err := projectFromDBX(projectDbx)
|
|
if err != nil {
|
|
errors = append(errors, err)
|
|
continue
|
|
}
|
|
|
|
projects = append(projects, *project)
|
|
}
|
|
|
|
return projects, utils.CombineErrors(errors...)
|
|
}
|