serverctl/services/entry/pkg/db/postgres/applicationsRepository.go

62 lines
1.7 KiB
Go
Raw Normal View History

2022-02-16 15:30:38 +01:00
package postgres
import (
"context"
"errors"
"github.com/jackc/pgx/v4"
"go.uber.org/zap"
"serverctl/pkg/application/applications"
"serverctl/pkg/db"
)
type ApplicationRepository struct {
db *db.Client
logger *zap.Logger
}
var _ applications.Repository = ApplicationRepository{}
func NewApplicationRepository(logger *zap.Logger, db *db.Client) applications.Repository {
return &ApplicationRepository{logger: logger, db: db}
}
func (a ApplicationRepository) CreateApplication(ctx context.Context, name string, userId int, projectId int) (int, error) {
conn := a.db.GetConn(ctx)
defer conn.Release()
var applicationId int
err := conn.BeginTxFunc(ctx, pgx.TxOptions{}, func(tx pgx.Tx) error {
var exists bool
err := tx.QueryRow(
ctx,
"select exists(select 1 from sctl_project_member where project_id = $1 and member_id = $2 and role = 'admin')", projectId, userId,
).Scan(&exists)
if err != nil {
a.logger.Info("cannot query project member status in database",
zap.Int("userId", userId),
zap.Int("projectId", projectId),
zap.String("error", err.Error()))
return err
}
if !exists {
a.logger.Info("cannot create application as user isn't admin for project, or project doesn't exist",
zap.Int("userId", userId),
zap.Int("projectId", projectId))
return errors.New("user isn't admin or admin of project")
}
err = tx.QueryRow(ctx, "insert into sctl_application(name, project_id) values($1, $2) returning id", name, projectId).Scan(&applicationId)
if err != nil {
a.logger.Info("could not create application", zap.String("error", err.Error()))
return err
}
return nil
})
if err != nil {
return -1, err
}
return applicationId, nil
}