Add projects
This commit is contained in:
66
pkg/db/postgres/projectsRepository.go
Normal file
66
pkg/db/postgres/projectsRepository.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package postgres
|
||||
|
||||
import (
|
||||
"context"
|
||||
"serverctl/pkg/application/projects"
|
||||
"serverctl/pkg/db"
|
||||
)
|
||||
|
||||
var _ projects.Repository = &ProjectsRepository{}
|
||||
|
||||
type ProjectsRepository struct {
|
||||
db *db.Client
|
||||
}
|
||||
|
||||
func NewProjectsRepository(db *db.Client) projects.Repository {
|
||||
return &ProjectsRepository{db: db}
|
||||
}
|
||||
|
||||
type projectData struct {
|
||||
Name string `json:"name"`
|
||||
MemberIds []int `json:"memberIds"`
|
||||
AdminIds []int `json:"adminIds"`
|
||||
}
|
||||
|
||||
func NewProjectData(project *projects.CreateProject) projectData {
|
||||
return projectData{
|
||||
Name: project.Name,
|
||||
AdminIds: project.AdminIds,
|
||||
MemberIds: project.MemberIds,
|
||||
}
|
||||
}
|
||||
|
||||
func (p ProjectsRepository) Create(ctx context.Context, project *projects.CreateProject) (int, error) {
|
||||
conn := p.db.GetConn(ctx)
|
||||
defer conn.Release()
|
||||
|
||||
var projectId int
|
||||
err := conn.QueryRow(ctx, "insert into sctl_project(data) values ($1) returning id", NewProjectData(project)).Scan(&projectId)
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
|
||||
return projectId, nil
|
||||
}
|
||||
|
||||
func (p ProjectsRepository) GetForMemberId(ctx context.Context, memberId int) ([]*projects.Project, error) {
|
||||
conn := p.db.GetConn(ctx)
|
||||
defer conn.Release()
|
||||
|
||||
rows, _ := conn.Query(ctx, "select id, data from sctl_project")
|
||||
projectsArr := make([]*projects.Project, 0)
|
||||
|
||||
for rows.Next() {
|
||||
var (
|
||||
id int
|
||||
data projectData
|
||||
)
|
||||
err := rows.Scan(&id, &data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
projectsArr = append(projectsArr, projects.NewProject(id, data.Name, data.MemberIds, data.AdminIds))
|
||||
}
|
||||
|
||||
return projectsArr, nil
|
||||
}
|
Reference in New Issue
Block a user