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.
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
package target
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/boundary/internal/db/timestamp"
|
|
"github.com/hashicorp/boundary/internal/errors"
|
|
"github.com/hashicorp/boundary/internal/oplog"
|
|
"github.com/hashicorp/boundary/internal/target/store"
|
|
"github.com/hashicorp/boundary/internal/types/subtypes"
|
|
)
|
|
|
|
// Target is a commmon interface for all target subtypes
|
|
type Target interface {
|
|
GetPublicId() string
|
|
GetScopeId() string
|
|
GetDefaultPort() uint32
|
|
GetName() string
|
|
GetDescription() string
|
|
GetVersion() uint32
|
|
GetType() subtypes.Subtype
|
|
GetCreateTime() *timestamp.Timestamp
|
|
GetUpdateTime() *timestamp.Timestamp
|
|
GetSessionMaxSeconds() uint32
|
|
GetSessionConnectionLimit() int32
|
|
GetWorkerFilter() string
|
|
Clone() Target
|
|
SetPublicId(context.Context, string) error
|
|
SetScopeId(string)
|
|
SetName(string)
|
|
SetDescription(string)
|
|
SetVersion(uint32)
|
|
SetDefaultPort(uint32)
|
|
SetCreateTime(*timestamp.Timestamp)
|
|
SetUpdateTime(*timestamp.Timestamp)
|
|
SetSessionMaxSeconds(uint32)
|
|
SetSessionConnectionLimit(int32)
|
|
SetWorkerFilter(string)
|
|
Oplog(op oplog.OpType) oplog.Metadata
|
|
}
|
|
|
|
// TargetType defines the possible types for targets.
|
|
type TargetType uint32
|
|
|
|
const (
|
|
UnknownTargetType TargetType = 0
|
|
TcpTargetType TargetType = 1
|
|
)
|
|
|
|
// String returns a string representation of the target type.
|
|
func (t TargetType) String() string {
|
|
return [...]string{
|
|
"unknown",
|
|
"tcp",
|
|
}[t]
|
|
}
|
|
|
|
const (
|
|
targetsViewDefaultTable = "target_all_subtypes"
|
|
)
|
|
|
|
// targetView provides a common way to return targets regardless of their
|
|
// underlying type.
|
|
type targetView struct {
|
|
*store.TargetView
|
|
tableName string `gorm:"-"`
|
|
}
|
|
|
|
// allocTargetView will allocate a target view
|
|
func allocTargetView() targetView {
|
|
return targetView{
|
|
TargetView: &store.TargetView{},
|
|
}
|
|
}
|
|
|
|
// TableName provides an overridden gorm table name for targets.
|
|
func (t *targetView) TableName() string {
|
|
if t.tableName != "" {
|
|
return t.tableName
|
|
}
|
|
return targetsViewDefaultTable
|
|
}
|
|
|
|
// SetTableName sets the table name for the resource. If the caller attempts to
|
|
// set the name to "" the name will be reset to the default name.
|
|
func (t *targetView) SetTableName(n string) {
|
|
switch n {
|
|
case "":
|
|
t.tableName = targetsViewDefaultTable
|
|
default:
|
|
t.tableName = n
|
|
}
|
|
}
|
|
|
|
func (t *targetView) Subtype() subtypes.Subtype {
|
|
return subtypes.Subtype(t.Type)
|
|
}
|
|
|
|
// targetSubtype converts the target view to the concrete subtype
|
|
func (t *targetView) targetSubtype(ctx context.Context) (Target, error) {
|
|
const op = "target.targetView.targetSubtype"
|
|
|
|
alloc, ok := subtypeRegistry.allocFunc(t.Subtype())
|
|
if !ok {
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, fmt.Sprintf("%s is an unknown target subtype of %s", t.PublicId, t.Type))
|
|
}
|
|
|
|
tt := alloc()
|
|
if err := tt.SetPublicId(ctx, t.PublicId); err != nil {
|
|
return nil, err
|
|
}
|
|
tt.SetVersion(t.Version)
|
|
tt.SetScopeId(t.ScopeId)
|
|
tt.SetName(t.Name)
|
|
tt.SetDescription(t.Description)
|
|
tt.SetDefaultPort(t.DefaultPort)
|
|
tt.SetCreateTime(t.CreateTime)
|
|
tt.SetUpdateTime(t.UpdateTime)
|
|
tt.SetSessionMaxSeconds(t.SessionMaxSeconds)
|
|
tt.SetSessionConnectionLimit(t.SessionConnectionLimit)
|
|
tt.SetWorkerFilter(t.WorkerFilter)
|
|
return tt, nil
|
|
}
|