Jim
6a37c93d77
* fix (auth): properly handle rows.Err() * fix (auth/ldap): properly handle rows.Err() * fix (auth/oidc): properly handle rows.Err() * fix (auth/password): properly handle rows.Err() * fix (authtoken): properly handle rows.Err() * fix (credential): properly handle rows.Err() * fix (credential/static): properly handle rows.Err() * fix (credential/vault): properly handle rows.Err() * fix (db): properly handle rows.Err() * fix (db/schema): properly handle rows.Err() * fix (db/migrations): properly handle rows.Err() * fix (host): properly handle rows.Err() * fix (host/plugin): properly handle rows.Err() * fix (host/static): properly handle rows.Err() * fix (iam): properly handle rows.Err() * fix (kms): properly handle rows.Err() * fix (oplog): properly handle rows.Err() * fix (pagination/purge): properly handle rows.Err() * fix (plugin): properly handle rows.Err() * fix (job/scheduler): properly handle rows.Err() * fix (server): properly handle rows.Err() * fix (session): properly handle rows.Err() * fix (target): properly handle rows.Err()
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package oplog
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/go-dbw"
|
|
|
|
"github.com/hashicorp/boundary/internal/errors"
|
|
)
|
|
|
|
// Writer provides a database writer for oplog operations
|
|
type Writer struct {
|
|
*dbw.DB
|
|
}
|
|
|
|
// HasTable checks if tableName exists
|
|
func (w *Writer) hasTable(ctx context.Context, tableName string) (bool, error) {
|
|
const op = "oplog.(Writer).hasTable"
|
|
if tableName == "" {
|
|
return false, errors.New(ctx, errors.InvalidParameter, op, "missing table name")
|
|
}
|
|
var count int64
|
|
rw := dbw.New(w.DB)
|
|
rows, err := rw.Query(context.Background(), "select count(*) from information_schema.tables where table_name = ? and table_type = ?", []any{tableName, "BASE TABLE"})
|
|
if err != nil {
|
|
return false, errors.Wrap(ctx, err, op)
|
|
}
|
|
if ok := rows.Next(); ok {
|
|
rw.ScanRows(rows, &count)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return false, errors.Wrap(ctx, err, op)
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
// CreateTableLike will create a newTableName like the model's table
|
|
// the new table should have all things like the existing model's table (defaults, constraints, indexes, etc)
|
|
func (w *Writer) createTableLike(ctx context.Context, existingTableName string, newTableName string) error {
|
|
const op = "oplog.(Writer).createTableLike"
|
|
if existingTableName == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing existing table name")
|
|
}
|
|
if newTableName == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing new table name")
|
|
}
|
|
|
|
sql := fmt.Sprintf(
|
|
`CREATE TABLE "%s" ( LIKE %s INCLUDING ALL );`,
|
|
newTableName,
|
|
existingTableName,
|
|
)
|
|
_, err := dbw.New(w.DB).Exec(ctx, sql, nil)
|
|
if err != nil {
|
|
return errors.Wrap(ctx, err, op)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DropTableIfExists will drop the table if it exists
|
|
func (w *Writer) dropTableIfExists(ctx context.Context, tableName string) error {
|
|
const op = "oplog.(Writer).dropTableIfExists"
|
|
if tableName == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing table name")
|
|
}
|
|
sql := fmt.Sprintf("drop table if exists %s ", tableName)
|
|
_, err := dbw.New(w.DB).Exec(ctx, sql, nil)
|
|
if err != nil {
|
|
return errors.Wrap(ctx, err, op)
|
|
}
|
|
return nil
|
|
}
|