client

package
v1.4.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 28, 2026 License: Apache-2.0 Imports: 26 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	ErrCarbideClientInvalidAddress    = errors.New("CarbideClient: invalid address")
	ErrCarbideClientInvalidDialOpts   = errors.New("CarbideClient: invalid dial options")
	ErrCarbideClientInvalidSecureOpts = errors.New("CarbideClient: invalid secure options")
	ErrCarbideClientInvalidServerCA   = errors.New("CarbideClient: invalid server CA")
	ErrCarbideClientInvalidClientCA   = errors.New("CarbideClient: invalid client CA")
	ErrCarbideClientInvalidClientKey  = errors.New("CarbideClient: invalid client key")
	ErrCarbideClientInvalidClientCert = errors.New("CarbideClient: invalid client cert")
	ErrClientNotConnected             = errors.New("gRPC client is not connected to the server")
)

Errors

View Source 
var (
	ErrRlaClientInvalidAddress    = errors.New("RlaClient: invalid address")
	ErrRlaClientInvalidDialOpts   = errors.New("RlaClient: invalid dial options")
	ErrRlaClientInvalidSecureOpts = errors.New("RlaClient: invalid secure options")
	ErrRlaClientInvalidServerCA   = errors.New("RlaClient: invalid server CA")
	ErrRlaClientInvalidClientCA   = errors.New("RlaClient: invalid client CA")
	ErrRlaClientInvalidClientKey  = errors.New("RlaClient: invalid client key")
	ErrRlaClientInvalidClientCert = errors.New("RlaClient: invalid client cert")
)

Errors

Functions

func SliceToChunks 

func SliceToChunks[T any](slice []T, chunkSize int) (chunks [][]T)

Types

type CarbideAtomicClient 

type CarbideAtomicClient struct {
	Config *CarbideClientConfig
	// contains filtered or unexported fields
}

CarbideAtomicClient is an atomic wrapper around the CarbideClient

func NewCarbideAtomicClient 

func NewCarbideAtomicClient(config *CarbideClientConfig) *CarbideAtomicClient

NewCarbideAtomicClient creates a new CarbideAtomicClient

func (*CarbideAtomicClient) CheckAndReloadCerts 

func (cac *CarbideAtomicClient) CheckAndReloadCerts(initialClientCertMD5, initialServerCAMD5 []byte)

CheckAndReloadCerts continuously monitors the TLS certificates for changes. If a change is detected, it reinitializes the CarbideClient with the new certificates to ensure secure communication.

func (*CarbideAtomicClient) CheckCertificates 

func (cac *CarbideAtomicClient) CheckCertificates(lastClientCertMD5, lastServerCAMD5 []byte) (bool, []byte, []byte, error)

CheckCertificates checks if the client and server CA certificates have changed

func (*CarbideAtomicClient) GetClient 

func (cac *CarbideAtomicClient) GetClient() *CarbideClient

GetClient returns the current version of Carbide client from the atomic value. Returns nil if the client has not been initialized yet.

func (*CarbideAtomicClient) GetInitialCertMD5 

func (cac *CarbideAtomicClient) GetInitialCertMD5() (clientCertMD5, serverCAMD5 []byte, err error)

GetInitialCertMD5 retrieves the MD5 hash of the initial set of certificate that the client is Using

func (*CarbideAtomicClient) SwapClient 

func (cac *CarbideAtomicClient) SwapClient(newClient *CarbideClient) *CarbideClient

SwapClient atomically replaces the current CarbideClient with a new one, returning the old client for the caller to manage.

func (*CarbideAtomicClient) Version 

func (cac *CarbideAtomicClient) Version() int64

Version returns the current version of the CarbideClient

type CarbideClient 

type CarbideClient struct {
	// contains filtered or unexported fields
}

CarbideClient is the data structure for the client

func NewCarbideClient 

func NewCarbideClient(config *CarbideClientConfig) (client *CarbideClient, err error)

NewCarbideClient creates a new CarbideClient

func NewMockCarbideClient 

func NewMockCarbideClient() *CarbideClient

NewMockCarbideClient creates a new mock CarbideClient

func (*CarbideClient) Carbide 

func (client *CarbideClient) Carbide() wflows.ForgeClient

Carbide client getter

func (*CarbideClient) Close 

func (cc *CarbideClient) Close() error

Close gracefully shuts down the client's gRPC connection.

type CarbideClientConfig 

type CarbideClientConfig struct {
	// The address of the server <host>:<port>
	Address string
	// Secure flag
	Secure SecureOptions
	// Skip Server Auth
	SkipServerAuth bool
	// The TLS certificate for the server
	ServerCAPath string
	// The TLS certificate for the client
	ClientCertPath string
	// The TLS key for the client
	ClientKeyPath string
	// client metrics interface
	ClientMetrics Metrics
}

CarbideClientConfig is the data structure for the client configuration

type Metrics 

type Metrics interface {
	// RecordRpcResponse call-back method that includes rpc method, response code, and duration
	RecordRpcResponse(method, code string, duration time.Duration)
}

Metrics interface that defines call-back functions for RPC metrics

type MockForgeClient 

type MockForgeClient struct {
	wflows.ForgeClient
}

MockForgeClient is a mock implementation of Forge gRPC protobuf Client

func (*MockForgeClient) AddExpectedMachine 

func (c *MockForgeClient) AddExpectedMachine(ctx context.Context, in *wflows.ExpectedMachine, opts ...grpc.CallOption) (*emptypb.Empty, error)

Expected Machine mock methods

func (*MockForgeClient) AddExpectedPowerShelf 

func (c *MockForgeClient) AddExpectedPowerShelf(ctx context.Context, in *wflows.ExpectedPowerShelf, opts ...grpc.CallOption) (*emptypb.Empty, error)

Expected Power Shelf mock methods

func (*MockForgeClient) AddExpectedSwitch 

func (c *MockForgeClient) AddExpectedSwitch(ctx context.Context, in *wflows.ExpectedSwitch, opts ...grpc.CallOption) (*emptypb.Empty, error)

Expected Switch mock methods

func (*MockForgeClient) AddMachineValidationTest 

func (c *MockForgeClient) AddMachineValidationTest(ctx context.Context, in *wflows.MachineValidationTestAddRequest, opts ...grpc.CallOption) (*wflows.MachineValidationTestAddUpdateResponse, error)

Machine Validation Test mock methods

func (*MockForgeClient) AddUpdateMachineValidationExternalConfig 

func (c *MockForgeClient) AddUpdateMachineValidationExternalConfig(ctx context.Context, in *wflows.AddUpdateMachineValidationExternalConfigRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) AllocateInstance 

func (c *MockForgeClient) AllocateInstance(ctx context.Context, in *wflows.InstanceAllocationRequest, opts ...grpc.CallOption) (*wflows.Instance, error)

Instance mock methods

func (*MockForgeClient) AllocateInstances 

func (c *MockForgeClient) AllocateInstances(ctx context.Context, in *wflows.BatchInstanceAllocationRequest, opts ...grpc.CallOption) (*wflows.BatchInstanceAllocationResponse, error)

func (*MockForgeClient) AssociateMachinesWithInstanceType 

func (c *MockForgeClient) AssociateMachinesWithInstanceType(ctx context.Context, in *wflows.AssociateMachinesWithInstanceTypeRequest, opts ...grpc.CallOption) (*wflows.AssociateMachinesWithInstanceTypeResponse, error)

func (*MockForgeClient) CreateDpuExtensionService 

func (c *MockForgeClient) CreateDpuExtensionService(ctx context.Context, in *wflows.CreateDpuExtensionServiceRequest, opts ...grpc.CallOption) (*wflows.DpuExtensionService, error)

DPU Extension Service mock methods

func (*MockForgeClient) CreateExpectedMachines 

func (c *MockForgeClient) CreateExpectedMachines(ctx context.Context, in *wflows.BatchExpectedMachineOperationRequest, opts ...grpc.CallOption) (*wflows.BatchExpectedMachineOperationResponse, error)

func (*MockForgeClient) CreateIBPartition 

func (c *MockForgeClient) CreateIBPartition(ctx context.Context, in *wflows.IBPartitionCreationRequest, opts ...grpc.CallOption) (*wflows.IBPartition, error)

InfiniBand Partition mock methods

func (*MockForgeClient) CreateInstanceType 

func (c *MockForgeClient) CreateInstanceType(ctx context.Context, in *wflows.CreateInstanceTypeRequest, opts ...grpc.CallOption) (*wflows.CreateInstanceTypeResponse, error)

Instance Type mock methods

func (*MockForgeClient) CreateNVLinkLogicalPartition 

func (c *MockForgeClient) CreateNVLinkLogicalPartition(ctx context.Context, in *wflows.NVLinkLogicalPartitionCreationRequest, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartition, error)

NVLink Logical Partition Mocks

func (*MockForgeClient) CreateNetworkSecurityGroup 

func (c *MockForgeClient) CreateNetworkSecurityGroup(ctx context.Context, in *wflows.CreateNetworkSecurityGroupRequest, opts ...grpc.CallOption) (*wflows.CreateNetworkSecurityGroupResponse, error)

func (*MockForgeClient) CreateNetworkSegment 

func (c *MockForgeClient) CreateNetworkSegment(ctx context.Context, in *wflows.NetworkSegmentCreationRequest, opts ...grpc.CallOption) (*wflows.NetworkSegment, error)

func (*MockForgeClient) CreateOsImage 

func (c *MockForgeClient) CreateOsImage(ctx context.Context, in *wflows.OsImageAttributes, opts ...grpc.CallOption) (*wflows.OsImage, error)

OS Image mock methods

func (*MockForgeClient) CreateTenant 

func (c *MockForgeClient) CreateTenant(ctx context.Context, in *wflows.CreateTenantRequest, opts ...grpc.CallOption) (*wflows.CreateTenantResponse, error)

Tenant mock methods

func (*MockForgeClient) CreateTenantKeyset 

func (c *MockForgeClient) CreateTenantKeyset(ctx context.Context, in *wflows.CreateTenantKeysetRequest, opts ...grpc.CallOption) (*wflows.CreateTenantKeysetResponse, error)

Tenant Keyset mock methods

func (*MockForgeClient) CreateVpc 

func (c *MockForgeClient) CreateVpc(ctx context.Context, in *wflows.VpcCreationRequest, opts ...grpc.CallOption) (*wflows.Vpc, error)

VPC mock methods

func (*MockForgeClient) CreateVpcPeering 

func (c *MockForgeClient) CreateVpcPeering(ctx context.Context, in *wflows.VpcPeeringCreationRequest, opts ...grpc.CallOption) (*wflows.VpcPeering, error)

VPC Peering mock methods

func (*MockForgeClient) CreateVpcPrefix 

func (c *MockForgeClient) CreateVpcPrefix(ctx context.Context, in *wflows.VpcPrefixCreationRequest, opts ...grpc.CallOption) (*wflows.VpcPrefix, error)

VPC Prefix mock methods

func (*MockForgeClient) DeleteDpuExtensionService 

func (c *MockForgeClient) DeleteDpuExtensionService(ctx context.Context, in *wflows.DeleteDpuExtensionServiceRequest, opts ...grpc.CallOption) (*wflows.DeleteDpuExtensionServiceResponse, error)

func (*MockForgeClient) DeleteExpectedMachine 

func (c *MockForgeClient) DeleteExpectedMachine(ctx context.Context, in *wflows.ExpectedMachineRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) DeleteExpectedPowerShelf 

func (c *MockForgeClient) DeleteExpectedPowerShelf(ctx context.Context, in *wflows.ExpectedPowerShelfRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) DeleteExpectedSwitch 

func (c *MockForgeClient) DeleteExpectedSwitch(ctx context.Context, in *wflows.ExpectedSwitchRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) DeleteIBPartition 

func (c *MockForgeClient) DeleteIBPartition(ctx context.Context, in *wflows.IBPartitionDeletionRequest, opts ...grpc.CallOption) (*wflows.IBPartitionDeletionResult, error)

func (*MockForgeClient) DeleteInstanceType 

func (c *MockForgeClient) DeleteInstanceType(ctx context.Context, in *wflows.DeleteInstanceTypeRequest, opts ...grpc.CallOption) (*wflows.DeleteInstanceTypeResponse, error)

func (*MockForgeClient) DeleteNVLinkLogicalPartition 

func (c *MockForgeClient) DeleteNVLinkLogicalPartition(ctx context.Context, in *wflows.NVLinkLogicalPartitionDeletionRequest, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartitionDeletionResult, error)

func (*MockForgeClient) DeleteNetworkSecurityGroup 

func (c *MockForgeClient) DeleteNetworkSecurityGroup(ctx context.Context, in *wflows.DeleteNetworkSecurityGroupRequest, opts ...grpc.CallOption) (*wflows.DeleteNetworkSecurityGroupResponse, error)

func (*MockForgeClient) DeleteNetworkSegment 

func (c *MockForgeClient) DeleteNetworkSegment(ctx context.Context, in *wflows.NetworkSegmentDeletionRequest, opts ...grpc.CallOption) (*wflows.NetworkSegmentDeletionResult, error)

func (*MockForgeClient) DeleteOsImage 

func (c *MockForgeClient) DeleteOsImage(ctx context.Context, in *wflows.DeleteOsImageRequest, opts ...grpc.CallOption) (*wflows.DeleteOsImageResponse, error)

func (*MockForgeClient) DeleteTenantKeyset 

func (c *MockForgeClient) DeleteTenantKeyset(ctx context.Context, in *wflows.DeleteTenantKeysetRequest, opts ...grpc.CallOption) (*wflows.DeleteTenantKeysetResponse, error)

func (*MockForgeClient) DeleteVpc 

func (c *MockForgeClient) DeleteVpc(ctx context.Context, in *wflows.VpcDeletionRequest, opts ...grpc.CallOption) (*wflows.VpcDeletionResult, error)

func (*MockForgeClient) DeleteVpcPeering 

func (c *MockForgeClient) DeleteVpcPeering(ctx context.Context, in *wflows.VpcPeeringDeletionRequest, opts ...grpc.CallOption) (*wflows.VpcPeeringDeletionResult, error)

func (*MockForgeClient) DeleteVpcPrefix 

func (c *MockForgeClient) DeleteVpcPrefix(ctx context.Context, in *wflows.VpcPrefixDeletionRequest, opts ...grpc.CallOption) (*wflows.VpcPrefixDeletionResult, error)

func (*MockForgeClient) FindDpuExtensionServiceIds 

func (c *MockForgeClient) FindDpuExtensionServiceIds(ctx context.Context, in *wflows.DpuExtensionServiceSearchFilter, opts ...grpc.CallOption) (*wflows.DpuExtensionServiceIdList, error)

func (*MockForgeClient) FindDpuExtensionServicesByIds 

func (c *MockForgeClient) FindDpuExtensionServicesByIds(ctx context.Context, in *wflows.DpuExtensionServicesByIdsRequest, opts ...grpc.CallOption) (*wflows.DpuExtensionServiceList, error)

func (*MockForgeClient) FindIBPartitionIds 

func (c *MockForgeClient) FindIBPartitionIds(ctx context.Context, in *wflows.IBPartitionSearchFilter, opts ...grpc.CallOption) (*wflows.IBPartitionIdList, error)

func (*MockForgeClient) FindIBPartitionsByIds 

func (c *MockForgeClient) FindIBPartitionsByIds(ctx context.Context, in *wflows.IBPartitionsByIdsRequest, opts ...grpc.CallOption) (*wflows.IBPartitionList, error)

func (*MockForgeClient) FindInstanceIds 

func (c *MockForgeClient) FindInstanceIds(ctx context.Context, in *wflows.InstanceSearchFilter, opts ...grpc.CallOption) (*wflows.InstanceIdList, error)

func (*MockForgeClient) FindInstanceTypeIds 

func (c *MockForgeClient) FindInstanceTypeIds(ctx context.Context, in *wflows.FindInstanceTypeIdsRequest, opts ...grpc.CallOption) (*wflows.FindInstanceTypeIdsResponse, error)

func (*MockForgeClient) FindInstanceTypesByIds 

func (c *MockForgeClient) FindInstanceTypesByIds(ctx context.Context, in *wflows.FindInstanceTypesByIdsRequest, opts ...grpc.CallOption) (*wflows.FindInstanceTypesByIdsResponse, error)

func (*MockForgeClient) FindInstancesByIds 

func (c *MockForgeClient) FindInstancesByIds(ctx context.Context, in *wflows.InstancesByIdsRequest, opts ...grpc.CallOption) (*wflows.InstanceList, error)

func (*MockForgeClient) FindMachineIds 

func (c *MockForgeClient) FindMachineIds(ctx context.Context, in *wflows.MachineSearchConfig, opts ...grpc.CallOption) (*wflows.MachineIdList, error)

func (*MockForgeClient) FindMachinesByIds 

func (c *MockForgeClient) FindMachinesByIds(ctx context.Context, in *wflows.MachinesByIdsRequest, opts ...grpc.CallOption) (*wflows.MachineList, error)

func (*MockForgeClient) FindNVLinkLogicalPartitionIds 

func (c *MockForgeClient) FindNVLinkLogicalPartitionIds(ctx context.Context, in *wflows.NVLinkLogicalPartitionSearchFilter, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartitionIdList, error)

func (*MockForgeClient) FindNVLinkLogicalPartitionsByIds 

func (c *MockForgeClient) FindNVLinkLogicalPartitionsByIds(ctx context.Context, in *wflows.NVLinkLogicalPartitionsByIdsRequest, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartitionList, error)

func (*MockForgeClient) FindNetworkSecurityGroupIds 

func (c *MockForgeClient) FindNetworkSecurityGroupIds(ctx context.Context, in *wflows.FindNetworkSecurityGroupIdsRequest, opts ...grpc.CallOption) (*wflows.FindNetworkSecurityGroupIdsResponse, error)

func (*MockForgeClient) FindNetworkSecurityGroupsByIds 

func (c *MockForgeClient) FindNetworkSecurityGroupsByIds(ctx context.Context, in *wflows.FindNetworkSecurityGroupsByIdsRequest, opts ...grpc.CallOption) (*wflows.FindNetworkSecurityGroupsByIdsResponse, error)

func (*MockForgeClient) FindNetworkSegmentIds 

func (c *MockForgeClient) FindNetworkSegmentIds(ctx context.Context, in *wflows.NetworkSegmentSearchFilter, opts ...grpc.CallOption) (*wflows.NetworkSegmentIdList, error)

func (*MockForgeClient) FindNetworkSegmentsByIds 

func (c *MockForgeClient) FindNetworkSegmentsByIds(ctx context.Context, in *wflows.NetworkSegmentsByIdsRequest, opts ...grpc.CallOption) (*wflows.NetworkSegmentList, error)

func (*MockForgeClient) FindSkusByIds 

func (c *MockForgeClient) FindSkusByIds(ctx context.Context, in *wflows.SkusByIdsRequest, opts ...grpc.CallOption) (*wflows.SkuList, error)

SKU mock methods

func (*MockForgeClient) FindTenant 

func (c *MockForgeClient) FindTenant(ctx context.Context, in *wflows.FindTenantRequest, opts ...grpc.CallOption) (*wflows.FindTenantResponse, error)

func (*MockForgeClient) FindTenantKeysetIds 

func (c *MockForgeClient) FindTenantKeysetIds(ctx context.Context, in *wflows.TenantKeysetSearchFilter, opts ...grpc.CallOption) (*wflows.TenantKeysetIdList, error)

func (*MockForgeClient) FindTenantKeysetsByIds 

func (c *MockForgeClient) FindTenantKeysetsByIds(ctx context.Context, in *wflows.TenantKeysetsByIdsRequest, opts ...grpc.CallOption) (*wflows.TenantKeySetList, error)

func (*MockForgeClient) FindTenantOrganizationIds 

func (c *MockForgeClient) FindTenantOrganizationIds(ctx context.Context, in *wflows.TenantSearchFilter, opts ...grpc.CallOption) (*wflows.TenantOrganizationIdList, error)

func (*MockForgeClient) FindTenantsByOrganizationIds 

func (c *MockForgeClient) FindTenantsByOrganizationIds(ctx context.Context, in *wflows.TenantByOrganizationIdsRequest, opts ...grpc.CallOption) (*wflows.TenantList, error)

func (*MockForgeClient) FindVpcIds 

func (c *MockForgeClient) FindVpcIds(ctx context.Context, in *wflows.VpcSearchFilter, opts ...grpc.CallOption) (*wflows.VpcIdList, error)

func (*MockForgeClient) FindVpcPeeringIds 

func (c *MockForgeClient) FindVpcPeeringIds(ctx context.Context, in *wflows.VpcPeeringSearchFilter, opts ...grpc.CallOption) (*wflows.VpcPeeringIdList, error)

func (*MockForgeClient) FindVpcPeeringsByIds 

func (c *MockForgeClient) FindVpcPeeringsByIds(ctx context.Context, in *wflows.VpcPeeringsByIdsRequest, opts ...grpc.CallOption) (*wflows.VpcPeeringList, error)

func (*MockForgeClient) FindVpcsByIds 

func (c *MockForgeClient) FindVpcsByIds(ctx context.Context, in *wflows.VpcsByIdsRequest, opts ...grpc.CallOption) (*wflows.VpcList, error)

func (*MockForgeClient) GetAllExpectedMachines 

func (c *MockForgeClient) GetAllExpectedMachines(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.ExpectedMachineList, error)

func (*MockForgeClient) GetAllExpectedMachinesLinked 

func (c *MockForgeClient) GetAllExpectedMachinesLinked(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.LinkedExpectedMachineList, error)

func (*MockForgeClient) GetAllExpectedPowerShelves 

func (c *MockForgeClient) GetAllExpectedPowerShelves(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.ExpectedPowerShelfList, error)

func (*MockForgeClient) GetAllExpectedPowerShelvesLinked 

func (c *MockForgeClient) GetAllExpectedPowerShelvesLinked(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.LinkedExpectedPowerShelfList, error)

func (*MockForgeClient) GetAllExpectedSwitches 

func (c *MockForgeClient) GetAllExpectedSwitches(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.ExpectedSwitchList, error)

func (*MockForgeClient) GetAllExpectedSwitchesLinked 

func (c *MockForgeClient) GetAllExpectedSwitchesLinked(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.LinkedExpectedSwitchList, error)

func (*MockForgeClient) GetAllSkuIds 

func (c *MockForgeClient) GetAllSkuIds(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*wflows.SkuIdList, error)

func (*MockForgeClient) GetDpuExtensionServiceVersionsInfo 

func (c *MockForgeClient) GetDpuExtensionServiceVersionsInfo(ctx context.Context, in *wflows.GetDpuExtensionServiceVersionsInfoRequest, opts ...grpc.CallOption) (*wflows.DpuExtensionServiceVersionInfoList, error)

func (*MockForgeClient) GetExpectedMachine 

func (c *MockForgeClient) GetExpectedMachine(ctx context.Context, in *wflows.ExpectedMachineRequest, opts ...grpc.CallOption) (*wflows.ExpectedMachine, error)

func (*MockForgeClient) GetMachineValidationExternalConfigs 

func (c *MockForgeClient) GetMachineValidationExternalConfigs(ctx context.Context, in *wflows.GetMachineValidationExternalConfigsRequest, opts ...grpc.CallOption) (*wflows.GetMachineValidationExternalConfigsResponse, error)

func (*MockForgeClient) GetMachineValidationResults 

func (c *MockForgeClient) GetMachineValidationResults(ctx context.Context, in *wflows.MachineValidationGetRequest, opts ...grpc.CallOption) (*wflows.MachineValidationResultList, error)

func (*MockForgeClient) GetMachineValidationRuns 

func (c *MockForgeClient) GetMachineValidationRuns(ctx context.Context, in *wflows.MachineValidationRunListGetRequest, opts ...grpc.CallOption) (*wflows.MachineValidationRunList, error)

func (*MockForgeClient) GetMachineValidationTests 

func (c *MockForgeClient) GetMachineValidationTests(ctx context.Context, in *wflows.MachineValidationTestsGetRequest, opts ...grpc.CallOption) (*wflows.MachineValidationTestsGetResponse, error)

func (*MockForgeClient) GetNetworkSecurityGroupAttachments 

func (c *MockForgeClient) GetNetworkSecurityGroupAttachments(ctx context.Context, in *wflows.GetNetworkSecurityGroupAttachmentsRequest, opts ...grpc.CallOption) (*wflows.GetNetworkSecurityGroupAttachmentsResponse, error)

func (*MockForgeClient) GetNetworkSecurityGroupPropagationStatus 

func (c *MockForgeClient) GetNetworkSecurityGroupPropagationStatus(ctx context.Context, in *wflows.GetNetworkSecurityGroupPropagationStatusRequest, opts ...grpc.CallOption) (*wflows.GetNetworkSecurityGroupPropagationStatusResponse, error)

func (*MockForgeClient) GetVpcPrefixes 

func (c *MockForgeClient) GetVpcPrefixes(ctx context.Context, in *wflows.VpcPrefixGetRequest, opts ...grpc.CallOption) (*wflows.VpcPrefixList, error)

func (*MockForgeClient) InvokeInstancePower 

func (c *MockForgeClient) InvokeInstancePower(ctx context.Context, in *wflows.InstancePowerRequest, opts ...grpc.CallOption) (*wflows.InstancePowerResult, error)

func (*MockForgeClient) ListOsImage 

func (c *MockForgeClient) ListOsImage(ctx context.Context, in *wflows.ListOsImageRequest, opts ...grpc.CallOption) (*wflows.ListOsImageResponse, error)

func (*MockForgeClient) MachineValidationTestEnableDisableTest 

func (c *MockForgeClient) MachineValidationTestEnableDisableTest(ctx context.Context, in *wflows.MachineValidationTestEnableDisableTestRequest, opts ...grpc.CallOption) (*wflows.MachineValidationTestEnableDisableTestResponse, error)

func (*MockForgeClient) NVLinkLogicalPartitionsForTenant 

func (c *MockForgeClient) NVLinkLogicalPartitionsForTenant(ctx context.Context, in *wflows.TenantSearchQuery, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartitionList, error)

func (*MockForgeClient) PersistValidationResult 

func (c *MockForgeClient) PersistValidationResult(ctx context.Context, in *wflows.MachineValidationResultPostRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) ReleaseInstance 

func (c *MockForgeClient) ReleaseInstance(ctx context.Context, in *wflows.InstanceReleaseRequest, opts ...grpc.CallOption) (*wflows.InstanceReleaseResult, error)

func (*MockForgeClient) RemoveMachineInstanceTypeAssociation 

func (c *MockForgeClient) RemoveMachineInstanceTypeAssociation(ctx context.Context, in *wflows.RemoveMachineInstanceTypeAssociationRequest, opts ...grpc.CallOption) (*wflows.RemoveMachineInstanceTypeAssociationResponse, error)

func (*MockForgeClient) RemoveMachineValidationExternalConfig 

func (c *MockForgeClient) RemoveMachineValidationExternalConfig(ctx context.Context, in *wflows.RemoveMachineValidationExternalConfigRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) SearchVpcPrefixes 

func (c *MockForgeClient) SearchVpcPrefixes(ctx context.Context, in *wflows.VpcPrefixSearchQuery, opts ...grpc.CallOption) (*wflows.VpcPrefixIdList, error)

func (*MockForgeClient) SetMaintenance 

func (c *MockForgeClient) SetMaintenance(ctx context.Context, in *wflows.MaintenanceRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

Machine mock methods

func (*MockForgeClient) UpdateDpuExtensionService 

func (c *MockForgeClient) UpdateDpuExtensionService(ctx context.Context, in *wflows.UpdateDpuExtensionServiceRequest, opts ...grpc.CallOption) (*wflows.DpuExtensionService, error)

func (*MockForgeClient) UpdateExpectedMachine 

func (c *MockForgeClient) UpdateExpectedMachine(ctx context.Context, in *wflows.ExpectedMachine, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) UpdateExpectedMachines 

func (c *MockForgeClient) UpdateExpectedMachines(ctx context.Context, in *wflows.BatchExpectedMachineOperationRequest, opts ...grpc.CallOption) (*wflows.BatchExpectedMachineOperationResponse, error)

func (*MockForgeClient) UpdateExpectedPowerShelf 

func (c *MockForgeClient) UpdateExpectedPowerShelf(ctx context.Context, in *wflows.ExpectedPowerShelf, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) UpdateExpectedSwitch 

func (c *MockForgeClient) UpdateExpectedSwitch(ctx context.Context, in *wflows.ExpectedSwitch, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) UpdateIBPartition added in v1.3.0 

func (c *MockForgeClient) UpdateIBPartition(ctx context.Context, in *wflows.IBPartitionUpdateRequest, opts ...grpc.CallOption) (*wflows.IBPartition, error)

func (*MockForgeClient) UpdateInstanceConfig 

func (c *MockForgeClient) UpdateInstanceConfig(ctx context.Context, in *wflows.InstanceConfigUpdateRequest, opts ...grpc.CallOption) (*wflows.Instance, error)

func (*MockForgeClient) UpdateInstanceType 

func (c *MockForgeClient) UpdateInstanceType(ctx context.Context, in *wflows.UpdateInstanceTypeRequest, opts ...grpc.CallOption) (*wflows.UpdateInstanceTypeResponse, error)

func (*MockForgeClient) UpdateMachineMetadata 

func (c *MockForgeClient) UpdateMachineMetadata(ctx context.Context, in *wflows.MachineMetadataUpdateRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockForgeClient) UpdateMachineValidationRun 

func (c *MockForgeClient) UpdateMachineValidationRun(ctx context.Context, in *wflows.MachineValidationRunRequest, opts ...grpc.CallOption) (*wflows.MachineValidationRunResponse, error)

Network Security Group mock methods

func (*MockForgeClient) UpdateMachineValidationTest 

func (c *MockForgeClient) UpdateMachineValidationTest(ctx context.Context, in *wflows.MachineValidationTestUpdateRequest, opts ...grpc.CallOption) (*wflows.MachineValidationTestAddUpdateResponse, error)

func (*MockForgeClient) UpdateNVLinkLogicalPartition 

func (c *MockForgeClient) UpdateNVLinkLogicalPartition(ctx context.Context, in *wflows.NVLinkLogicalPartitionUpdateRequest, opts ...grpc.CallOption) (*wflows.NVLinkLogicalPartitionUpdateResult, error)

func (*MockForgeClient) UpdateNetworkSecurityGroup 

func (c *MockForgeClient) UpdateNetworkSecurityGroup(ctx context.Context, in *wflows.UpdateNetworkSecurityGroupRequest, opts ...grpc.CallOption) (*wflows.UpdateNetworkSecurityGroupResponse, error)

func (*MockForgeClient) UpdateOsImage 

func (c *MockForgeClient) UpdateOsImage(ctx context.Context, in *wflows.OsImageAttributes, opts ...grpc.CallOption) (*wflows.OsImage, error)

func (*MockForgeClient) UpdateTenant 

func (c *MockForgeClient) UpdateTenant(ctx context.Context, in *wflows.UpdateTenantRequest, opts ...grpc.CallOption) (*wflows.UpdateTenantResponse, error)

func (*MockForgeClient) UpdateTenantKeyset 

func (c *MockForgeClient) UpdateTenantKeyset(ctx context.Context, in *wflows.UpdateTenantKeysetRequest, opts ...grpc.CallOption) (*wflows.UpdateTenantKeysetResponse, error)

func (*MockForgeClient) UpdateVpc 

func (c *MockForgeClient) UpdateVpc(ctx context.Context, in *wflows.VpcUpdateRequest, opts ...grpc.CallOption) (*wflows.VpcUpdateResult, error)

func (*MockForgeClient) UpdateVpcPrefix 

func (c *MockForgeClient) UpdateVpcPrefix(ctx context.Context, in *wflows.VpcPrefixUpdateRequest, opts ...grpc.CallOption) (*wflows.VpcPrefix, error)

func (*MockForgeClient) UpdateVpcVirtualization 

func (c *MockForgeClient) UpdateVpcVirtualization(ctx context.Context, in *wflows.VpcUpdateVirtualizationRequest, opts ...grpc.CallOption) (*wflows.VpcUpdateVirtualizationResult, error)

func (*MockForgeClient) Version 

func (c *MockForgeClient) Version(ctx context.Context, in *wflows.VersionRequest, opts ...grpc.CallOption) (*wflows.BuildInfo, error)

Version mock methods

type MockRLAClient 

type MockRLAClient struct {
	rlav1.RLAClient
}

MockRLAClient is a mock implementation of RLA gRPC protobuf Client

func (*MockRLAClient) AddComponent added in v1.2.0 

func (c *MockRLAClient) AddComponent(ctx context.Context, in *rlav1.AddComponentRequest, opts ...grpc.CallOption) (*rlav1.AddComponentResponse, error)

Component mutation mock methods

func (*MockRLAClient) AssociateRuleWithRack added in v1.2.0 

func (c *MockRLAClient) AssociateRuleWithRack(ctx context.Context, in *rlav1.AssociateRuleWithRackRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

Rack-rule association mock methods

func (*MockRLAClient) AttachRacksToNVLDomain 

func (c *MockRLAClient) AttachRacksToNVLDomain(ctx context.Context, in *rlav1.AttachRacksToNVLDomainRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) BringUpRack 

func (c *MockRLAClient) BringUpRack(ctx context.Context, in *rlav1.BringUpRackRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

func (*MockRLAClient) CancelTask added in v1.2.0 

func (c *MockRLAClient) CancelTask(ctx context.Context, in *rlav1.CancelTaskRequest, opts ...grpc.CallOption) (*rlav1.CancelTaskResponse, error)

func (*MockRLAClient) CreateExpectedRack 

func (c *MockRLAClient) CreateExpectedRack(ctx context.Context, in *rlav1.CreateExpectedRackRequest, opts ...grpc.CallOption) (*rlav1.CreateExpectedRackResponse, error)

Rack mock methods

func (*MockRLAClient) CreateNVLDomain 

func (c *MockRLAClient) CreateNVLDomain(ctx context.Context, in *rlav1.CreateNVLDomainRequest, opts ...grpc.CallOption) (*rlav1.CreateNVLDomainResponse, error)

NVL Domain mock methods

func (*MockRLAClient) CreateOperationRule added in v1.2.0 

func (c *MockRLAClient) CreateOperationRule(ctx context.Context, in *rlav1.CreateOperationRuleRequest, opts ...grpc.CallOption) (*rlav1.CreateOperationRuleResponse, error)

Operation rule mock methods

func (*MockRLAClient) DeleteComponent added in v1.2.0 

func (c *MockRLAClient) DeleteComponent(ctx context.Context, in *rlav1.DeleteComponentRequest, opts ...grpc.CallOption) (*rlav1.DeleteComponentResponse, error)

func (*MockRLAClient) DeleteOperationRule added in v1.2.0 

func (c *MockRLAClient) DeleteOperationRule(ctx context.Context, in *rlav1.DeleteOperationRuleRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) DetachRacksFromNVLDomain 

func (c *MockRLAClient) DetachRacksFromNVLDomain(ctx context.Context, in *rlav1.DetachRacksFromNVLDomainRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) DisassociateRuleFromRack added in v1.2.0 

func (c *MockRLAClient) DisassociateRuleFromRack(ctx context.Context, in *rlav1.DisassociateRuleFromRackRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) GetComponentInfoByID 

func (c *MockRLAClient) GetComponentInfoByID(ctx context.Context, in *rlav1.GetComponentInfoByIDRequest, opts ...grpc.CallOption) (*rlav1.GetComponentInfoResponse, error)

Component mock methods

func (*MockRLAClient) GetComponentInfoBySerial 

func (c *MockRLAClient) GetComponentInfoBySerial(ctx context.Context, in *rlav1.GetComponentInfoBySerialRequest, opts ...grpc.CallOption) (*rlav1.GetComponentInfoResponse, error)

func (*MockRLAClient) GetComponents 

func (c *MockRLAClient) GetComponents(ctx context.Context, in *rlav1.GetComponentsRequest, opts ...grpc.CallOption) (*rlav1.GetComponentsResponse, error)

func (*MockRLAClient) GetListOfNVLDomains 

func (c *MockRLAClient) GetListOfNVLDomains(ctx context.Context, in *rlav1.GetListOfNVLDomainsRequest, opts ...grpc.CallOption) (*rlav1.GetListOfNVLDomainsResponse, error)

func (*MockRLAClient) GetListOfRacks 

func (c *MockRLAClient) GetListOfRacks(ctx context.Context, in *rlav1.GetListOfRacksRequest, opts ...grpc.CallOption) (*rlav1.GetListOfRacksResponse, error)

func (*MockRLAClient) GetOperationRule added in v1.2.0 

func (c *MockRLAClient) GetOperationRule(ctx context.Context, in *rlav1.GetOperationRuleRequest, opts ...grpc.CallOption) (*rlav1.OperationRule, error)

func (*MockRLAClient) GetRackInfoByID 

func (c *MockRLAClient) GetRackInfoByID(ctx context.Context, in *rlav1.GetRackInfoByIDRequest, opts ...grpc.CallOption) (*rlav1.GetRackInfoResponse, error)

func (*MockRLAClient) GetRackInfoBySerial 

func (c *MockRLAClient) GetRackInfoBySerial(ctx context.Context, in *rlav1.GetRackInfoBySerialRequest, opts ...grpc.CallOption) (*rlav1.GetRackInfoResponse, error)

func (*MockRLAClient) GetRackRuleAssociation added in v1.2.0 

func (c *MockRLAClient) GetRackRuleAssociation(ctx context.Context, in *rlav1.GetRackRuleAssociationRequest, opts ...grpc.CallOption) (*rlav1.GetRackRuleAssociationResponse, error)

func (*MockRLAClient) GetRacksForNVLDomain 

func (c *MockRLAClient) GetRacksForNVLDomain(ctx context.Context, in *rlav1.GetRacksForNVLDomainRequest, opts ...grpc.CallOption) (*rlav1.GetRacksForNVLDomainResponse, error)

func (*MockRLAClient) GetTasksByIDs 

func (c *MockRLAClient) GetTasksByIDs(ctx context.Context, in *rlav1.GetTasksByIDsRequest, opts ...grpc.CallOption) (*rlav1.GetTasksByIDsResponse, error)

func (*MockRLAClient) IngestRack added in v1.2.0 

func (c *MockRLAClient) IngestRack(ctx context.Context, in *rlav1.IngestRackRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

func (*MockRLAClient) ListOperationRules added in v1.2.0 

func (c *MockRLAClient) ListOperationRules(ctx context.Context, in *rlav1.ListOperationRulesRequest, opts ...grpc.CallOption) (*rlav1.ListOperationRulesResponse, error)

func (*MockRLAClient) ListRackRuleAssociations added in v1.2.0 

func (c *MockRLAClient) ListRackRuleAssociations(ctx context.Context, in *rlav1.ListRackRuleAssociationsRequest, opts ...grpc.CallOption) (*rlav1.ListRackRuleAssociationsResponse, error)

func (*MockRLAClient) ListTasks 

func (c *MockRLAClient) ListTasks(ctx context.Context, in *rlav1.ListTasksRequest, opts ...grpc.CallOption) (*rlav1.ListTasksResponse, error)

func (*MockRLAClient) PatchComponent added in v1.2.0 

func (c *MockRLAClient) PatchComponent(ctx context.Context, in *rlav1.PatchComponentRequest, opts ...grpc.CallOption) (*rlav1.PatchComponentResponse, error)

func (*MockRLAClient) PatchRack 

func (c *MockRLAClient) PatchRack(ctx context.Context, in *rlav1.PatchRackRequest, opts ...grpc.CallOption) (*rlav1.PatchRackResponse, error)

func (*MockRLAClient) PowerOffRack 

func (c *MockRLAClient) PowerOffRack(ctx context.Context, in *rlav1.PowerOffRackRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

func (*MockRLAClient) PowerOnRack 

func (c *MockRLAClient) PowerOnRack(ctx context.Context, in *rlav1.PowerOnRackRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

func (*MockRLAClient) PowerResetRack 

func (c *MockRLAClient) PowerResetRack(ctx context.Context, in *rlav1.PowerResetRackRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

func (*MockRLAClient) SetRuleAsDefault added in v1.2.0 

func (c *MockRLAClient) SetRuleAsDefault(ctx context.Context, in *rlav1.SetRuleAsDefaultRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) UpdateOperationRule added in v1.2.0 

func (c *MockRLAClient) UpdateOperationRule(ctx context.Context, in *rlav1.UpdateOperationRuleRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)

func (*MockRLAClient) UpgradeFirmware 

func (c *MockRLAClient) UpgradeFirmware(ctx context.Context, in *rlav1.UpgradeFirmwareRequest, opts ...grpc.CallOption) (*rlav1.SubmitTaskResponse, error)

Task mock methods

func (*MockRLAClient) ValidateComponents 

func (c *MockRLAClient) ValidateComponents(ctx context.Context, in *rlav1.ValidateComponentsRequest, opts ...grpc.CallOption) (*rlav1.ValidateComponentsResponse, error)

func (*MockRLAClient) Version 

func (c *MockRLAClient) Version(ctx context.Context, in *rlav1.VersionRequest, opts ...grpc.CallOption) (*rlav1.BuildInfo, error)

Version mock methods

type RlaAtomicClient 

type RlaAtomicClient struct {
	Config *RlaClientConfig
	// contains filtered or unexported fields
}

RlaAtomicClient is an atomic wrapper around the RlaClient

func NewRlaAtomicClient 

func NewRlaAtomicClient(config *RlaClientConfig) *RlaAtomicClient

NewRlaAtomicClient creates a new RlaAtomicClient

func (*RlaAtomicClient) CheckAndReloadCerts 

func (rac *RlaAtomicClient) CheckAndReloadCerts(initialClientCertMD5, initialServerCAMD5 []byte)

CheckAndReloadCerts continuously monitors the TLS certificates for changes. If a change is detected, it reinitializes the RlaClient with the new certificates to ensure secure communication.

func (*RlaAtomicClient) CheckCertificates 

func (rac *RlaAtomicClient) CheckCertificates(lastClientCertMD5, lastServerCAMD5 []byte) (bool, []byte, []byte, error)

CheckCertificates checks if the client and server CA certificates have changed

func (*RlaAtomicClient) GetClient 

func (rac *RlaAtomicClient) GetClient() *RlaClient

GetClient returns the current version of Rla client from the atomic value. Returns nil if the client has not been initialized yet.

func (*RlaAtomicClient) GetInitialCertMD5 

func (rac *RlaAtomicClient) GetInitialCertMD5() (clientCertMD5, serverCAMD5 []byte, err error)

GetInitialCertMD5 retrieves the MD5 hash of the initial set of certificate that the client is Using

func (*RlaAtomicClient) SwapClient 

func (rac *RlaAtomicClient) SwapClient(newClient *RlaClient) *RlaClient

SwapClient atomically replaces the current RlaClient with a new one, returning the old client for the caller to manage.

func (*RlaAtomicClient) Version 

func (rac *RlaAtomicClient) Version() int64

Version returns the current version of the RlaClient

type RlaClient 

type RlaClient struct {
	// contains filtered or unexported fields
}

RlaClient is the data structure for the client

func NewMockRlaClient 

func NewMockRlaClient() *RlaClient

NewMockRlaClient creates a new mock RlaClient that can be used with RlaAtomicClient.SwapClient

func NewRlaClient 

func NewRlaClient(config *RlaClientConfig) (client *RlaClient, err error)

NewRlaClient creates a new RlaClient

func (*RlaClient) Close 

func (cc *RlaClient) Close() error

Close gracefully shuts down the client's gRPC connection.

func (*RlaClient) Rla 

func (client *RlaClient) Rla() rlav1.RLAClient

Rla client getter

type RlaClientConfig 

type RlaClientConfig struct {
	// The address of the server <host>:<port>
	Address string
	// Secure flag
	Secure RlaClientSecureOptions
	// Skip Server Auth
	SkipServerAuth bool
	// The TLS certificate for the server
	ServerCAPath string
	// The TLS certificate for the client
	ClientCertPath string
	// The TLS key for the client
	ClientKeyPath string
	// client metrics interface
	ClientMetrics Metrics
}

RlaClientConfig is the data structure for the client configuration

type RlaClientSecureOptions 

type RlaClientSecureOptions int

SecureOptions is the enum for the secure options 

const (
	// RlaInsecureGrpc is the insecure dial option
	RlaInsecureGrpc RlaClientSecureOptions = iota
	// RlaServerTLS is the secure dial option for server tls
	RlaServerTLS
	// RlaMutualTLS for mutual tls
	RlaMutualTLS
)

type SecureOptions 

type SecureOptions int

SecureOptions is the enum for the secure options 

const (
	// InsecuregRPC is the insecure dial option
	InsecuregRPC SecureOptions = iota
	// ServerTLS is the secure dial option for server tls
	ServerTLS
	// MutualTLS for mutual tls
	MutualTLS
)

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.