mirror of https://github.com/harness/drone.git
341 lines
12 KiB
Go
341 lines
12 KiB
Go
// Copyright 2023 Harness, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/harness/gitness/app/store"
|
|
"github.com/harness/gitness/store/database"
|
|
"github.com/harness/gitness/store/database/dbtx"
|
|
"github.com/harness/gitness/types"
|
|
"github.com/harness/gitness/types/enum"
|
|
|
|
"github.com/Masterminds/squirrel"
|
|
"github.com/guregu/null"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
infraProviderResourceIDColumn = `ipreso_id`
|
|
infraProviderResourceInsertColumns = `
|
|
ipreso_uid,
|
|
ipreso_display_name,
|
|
ipreso_infra_provider_config_id,
|
|
ipreso_type,
|
|
ipreso_space_id,
|
|
ipreso_created,
|
|
ipreso_updated,
|
|
ipreso_cpu,
|
|
ipreso_memory,
|
|
ipreso_disk,
|
|
ipreso_network,
|
|
ipreso_region,
|
|
ipreso_opentofu_params
|
|
`
|
|
infraProviderResourceSelectColumns = "ipreso_id," + infraProviderResourceInsertColumns
|
|
infraProviderResourceTable = `infra_provider_resources`
|
|
)
|
|
|
|
type infraProviderResource struct {
|
|
ID int64 `db:"ipreso_id"`
|
|
Identifier string `db:"ipreso_uid"`
|
|
Name string `db:"ipreso_display_name"`
|
|
InfraProviderConfigID int64 `db:"ipreso_infra_provider_config_id"`
|
|
InfraProviderType enum.InfraProviderType `db:"ipreso_type"`
|
|
SpaceID int64 `db:"ipreso_space_id"`
|
|
CPU null.String `db:"ipreso_cpu"`
|
|
Memory null.String `db:"ipreso_memory"`
|
|
Disk null.String `db:"ipreso_disk"`
|
|
Network null.String `db:"ipreso_network"`
|
|
Region string `db:"ipreso_region"` // need list maybe
|
|
OpenTofuParams []byte `db:"ipreso_opentofu_params"`
|
|
Created int64 `db:"ipreso_created"`
|
|
Updated int64 `db:"ipreso_updated"`
|
|
}
|
|
|
|
var _ store.InfraProviderResourceStore = (*infraProviderResourceStore)(nil)
|
|
|
|
// NewGitspaceConfigStore returns a new GitspaceConfigStore.
|
|
func NewInfraProviderResourceStore(db *sqlx.DB) store.InfraProviderResourceStore {
|
|
return &infraProviderResourceStore{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
type infraProviderResourceStore struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func (s infraProviderResourceStore) List(ctx context.Context, infraProviderConfigID int64,
|
|
_ types.ListQueryFilter) ([]*types.InfraProviderResource, error) {
|
|
stmt := database.Builder.
|
|
Select(infraProviderResourceSelectColumns).
|
|
From(infraProviderResourceTable).
|
|
Where("ipreso_infra_provider_config_id = $1", infraProviderConfigID)
|
|
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
dst := new([]infraProviderResource)
|
|
if err := db.SelectContext(ctx, dst, sql, args...); err != nil {
|
|
return nil, database.ProcessSQLErrorf(ctx, err, "Failed to list infraprovider resources")
|
|
}
|
|
return mapToInfraProviderResources(ctx, *dst)
|
|
}
|
|
|
|
func (s infraProviderResourceStore) Find(ctx context.Context, id int64) (*types.InfraProviderResource, error) {
|
|
stmt := database.Builder.
|
|
Select(infraProviderResourceSelectColumns).
|
|
From(infraProviderResourceTable).
|
|
Where(infraProviderResourceIDColumn+" = $1", id)
|
|
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
dst := new(infraProviderResource)
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
if err := db.GetContext(ctx, dst, sql, args...); err != nil {
|
|
return nil, database.ProcessSQLErrorf(ctx, err, "Failed to find infraprovider resource %d", id)
|
|
}
|
|
return mapToInfraProviderResource(ctx, dst)
|
|
}
|
|
|
|
func (s infraProviderResourceStore) FindByIdentifier(
|
|
ctx context.Context,
|
|
spaceID int64,
|
|
identifier string,
|
|
) (*types.InfraProviderResource, error) {
|
|
stmt := database.Builder.
|
|
Select(infraProviderResourceSelectColumns).
|
|
From(infraProviderResourceTable).
|
|
Where("ipreso_uid = $1", identifier).
|
|
Where("ipreso_space_id = $2", spaceID)
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
dst := new(infraProviderResource)
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
if err := db.GetContext(ctx, dst, sql, args...); err != nil {
|
|
return nil, database.ProcessSQLErrorf(ctx, err, "Failed to find infraprovider resource %s", identifier)
|
|
}
|
|
return mapToInfraProviderResource(ctx, dst)
|
|
}
|
|
|
|
func (s infraProviderResourceStore) Create(
|
|
ctx context.Context,
|
|
infraProviderResource *types.InfraProviderResource,
|
|
) error {
|
|
jsonBytes, marshalErr := json.Marshal(infraProviderResource.Metadata)
|
|
if marshalErr != nil {
|
|
return marshalErr
|
|
}
|
|
stmt := database.Builder.
|
|
Insert(infraProviderResourceTable).
|
|
Columns(infraProviderResourceInsertColumns).
|
|
Values(
|
|
infraProviderResource.UID,
|
|
infraProviderResource.Name,
|
|
infraProviderResource.InfraProviderConfigID,
|
|
infraProviderResource.InfraProviderType,
|
|
infraProviderResource.SpaceID,
|
|
infraProviderResource.Created,
|
|
infraProviderResource.Updated,
|
|
infraProviderResource.CPU,
|
|
infraProviderResource.Memory,
|
|
infraProviderResource.Disk,
|
|
infraProviderResource.Network,
|
|
infraProviderResource.Region,
|
|
jsonBytes,
|
|
).
|
|
Suffix(ReturningClause + infraProviderResourceIDColumn)
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
if err = db.QueryRowContext(ctx, sql, args...).Scan(&infraProviderResource.ID); err != nil {
|
|
return database.ProcessSQLErrorf(
|
|
ctx, err, "infra provider resource create failed %s", infraProviderResource.UID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s infraProviderResourceStore) Update(
|
|
ctx context.Context,
|
|
infraProviderResource *types.InfraProviderResource,
|
|
) error {
|
|
dbinfraProviderResource, err := s.mapToInternalInfraProviderResource(ctx, infraProviderResource)
|
|
if err != nil {
|
|
return fmt.Errorf(
|
|
"failed to map to DB Obj for infraprovider resource %s", infraProviderResource.UID)
|
|
}
|
|
stmt := database.Builder.
|
|
Update(infraProviderResourceTable).
|
|
Set("ipreso_display_name", dbinfraProviderResource.Name).
|
|
Set("ipreso_memory", dbinfraProviderResource.Memory).
|
|
Set("ipreso_disk", dbinfraProviderResource.Disk).
|
|
Set("ipreso_network", dbinfraProviderResource.Network).
|
|
Set("ipreso_region", dbinfraProviderResource.Region).
|
|
Set("ipreso_opentofu_params", dbinfraProviderResource.OpenTofuParams).
|
|
Set("ipreso_updated", dbinfraProviderResource.Updated).
|
|
Where("ipreso_id = ?", infraProviderResource.ID)
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
if _, err := db.ExecContext(ctx, sql, args...); err != nil {
|
|
return database.ProcessSQLErrorf(
|
|
ctx, err, "Failed to update infraprovider resource %s", infraProviderResource.UID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s infraProviderResourceStore) DeleteByIdentifier(ctx context.Context, spaceID int64, identifier string) error {
|
|
stmt := database.Builder.
|
|
Delete(infraProviderResourceTable).
|
|
Where("ipreso_uid = $1", identifier).
|
|
Where("ipreso_space_id = $2", spaceID)
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
db := dbtx.GetAccessor(ctx, s.db)
|
|
if _, err := db.ExecContext(ctx, sql, args...); err != nil {
|
|
return database.ProcessSQLErrorf(
|
|
ctx, err, "Failed to delete infra provider resource %s", identifier)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func mapToInfraProviderResource(_ context.Context,
|
|
in *infraProviderResource) (*types.InfraProviderResource, error) {
|
|
metadataParamsMap := make(map[string]string)
|
|
marshalErr := json.Unmarshal(in.OpenTofuParams, &metadataParamsMap)
|
|
if marshalErr != nil {
|
|
return nil, marshalErr
|
|
}
|
|
return &types.InfraProviderResource{
|
|
UID: in.Identifier,
|
|
InfraProviderConfigID: in.InfraProviderConfigID,
|
|
ID: in.ID,
|
|
InfraProviderType: in.InfraProviderType,
|
|
Name: in.Name,
|
|
SpaceID: in.SpaceID,
|
|
CPU: in.CPU.Ptr(),
|
|
Memory: in.Memory.Ptr(),
|
|
Disk: in.Disk.Ptr(),
|
|
Network: in.Network.Ptr(),
|
|
Region: in.Region,
|
|
Metadata: metadataParamsMap,
|
|
Created: in.Created,
|
|
Updated: in.Updated,
|
|
}, nil
|
|
}
|
|
|
|
func (s infraProviderResourceStore) mapToInternalInfraProviderResource(_ context.Context,
|
|
in *types.InfraProviderResource) (*infraProviderResource, error) {
|
|
jsonBytes, marshalErr := json.Marshal(in.Metadata)
|
|
if marshalErr != nil {
|
|
return nil, marshalErr
|
|
}
|
|
return &infraProviderResource{
|
|
Identifier: in.UID,
|
|
InfraProviderConfigID: in.InfraProviderConfigID,
|
|
InfraProviderType: in.InfraProviderType,
|
|
Name: in.Name,
|
|
SpaceID: in.SpaceID,
|
|
CPU: null.StringFromPtr(in.CPU),
|
|
Memory: null.StringFromPtr(in.Memory),
|
|
Disk: null.StringFromPtr(in.Disk),
|
|
Network: null.StringFromPtr(in.Network),
|
|
Region: in.Region,
|
|
OpenTofuParams: jsonBytes,
|
|
Created: in.Created,
|
|
Updated: in.Updated,
|
|
}, nil
|
|
}
|
|
|
|
func mapToInfraProviderResources(ctx context.Context,
|
|
resources []infraProviderResource) ([]*types.InfraProviderResource, error) {
|
|
var err error
|
|
res := make([]*types.InfraProviderResource, len(resources))
|
|
for i := range resources {
|
|
res[i], err = mapToInfraProviderResource(ctx, &resources[i])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
var _ store.InfraProviderResourceView = (*InfraProviderResourceView)(nil)
|
|
|
|
// NewInfraProviderResourceView returns a new InfraProviderResourceView.
|
|
// It's used by the infraprovider resource cache.
|
|
func NewInfraProviderResourceView(db *sqlx.DB) *InfraProviderResourceView {
|
|
return &InfraProviderResourceView{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
type InfraProviderResourceView struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func (i InfraProviderResourceView) Find(ctx context.Context, id int64) (*types.InfraProviderResource, error) {
|
|
stmt := database.Builder.
|
|
Select(infraProviderResourceSelectColumns).
|
|
From(infraProviderResourceTable).
|
|
Where(infraProviderResourceIDColumn+" = $1", id)
|
|
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
dst := new(infraProviderResource)
|
|
db := dbtx.GetAccessor(ctx, i.db)
|
|
if err := db.GetContext(ctx, dst, sql, args...); err != nil {
|
|
return nil, database.ProcessSQLErrorf(ctx, err, "Failed to find infraprovider resource %d", id)
|
|
}
|
|
return mapToInfraProviderResource(ctx, dst)
|
|
}
|
|
|
|
func (i InfraProviderResourceView) FindMany(ctx context.Context, ids []int64) ([]*types.InfraProviderResource, error) {
|
|
stmt := database.Builder.
|
|
Select(infraProviderResourceSelectColumns).
|
|
From(infraProviderResourceTable).
|
|
Where(squirrel.Eq{infraProviderResourceIDColumn: ids})
|
|
|
|
sql, args, err := stmt.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
|
|
}
|
|
dst := new([]infraProviderResource)
|
|
db := dbtx.GetAccessor(ctx, i.db)
|
|
if err := db.GetContext(ctx, dst, sql, args...); err != nil {
|
|
return nil, database.ProcessSQLErrorf(ctx, err, "Failed to find infraprovider resources")
|
|
}
|
|
return mapToInfraProviderResources(ctx, *dst)
|
|
}
|