mirror of
https://github.com/hanzozt/zt.git
synced 2026-10-30 15:04:55 +00:00
zeekay
241625e7d9
- Repo names: ziti-* → zt-* - Go modules: github.com/hanzozt/ziti → github.com/hanzozt/zt - Docker images: hanzozt/ziti-* → hanzozt/zt-* - C/Java/JS symbols: ziti_* → zt_* - File and directory names updated
102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/michaelquigley/pfxlog"
|
|
"github.com/hanzozt/sdk-golang/zt/edge"
|
|
"github.com/hanzozt/storage/boltz"
|
|
"github.com/hanzozt/zt/v2/common/pb/cmd_pb"
|
|
"github.com/hanzozt/zt/v2/common/pb/edge_cmd_pb"
|
|
"github.com/hanzozt/zt/v2/controller/change"
|
|
"github.com/hanzozt/zt/v2/controller/command"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// InvalidTerminatorInstanceIdError indicates a non-retriable terminator validation failure
|
|
type InvalidTerminatorInstanceIdError struct {
|
|
msg string
|
|
}
|
|
|
|
func (e InvalidTerminatorInstanceIdError) ErrorCode() uint32 {
|
|
return edge.ErrorCodeInvalidInstanceId
|
|
}
|
|
|
|
func (e InvalidTerminatorInstanceIdError) GetRetryHint() edge.RetryHint {
|
|
return edge.RetryNotRetriable
|
|
}
|
|
|
|
func (e InvalidTerminatorInstanceIdError) Error() string {
|
|
return e.msg
|
|
}
|
|
|
|
type CreateEdgeTerminatorCmd struct {
|
|
Env Env
|
|
Entity *Terminator
|
|
Context *change.Context
|
|
}
|
|
|
|
func (self *CreateEdgeTerminatorCmd) Apply(ctx boltz.MutateContext) error {
|
|
createCmd := &command.CreateEntityCommand[*Terminator]{
|
|
Creator: self.Env.GetManagers().Terminator,
|
|
Entity: self.Entity,
|
|
PostCreateHook: self.validateTerminatorIdentity,
|
|
Context: self.Context,
|
|
}
|
|
return self.Env.GetManagers().Terminator.ApplyCreate(createCmd, ctx)
|
|
}
|
|
|
|
func (self *CreateEdgeTerminatorCmd) validateTerminatorIdentity(ctx boltz.MutateContext, terminator *Terminator) error {
|
|
tx := ctx.Tx()
|
|
|
|
if terminator.GetInstanceId() == "" {
|
|
return nil
|
|
}
|
|
|
|
identityTerminators, err := self.Env.GetStores().Terminator.GetTerminatorsInIdentityGroup(tx, terminator.GetId())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, otherTerminator := range identityTerminators {
|
|
if otherTerminator.HostId != terminator.HostId {
|
|
pfxlog.Logger().WithFields(logrus.Fields{
|
|
"terminatorId": terminator.GetId(),
|
|
"siblingId": otherTerminator.GetId(),
|
|
"instanceId": terminator.InstanceId,
|
|
"terminatorIdentity": terminator.HostId,
|
|
"existingIdentity": otherTerminator.HostId,
|
|
}).Warn("validation of terminator failed, shared identity belongs to different identity")
|
|
return InvalidTerminatorInstanceIdError{
|
|
msg: fmt.Sprintf("sibling terminator %v with shared identity %v belongs to different identity", otherTerminator.GetId(), terminator.GetInstanceId()),
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (self *CreateEdgeTerminatorCmd) GetChangeContext() *change.Context {
|
|
return self.Context
|
|
}
|
|
|
|
func (self *CreateEdgeTerminatorCmd) Encode() ([]byte, error) {
|
|
terminatorData, err := self.Env.GetManagers().Terminator.Marshall(self.Entity)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cmd := &edge_cmd_pb.CreateEdgeTerminatorCommand{
|
|
TerminatorData: terminatorData,
|
|
Ctx: ContextToProtobuf(self.Context),
|
|
}
|
|
return cmd_pb.EncodeProtobuf(cmd)
|
|
}
|
|
|
|
func (self *CreateEdgeTerminatorCmd) Decode(env Env, msg *edge_cmd_pb.CreateEdgeTerminatorCommand) error {
|
|
var err error
|
|
self.Env = env
|
|
self.Entity, err = env.GetManagers().Terminator.Unmarshall(msg.TerminatorData)
|
|
self.Context = ProtobufToContext(msg.Ctx)
|
|
return err
|
|
}
|