70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/cloudbase/garm/database/common"
|
|
"github.com/cloudbase/garm/params"
|
|
)
|
|
|
|
func (s *sqlDatabase) CreateScaleSetInstance(_ context.Context, scaleSetID uint, param params.CreateInstanceParams) (instance params.Instance, err error) {
|
|
scaleSet, err := s.getScaleSetByID(s.conn, scaleSetID)
|
|
if err != nil {
|
|
return params.Instance{}, errors.Wrap(err, "fetching scale set")
|
|
}
|
|
|
|
defer func() {
|
|
if err == nil {
|
|
s.sendNotify(common.InstanceEntityType, common.CreateOperation, instance)
|
|
}
|
|
}()
|
|
|
|
var secret []byte
|
|
if len(param.JitConfiguration) > 0 {
|
|
secret, err = s.marshalAndSeal(param.JitConfiguration)
|
|
if err != nil {
|
|
return params.Instance{}, errors.Wrap(err, "marshalling jit config")
|
|
}
|
|
}
|
|
|
|
newInstance := Instance{
|
|
ScaleSet: scaleSet,
|
|
Name: param.Name,
|
|
Status: param.Status,
|
|
RunnerStatus: param.RunnerStatus,
|
|
OSType: param.OSType,
|
|
OSArch: param.OSArch,
|
|
CallbackURL: param.CallbackURL,
|
|
MetadataURL: param.MetadataURL,
|
|
GitHubRunnerGroup: param.GitHubRunnerGroup,
|
|
JitConfiguration: secret,
|
|
AgentID: param.AgentID,
|
|
}
|
|
q := s.conn.Create(&newInstance)
|
|
if q.Error != nil {
|
|
return params.Instance{}, errors.Wrap(q.Error, "creating instance")
|
|
}
|
|
|
|
return s.sqlToParamsInstance(newInstance)
|
|
}
|
|
|
|
func (s *sqlDatabase) ListScaleSetInstances(_ context.Context, scalesetID uint) ([]params.Instance, error) {
|
|
var instances []Instance
|
|
query := s.conn.Model(&Instance{}).Preload("Job").Where("scale_set_fk_id = ?", scalesetID)
|
|
|
|
if err := query.Find(&instances); err.Error != nil {
|
|
return nil, errors.Wrap(err.Error, "fetching instances")
|
|
}
|
|
|
|
var err error
|
|
ret := make([]params.Instance, len(instances))
|
|
for idx, inst := range instances {
|
|
ret[idx], err = s.sqlToParamsInstance(inst)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "converting instance")
|
|
}
|
|
}
|
|
return ret, nil
|
|
}
|