53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
 
|
package plugin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
|
||
|
|
"github.com/hashicorp/boundary/internal/errors"
|
||
|
|
"github.com/hashicorp/boundary/internal/host/plugin/store"
|
||
|
|
)
|
||
|
|
|
||
|
|
// A HostSet is a collection of hosts from the set's catalog.
|
||
|
|
type HostSet struct {
|
||
|
|
*store.HostSet
|
||
|
|
tableName string `gorm:"-"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewHostSet creates a new in memory HostSet assigned to catalogId.
|
||
|
|
// Attributes, name, and description are the only valid options. All other
|
||
|
|
// options are ignored.
|
||
|
|
func NewHostSet(ctx context.Context, catalogId string, opt ...Option) (*HostSet, error) {
|
||
|
|
const op = "plugin.NewHostSet"
|
||
|
|
opts := getOpts(opt...)
|
||
|
|
set := &HostSet{
|
||
|
|
HostSet: &store.HostSet{
|
||
|
|
CatalogId: catalogId,
|
||
|
|
Name: opts.withName,
|
||
|
|
Description: opts.withDescription,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
if opts.withAttributes != nil {
|
||
|
|
attrs, err := json.Marshal(opts.withAttributes)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(ctx, err, op, errors.WithCode(errors.InvalidParameter))
|
||
|
|
}
|
||
|
|
set.Attributes = attrs
|
||
|
|
}
|
||
|
|
return set, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// TableName returns the table name for the host set.
|
||
|
|
func (s *HostSet) TableName() string {
|
||
|
|
if s.tableName != "" {
|
||
|
|
return s.tableName
|
||
|
|
}
|
||
|
|
return "host_plugin_set"
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetTableName sets the table name. If the caller attempts to
|
||
|
|
// set the name to "" the name will be reset to the default name.
|
||
|
|
func (s *HostSet) SetTableName(n string) {
|
||
|
|
s.tableName = n
|
||
|
|
}
|