Timothy Messier
eac433f7ff
This moves all of the tcp target functionality into a tcp package. It also establishes a Register function in the target package to allow specific target types to be registered at initilazation time. When registering, a target type needs to provide its Subtype, as well as some RepositoryHooks that are used by the target Repository.
113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
package target
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
"github.com/hashicorp/boundary/internal/errors"
|
|
hostStore "github.com/hashicorp/boundary/internal/host/store"
|
|
"github.com/hashicorp/boundary/internal/oplog"
|
|
"github.com/hashicorp/boundary/internal/target/store"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
var _ HostSource = (*TargetSet)(nil)
|
|
|
|
const (
|
|
DefaultTargetHostSetTableName = "target_host_set"
|
|
)
|
|
|
|
type TargetHostSet struct {
|
|
*store.TargetHostSet
|
|
tableName string `gorm:"-"`
|
|
}
|
|
|
|
var _ db.VetForWriter = (*TargetHostSet)(nil)
|
|
|
|
// NewTargetHostSet creates a new in memory target host set. No options are
|
|
// currently supported.
|
|
func NewTargetHostSet(targetId, hostSetId string, _ ...Option) (*TargetHostSet, error) {
|
|
const op = "target.NewTargetHostSet"
|
|
if targetId == "" {
|
|
return nil, errors.NewDeprecated(errors.InvalidParameter, op, "missing target id")
|
|
}
|
|
if hostSetId == "" {
|
|
return nil, errors.NewDeprecated(errors.InvalidParameter, op, "missing hostSetId id")
|
|
}
|
|
t := &TargetHostSet{
|
|
TargetHostSet: &store.TargetHostSet{
|
|
TargetId: targetId,
|
|
HostSetId: hostSetId,
|
|
},
|
|
}
|
|
return t, nil
|
|
}
|
|
|
|
// Clone creates a clone of the target host set
|
|
func (t *TargetHostSet) Clone() interface{} {
|
|
cp := proto.Clone(t.TargetHostSet)
|
|
return &TargetHostSet{
|
|
TargetHostSet: cp.(*store.TargetHostSet),
|
|
}
|
|
}
|
|
|
|
// VetForWrite implements db.VetForWrite() interface and validates the target
|
|
// host set before it's written.
|
|
func (t *TargetHostSet) VetForWrite(ctx context.Context, _ db.Reader, opType db.OpType, _ ...db.Option) error {
|
|
const op = "target.(TargetHostSet).VetForWrite"
|
|
if opType == db.CreateOp {
|
|
if t.TargetId == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing target id")
|
|
}
|
|
if t.HostSetId == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing host set id")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// TableName returns the tablename to override the default gorm table name
|
|
func (t *TargetHostSet) TableName() string {
|
|
if t.tableName != "" {
|
|
return t.tableName
|
|
}
|
|
return DefaultTargetHostSetTableName
|
|
}
|
|
|
|
// SetTableName sets the tablename and satisfies the ReplayableMessage
|
|
// interface. If the caller attempts to set the name to "" the name will be
|
|
// reset to the default name.
|
|
func (t *TargetHostSet) SetTableName(n string) {
|
|
t.tableName = n
|
|
}
|
|
|
|
func (t *TargetHostSet) oplog(op oplog.OpType) oplog.Metadata {
|
|
metadata := oplog.Metadata{
|
|
"resource-public-id": []string{fmt.Sprintf("%s:%s", t.TargetId, t.HostSetId)},
|
|
"resource-type": []string{"target host set"},
|
|
"op-type": []string{op.String()},
|
|
}
|
|
return metadata
|
|
}
|
|
|
|
// TargetSet is returned from most repo operations as the target's host set.
|
|
type TargetSet struct {
|
|
*hostStore.Set
|
|
}
|
|
|
|
// TableName returns the tablename to override the default gorm table name
|
|
func (ts *TargetSet) TableName() string {
|
|
return "target_set"
|
|
}
|
|
|
|
// Id returns the ID of the host set
|
|
func (ts *TargetSet) Id() string {
|
|
return ts.PublicId
|
|
}
|
|
|
|
// HostCatalogId returns the ID of the catalog containing the set
|
|
func (ts *TargetSet) HostCatalogId() string {
|
|
return ts.CatalogId
|
|
}
|