proxy

Documentation

Index

• Constants
• type AgentConfig
• type AgentServer
  • func NewAgentServer(l logrus.FieldLogger) *AgentServer
  • func (a *AgentServer) AgentEvent(ctx context.Context, e *pba.ModelEventMessage) (*pba.ModelEventResponse, error)
  • func (a *AgentServer) HandleModelEvent(event ModelEvent) error
  • func (a *AgentServer) Start(agentPort uint) error
  • func (a *AgentServer) Subscribe(request *pba.AgentSubscribeRequest, stream pba.AgentService_SubscribeServer) error
• type AgentSubscriber
• type ChainerConfig
• type Config
  • func GetConfig() *Config
  • func ParseArgs() *Config
• type LogConfig
• type ModelEvent
• type ProxyChainer
  • func New(logger log.FieldLogger) *ProxyChainer
  • func (pc *ProxyChainer) PipelineUpdateEvent(_ context.Context, msg *chainer.PipelineUpdateStatusMessage) (*chainer.PipelineUpdateStatusResponse, error)
  • func (pc *ProxyChainer) Start(port uint) error
  • func (pc *ProxyChainer) SubscribePipelineUpdates(request *chainer.PipelineSubscriptionRequest, ...) error
• type ProxyConfig
• type ProxyServer
  • func NewProxyServer(logger log.FieldLogger, es chan<- ModelEvent) *ProxyServer
  • func (p *ProxyServer) LoadModel(ctx context.Context, r *pb.LoadModelRequest) (*pb.LoadModelResponse, error)
  • func (p *ProxyServer) Start(port uint) error
  • func (p *ProxyServer) UnloadModel(ctx context.Context, r *pb.UnloadModelRequest) (*pb.UnloadModelResponse, error)
• type Scheduler
  • func NewScheduler(l logrus.FieldLogger, es <-chan ModelEvent, agentServer *AgentServer) *Scheduler
  • func (s *Scheduler) Start()
• type ServerKey

Constants

const (
	ProxyDefaultListenPort      = 8100
	ProxyDefaultEventBufferSize = 100
)

const (
	AgentDefaultListenPort = 9005
)

const (
	ChainerDefaultListenPort = 10101
)

const (
	LogDefaultLevel = logrus.InfoLevel
)

Types

type AgentConfig

type AgentConfig struct {
	AgentListenPort uint
}

type AgentServer

type AgentServer struct {
	pba.UnimplementedAgentServiceServer
	// contains filtered or unexported fields
}

func NewAgentServer

func NewAgentServer(l logrus.FieldLogger) *AgentServer

func (*AgentServer) AgentEvent

func (a *AgentServer) AgentEvent(ctx context.Context, e *pba.ModelEventMessage) (*pba.ModelEventResponse, error)

func (*AgentServer) HandleModelEvent

func (a *AgentServer) HandleModelEvent(event ModelEvent) error

func (*AgentServer) Start

func (a *AgentServer) Start(agentPort uint) error

func (*AgentServer) Subscribe

func (a *AgentServer) Subscribe(request *pba.AgentSubscribeRequest, stream pba.AgentService_SubscribeServer) error

type AgentSubscriber

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

type ChainerConfig

type ChainerConfig struct {
	ChainerListenPort uint
}

type Config

type Config struct {
	LogConfig
	ProxyConfig
	AgentConfig
	ChainerConfig
}

func GetConfig

func GetConfig() *Config

func ParseArgs

func ParseArgs() *Config

type LogConfig

type LogConfig struct {
	LogLevel logrus.Level
}

type ModelEvent

type ModelEvent struct {
	*agent.ModelOperationMessage
}

type ProxyChainer

type ProxyChainer struct {
	chainer.UnimplementedChainerServer
	// contains filtered or unexported fields
}

func New

func New(logger log.FieldLogger) *ProxyChainer

func (*ProxyChainer) PipelineUpdateEvent

func (pc *ProxyChainer) PipelineUpdateEvent(
	_ context.Context,
	msg *chainer.PipelineUpdateStatusMessage,
) (*chainer.PipelineUpdateStatusResponse, error)

func (*ProxyChainer) Start

func (pc *ProxyChainer) Start(port uint) error

func (*ProxyChainer) SubscribePipelineUpdates

func (pc *ProxyChainer) SubscribePipelineUpdates(
	request *chainer.PipelineSubscriptionRequest,
	subscription chainer.Chainer_SubscribePipelineUpdatesServer,
) error

type ProxyConfig

type ProxyConfig struct {
	ProxyListenPort      uint
	ProxyEventBufferSize uint
}

type ProxyServer

type ProxyServer struct {
	pb.UnimplementedSchedulerProxyServer
	// contains filtered or unexported fields
}

func NewProxyServer

func NewProxyServer(logger log.FieldLogger, es chan<- ModelEvent) *ProxyServer

func (*ProxyServer) LoadModel

func (p *ProxyServer) LoadModel(ctx context.Context, r *pb.LoadModelRequest) (*pb.LoadModelResponse, error)

func (*ProxyServer) Start

func (p *ProxyServer) Start(port uint) error

func (*ProxyServer) UnloadModel

func (p *ProxyServer) UnloadModel(ctx context.Context, r *pb.UnloadModelRequest) (*pb.UnloadModelResponse, error)

type Scheduler

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

func NewScheduler

func NewScheduler(l logrus.FieldLogger, es <-chan ModelEvent, agentServer *AgentServer) *Scheduler

func (*Scheduler) Start

func (s *Scheduler) Start()

type ServerKey

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