Skip to content

Commit

Permalink
Removed sync store for token/service/user/service-acc (#139)
Browse files Browse the repository at this point in the history
  • Loading branch information
marko-gacesa committed Dec 29, 2022
1 parent e4ae2b2 commit ae6e864
Show file tree
Hide file tree
Showing 11 changed files with 151 additions and 438 deletions.
48 changes: 30 additions & 18 deletions internal/store/database/service.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ package database

import (
"context"
"database/sql"
"fmt"

"github.com/harness/gitness/internal/store"
"github.com/harness/gitness/internal/store/database/dbtx"
"github.com/harness/gitness/types"
"github.com/harness/gitness/types/enum"

Expand Down Expand Up @@ -43,10 +43,13 @@ type ServiceStore struct {

// Find finds the service by id.
func (s *ServiceStore) Find(ctx context.Context, id int64) (*types.Service, error) {
db := dbtx.GetAccessor(ctx, s.db)

dst := new(service)
if err := s.db.GetContext(ctx, dst, serviceSelectID, id); err != nil {
if err := db.GetContext(ctx, dst, serviceSelectID, id); err != nil {
return nil, processSQLErrorf(err, "Select by id query failed")
}

return s.mapDBService(dst), nil
}

Expand All @@ -60,10 +63,13 @@ func (s *ServiceStore) FindUID(ctx context.Context, uid string) (*types.Service,
return nil, store.ErrResourceNotFound
}

db := dbtx.GetAccessor(ctx, s.db)

dst := new(service)
if err = s.db.GetContext(ctx, dst, serviceSelectUIDUnique, uidUnique); err != nil {
if err = db.GetContext(ctx, dst, serviceSelectUIDUnique, uidUnique); err != nil {
return nil, processSQLErrorf(err, "Select by uid query failed")
}

return s.mapDBService(dst), nil
}

Expand All @@ -74,12 +80,14 @@ func (s *ServiceStore) Create(ctx context.Context, svc *types.Service) error {
return fmt.Errorf("failed to map db service: %w", err)
}

query, arg, err := s.db.BindNamed(serviceInsert, dbSVC)
db := dbtx.GetAccessor(ctx, s.db)

query, arg, err := db.BindNamed(serviceInsert, dbSVC)
if err != nil {
return processSQLErrorf(err, "Failed to bind service object")
}

if err = s.db.QueryRowContext(ctx, query, arg...).Scan(&svc.ID); err != nil {
if err = db.QueryRowContext(ctx, query, arg...).Scan(&svc.ID); err != nil {
return processSQLErrorf(err, "Insert query failed")
}

Expand All @@ -93,12 +101,14 @@ func (s *ServiceStore) Update(ctx context.Context, svc *types.Service) error {
return fmt.Errorf("failed to map db service: %w", err)
}

query, arg, err := s.db.BindNamed(serviceUpdate, dbSVC)
db := dbtx.GetAccessor(ctx, s.db)

query, arg, err := db.BindNamed(serviceUpdate, dbSVC)
if err != nil {
return processSQLErrorf(err, "Failed to bind service object")
}

if _, err = s.db.ExecContext(ctx, query, arg...); err != nil {
if _, err = db.ExecContext(ctx, query, arg...); err != nil {
return processSQLErrorf(err, "Update query failed")
}

Expand All @@ -107,38 +117,40 @@ func (s *ServiceStore) Update(ctx context.Context, svc *types.Service) error {

// Delete deletes the service.
func (s *ServiceStore) Delete(ctx context.Context, id int64) error {
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return processSQLErrorf(err, "Failed to start a new transaction")
}
defer func(tx *sql.Tx) {
_ = tx.Rollback()
}(tx)
db := dbtx.GetAccessor(ctx, s.db)

// delete the service
if _, err = tx.ExecContext(ctx, serviceDelete, id); err != nil {
if _, err := db.ExecContext(ctx, serviceDelete, id); err != nil {
return processSQLErrorf(err, "The delete query failed")
}
return tx.Commit()

return nil
}

// List returns a list of service for a specific parent.
func (s *ServiceStore) List(ctx context.Context) ([]*types.Service, error) {
db := dbtx.GetAccessor(ctx, s.db)

dst := []*service{}

err := s.db.SelectContext(ctx, &dst, serviceSelect)
err := db.SelectContext(ctx, &dst, serviceSelect)
if err != nil {
return nil, processSQLErrorf(err, "Failed executing default list query")
}

return s.mapDBServices(dst), nil
}

// Count returns a count of service for a specific parent.
func (s *ServiceStore) Count(ctx context.Context) (int64, error) {
db := dbtx.GetAccessor(ctx, s.db)

var count int64
err := s.db.QueryRowContext(ctx, serviceCount).Scan(&count)
err := db.QueryRowContext(ctx, serviceCount).Scan(&count)
if err != nil {
return 0, processSQLErrorf(err, "Failed executing count query")
}

return count, nil
}

Expand Down
49 changes: 29 additions & 20 deletions internal/store/database/service_account.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ package database

import (
"context"
"database/sql"
"fmt"

"github.com/harness/gitness/internal/store"
"github.com/harness/gitness/internal/store/database/dbtx"
"github.com/harness/gitness/types"
"github.com/harness/gitness/types/enum"

Expand Down Expand Up @@ -43,8 +43,10 @@ type ServiceAccountStore struct {

// Find finds the service account by id.
func (s *ServiceAccountStore) Find(ctx context.Context, id int64) (*types.ServiceAccount, error) {
db := dbtx.GetAccessor(ctx, s.db)

dst := new(serviceAccount)
if err := s.db.GetContext(ctx, dst, serviceAccountSelectID, id); err != nil {
if err := db.GetContext(ctx, dst, serviceAccountSelectID, id); err != nil {
return nil, processSQLErrorf(err, "Select by id query failed")
}
return s.mapDBServiceAccount(dst), nil
Expand All @@ -60,10 +62,13 @@ func (s *ServiceAccountStore) FindUID(ctx context.Context, uid string) (*types.S
return nil, store.ErrResourceNotFound
}

db := dbtx.GetAccessor(ctx, s.db)

dst := new(serviceAccount)
if err = s.db.GetContext(ctx, dst, serviceAccountSelectUIDUnique, uidUnique); err != nil {
if err = db.GetContext(ctx, dst, serviceAccountSelectUIDUnique, uidUnique); err != nil {
return nil, processSQLErrorf(err, "Select by uid query failed")
}

return s.mapDBServiceAccount(dst), nil
}

Expand All @@ -74,12 +79,14 @@ func (s *ServiceAccountStore) Create(ctx context.Context, sa *types.ServiceAccou
return fmt.Errorf("failed to map db service account: %w", err)
}

query, arg, err := s.db.BindNamed(serviceAccountInsert, dbSA)
db := dbtx.GetAccessor(ctx, s.db)

query, arg, err := db.BindNamed(serviceAccountInsert, dbSA)
if err != nil {
return processSQLErrorf(err, "Failed to bind service account object")
}

if err = s.db.QueryRowContext(ctx, query, arg...).Scan(&sa.ID); err != nil {
if err = db.QueryRowContext(ctx, query, arg...).Scan(&sa.ID); err != nil {
return processSQLErrorf(err, "Insert query failed")
}

Expand All @@ -93,12 +100,14 @@ func (s *ServiceAccountStore) Update(ctx context.Context, sa *types.ServiceAccou
return fmt.Errorf("failed to map db service account: %w", err)
}

query, arg, err := s.db.BindNamed(serviceAccountUpdate, dbSA)
db := dbtx.GetAccessor(ctx, s.db)

query, arg, err := db.BindNamed(serviceAccountUpdate, dbSA)
if err != nil {
return processSQLErrorf(err, "Failed to bind service account object")
}

if _, err = s.db.ExecContext(ctx, query, arg...); err != nil {
if _, err = db.ExecContext(ctx, query, arg...); err != nil {
return processSQLErrorf(err, "Update query failed")
}

Expand All @@ -107,40 +116,40 @@ func (s *ServiceAccountStore) Update(ctx context.Context, sa *types.ServiceAccou

// Delete deletes the service account.
func (s *ServiceAccountStore) Delete(ctx context.Context, id int64) error {
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return processSQLErrorf(err, "Failed to start a new transaction")
}
defer func(tx *sql.Tx) {
_ = tx.Rollback()
}(tx)
// delete the service account
if _, err = tx.ExecContext(ctx, serviceAccountDelete, id); err != nil {
db := dbtx.GetAccessor(ctx, s.db)

if _, err := db.ExecContext(ctx, serviceAccountDelete, id); err != nil {
return processSQLErrorf(err, "The delete query failed")
}
return tx.Commit()

return nil
}

// List returns a list of service accounts for a specific parent.
func (s *ServiceAccountStore) List(ctx context.Context, parentType enum.ParentResourceType,
parentID int64) ([]*types.ServiceAccount, error) {
dst := []*serviceAccount{}
db := dbtx.GetAccessor(ctx, s.db)

err := s.db.SelectContext(ctx, &dst, serviceAccountSelectByParentTypeAndID, parentType, parentID)
dst := []*serviceAccount{}
err := db.SelectContext(ctx, &dst, serviceAccountSelectByParentTypeAndID, parentType, parentID)
if err != nil {
return nil, processSQLErrorf(err, "Failed executing default list query")
}

return s.mapDBServiceAccounts(dst), nil
}

// Count returns a count of service accounts for a specific parent.
func (s *ServiceAccountStore) Count(ctx context.Context,
parentType enum.ParentResourceType, parentID int64) (int64, error) {
db := dbtx.GetAccessor(ctx, s.db)

var count int64
err := s.db.QueryRowContext(ctx, serviceAccountCountByParentTypeAndID, parentType, parentID).Scan(&count)
err := db.QueryRowContext(ctx, serviceAccountCountByParentTypeAndID, parentType, parentID).Scan(&count)
if err != nil {
return 0, processSQLErrorf(err, "Failed executing count query")
}

return count, nil
}

Expand Down
79 changes: 0 additions & 79 deletions internal/store/database/service_account_sync.go

This file was deleted.

Loading

0 comments on commit ae6e864

Please sign in to comment.