Todd
29b4537d3b
* Reference PreferredEndpoints by top level struct name instead of embedded proto GetPreferredEndpoint method. Reorganize the sql files a little. * Make PreferredEndpoints replayable. Test refactoring to remove variables that persist across test calls. DeleteItems instead of trying to use WithWhere.
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package host
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hashicorp/boundary/internal/errors"
|
|
"github.com/hashicorp/boundary/internal/host/store"
|
|
"github.com/hashicorp/boundary/internal/libs/endpoint"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// defaultPreferredEndpointTableName defines the default table name for a
|
|
// PreferredEndpoint
|
|
const defaultPreferredEndpointTableName = "host_set_preferred_endpoint"
|
|
|
|
// PreferredEndpoint defines an endpoint condition for a host set.
|
|
// PreferredEndpoints are value objects of an HostSet, therefore there's no need
|
|
// for oplog metadata, since only the HostSet will have metadata because it's
|
|
// the root aggregate.
|
|
type PreferredEndpoint struct {
|
|
*store.PreferredEndpoint
|
|
tableName string
|
|
}
|
|
|
|
// NewPreferredEndpoint creates a new in memory preferred endpoint assigned to a
|
|
// HostSet. It supports no options.
|
|
func NewPreferredEndpoint(ctx context.Context, hostSetId string, priority uint32, condition string) (*PreferredEndpoint, error) {
|
|
const op = "host.NewPreferredEndpoint"
|
|
|
|
pe := &PreferredEndpoint{
|
|
PreferredEndpoint: &store.PreferredEndpoint{
|
|
HostSetId: hostSetId,
|
|
Priority: priority,
|
|
Condition: condition,
|
|
},
|
|
}
|
|
if err := pe.validate(ctx, op); err != nil {
|
|
return nil, err // intentionally not wrapped
|
|
}
|
|
return pe, nil
|
|
}
|
|
|
|
// validate the PreferredEndpoint. On success, it will return nil.
|
|
func (pe *PreferredEndpoint) validate(ctx context.Context, caller errors.Op) error {
|
|
if pe.HostSetId == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, caller, "missing host set id")
|
|
}
|
|
if pe.Priority < 1 {
|
|
return errors.New(ctx, errors.InvalidParameter, caller, "invalid priority value")
|
|
}
|
|
if pe.Condition == "" {
|
|
return errors.New(ctx, errors.InvalidParameter, caller, "missing condition")
|
|
}
|
|
if _, err := endpoint.NewPreferencer(ctx, endpoint.WithPreferenceOrder([]string{pe.Condition})); err != nil {
|
|
return errors.Wrap(ctx, err, caller)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// AllocPreferredEndpoint makes an empty one in-memory
|
|
func AllocPreferredEndpoint() *PreferredEndpoint {
|
|
return &PreferredEndpoint{
|
|
PreferredEndpoint: &store.PreferredEndpoint{},
|
|
}
|
|
}
|
|
|
|
// Clone a PreferredEndpoint
|
|
func (c *PreferredEndpoint) Clone() *PreferredEndpoint {
|
|
cp := proto.Clone(c.PreferredEndpoint)
|
|
return &PreferredEndpoint{
|
|
PreferredEndpoint: cp.(*store.PreferredEndpoint),
|
|
}
|
|
}
|
|
|
|
// TableName returns the table name.
|
|
func (c *PreferredEndpoint) TableName() string {
|
|
if c.tableName != "" {
|
|
return c.tableName
|
|
}
|
|
return defaultPreferredEndpointTableName
|
|
}
|
|
|
|
// SetTableName sets the table name.
|
|
func (c *PreferredEndpoint) SetTableName(n string) {
|
|
c.tableName = n
|
|
}
|