pptx704
6f0c365d44
Implement the host agent as a Connect RPC server that orchestrates sandbox creation, destruction, pause/resume, and command execution. Includes sandbox manager with TTL-based reaper, network slot allocator, rootfs cloning, hostagent proto definition with generated stubs, and test/debug scripts. Fix Firecracker process lifetime bug where VM was tied to HTTP request context instead of background context.
311 lines
15 KiB
Go
311 lines
15 KiB
Go
// Code generated by protoc-gen-connect-go. DO NOT EDIT.
|
|
//
|
|
// Source: process.proto
|
|
|
|
package specconnect
|
|
|
|
import (
|
|
connect "connectrpc.com/connect"
|
|
context "context"
|
|
errors "errors"
|
|
spec "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec"
|
|
http "net/http"
|
|
strings "strings"
|
|
)
|
|
|
|
// This is a compile-time assertion to ensure that this generated file and the connect package are
|
|
// compatible. If you get a compiler error that this constant is not defined, this code was
|
|
// generated with a version of connect newer than the one compiled into your binary. You can fix the
|
|
// problem by either regenerating this code with an older version of connect or updating the connect
|
|
// version compiled into your binary.
|
|
const _ = connect.IsAtLeastVersion1_13_0
|
|
|
|
const (
|
|
// ProcessName is the fully-qualified name of the Process service.
|
|
ProcessName = "process.Process"
|
|
)
|
|
|
|
// These constants are the fully-qualified names of the RPCs defined in this package. They're
|
|
// exposed at runtime as Spec.Procedure and as the final two segments of the HTTP route.
|
|
//
|
|
// Note that these are different from the fully-qualified method names used by
|
|
// google.golang.org/protobuf/reflect/protoreflect. To convert from these constants to
|
|
// reflection-formatted method names, remove the leading slash and convert the remaining slash to a
|
|
// period.
|
|
const (
|
|
// ProcessListProcedure is the fully-qualified name of the Process's List RPC.
|
|
ProcessListProcedure = "/process.Process/List"
|
|
// ProcessConnectProcedure is the fully-qualified name of the Process's Connect RPC.
|
|
ProcessConnectProcedure = "/process.Process/Connect"
|
|
// ProcessStartProcedure is the fully-qualified name of the Process's Start RPC.
|
|
ProcessStartProcedure = "/process.Process/Start"
|
|
// ProcessUpdateProcedure is the fully-qualified name of the Process's Update RPC.
|
|
ProcessUpdateProcedure = "/process.Process/Update"
|
|
// ProcessStreamInputProcedure is the fully-qualified name of the Process's StreamInput RPC.
|
|
ProcessStreamInputProcedure = "/process.Process/StreamInput"
|
|
// ProcessSendInputProcedure is the fully-qualified name of the Process's SendInput RPC.
|
|
ProcessSendInputProcedure = "/process.Process/SendInput"
|
|
// ProcessSendSignalProcedure is the fully-qualified name of the Process's SendSignal RPC.
|
|
ProcessSendSignalProcedure = "/process.Process/SendSignal"
|
|
// ProcessCloseStdinProcedure is the fully-qualified name of the Process's CloseStdin RPC.
|
|
ProcessCloseStdinProcedure = "/process.Process/CloseStdin"
|
|
)
|
|
|
|
// ProcessClient is a client for the process.Process service.
|
|
type ProcessClient interface {
|
|
List(context.Context, *connect.Request[spec.ListRequest]) (*connect.Response[spec.ListResponse], error)
|
|
Connect(context.Context, *connect.Request[spec.ConnectRequest]) (*connect.ServerStreamForClient[spec.ConnectResponse], error)
|
|
Start(context.Context, *connect.Request[spec.StartRequest]) (*connect.ServerStreamForClient[spec.StartResponse], error)
|
|
Update(context.Context, *connect.Request[spec.UpdateRequest]) (*connect.Response[spec.UpdateResponse], error)
|
|
// Client input stream ensures ordering of messages
|
|
StreamInput(context.Context) *connect.ClientStreamForClient[spec.StreamInputRequest, spec.StreamInputResponse]
|
|
SendInput(context.Context, *connect.Request[spec.SendInputRequest]) (*connect.Response[spec.SendInputResponse], error)
|
|
SendSignal(context.Context, *connect.Request[spec.SendSignalRequest]) (*connect.Response[spec.SendSignalResponse], error)
|
|
// Close stdin to signal EOF to the process.
|
|
// Only works for non-PTY processes. For PTY, send Ctrl+D (0x04) instead.
|
|
CloseStdin(context.Context, *connect.Request[spec.CloseStdinRequest]) (*connect.Response[spec.CloseStdinResponse], error)
|
|
}
|
|
|
|
// NewProcessClient constructs a client for the process.Process service. By default, it uses the
|
|
// Connect protocol with the binary Protobuf Codec, asks for gzipped responses, and sends
|
|
// uncompressed requests. To use the gRPC or gRPC-Web protocols, supply the connect.WithGRPC() or
|
|
// connect.WithGRPCWeb() options.
|
|
//
|
|
// The URL supplied here should be the base URL for the Connect or gRPC server (for example,
|
|
// http://api.acme.com or https://acme.com/grpc).
|
|
func NewProcessClient(httpClient connect.HTTPClient, baseURL string, opts ...connect.ClientOption) ProcessClient {
|
|
baseURL = strings.TrimRight(baseURL, "/")
|
|
processMethods := spec.File_process_proto.Services().ByName("Process").Methods()
|
|
return &processClient{
|
|
list: connect.NewClient[spec.ListRequest, spec.ListResponse](
|
|
httpClient,
|
|
baseURL+ProcessListProcedure,
|
|
connect.WithSchema(processMethods.ByName("List")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
connect: connect.NewClient[spec.ConnectRequest, spec.ConnectResponse](
|
|
httpClient,
|
|
baseURL+ProcessConnectProcedure,
|
|
connect.WithSchema(processMethods.ByName("Connect")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
start: connect.NewClient[spec.StartRequest, spec.StartResponse](
|
|
httpClient,
|
|
baseURL+ProcessStartProcedure,
|
|
connect.WithSchema(processMethods.ByName("Start")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
update: connect.NewClient[spec.UpdateRequest, spec.UpdateResponse](
|
|
httpClient,
|
|
baseURL+ProcessUpdateProcedure,
|
|
connect.WithSchema(processMethods.ByName("Update")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
streamInput: connect.NewClient[spec.StreamInputRequest, spec.StreamInputResponse](
|
|
httpClient,
|
|
baseURL+ProcessStreamInputProcedure,
|
|
connect.WithSchema(processMethods.ByName("StreamInput")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
sendInput: connect.NewClient[spec.SendInputRequest, spec.SendInputResponse](
|
|
httpClient,
|
|
baseURL+ProcessSendInputProcedure,
|
|
connect.WithSchema(processMethods.ByName("SendInput")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
sendSignal: connect.NewClient[spec.SendSignalRequest, spec.SendSignalResponse](
|
|
httpClient,
|
|
baseURL+ProcessSendSignalProcedure,
|
|
connect.WithSchema(processMethods.ByName("SendSignal")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
closeStdin: connect.NewClient[spec.CloseStdinRequest, spec.CloseStdinResponse](
|
|
httpClient,
|
|
baseURL+ProcessCloseStdinProcedure,
|
|
connect.WithSchema(processMethods.ByName("CloseStdin")),
|
|
connect.WithClientOptions(opts...),
|
|
),
|
|
}
|
|
}
|
|
|
|
// processClient implements ProcessClient.
|
|
type processClient struct {
|
|
list *connect.Client[spec.ListRequest, spec.ListResponse]
|
|
connect *connect.Client[spec.ConnectRequest, spec.ConnectResponse]
|
|
start *connect.Client[spec.StartRequest, spec.StartResponse]
|
|
update *connect.Client[spec.UpdateRequest, spec.UpdateResponse]
|
|
streamInput *connect.Client[spec.StreamInputRequest, spec.StreamInputResponse]
|
|
sendInput *connect.Client[spec.SendInputRequest, spec.SendInputResponse]
|
|
sendSignal *connect.Client[spec.SendSignalRequest, spec.SendSignalResponse]
|
|
closeStdin *connect.Client[spec.CloseStdinRequest, spec.CloseStdinResponse]
|
|
}
|
|
|
|
// List calls process.Process.List.
|
|
func (c *processClient) List(ctx context.Context, req *connect.Request[spec.ListRequest]) (*connect.Response[spec.ListResponse], error) {
|
|
return c.list.CallUnary(ctx, req)
|
|
}
|
|
|
|
// Connect calls process.Process.Connect.
|
|
func (c *processClient) Connect(ctx context.Context, req *connect.Request[spec.ConnectRequest]) (*connect.ServerStreamForClient[spec.ConnectResponse], error) {
|
|
return c.connect.CallServerStream(ctx, req)
|
|
}
|
|
|
|
// Start calls process.Process.Start.
|
|
func (c *processClient) Start(ctx context.Context, req *connect.Request[spec.StartRequest]) (*connect.ServerStreamForClient[spec.StartResponse], error) {
|
|
return c.start.CallServerStream(ctx, req)
|
|
}
|
|
|
|
// Update calls process.Process.Update.
|
|
func (c *processClient) Update(ctx context.Context, req *connect.Request[spec.UpdateRequest]) (*connect.Response[spec.UpdateResponse], error) {
|
|
return c.update.CallUnary(ctx, req)
|
|
}
|
|
|
|
// StreamInput calls process.Process.StreamInput.
|
|
func (c *processClient) StreamInput(ctx context.Context) *connect.ClientStreamForClient[spec.StreamInputRequest, spec.StreamInputResponse] {
|
|
return c.streamInput.CallClientStream(ctx)
|
|
}
|
|
|
|
// SendInput calls process.Process.SendInput.
|
|
func (c *processClient) SendInput(ctx context.Context, req *connect.Request[spec.SendInputRequest]) (*connect.Response[spec.SendInputResponse], error) {
|
|
return c.sendInput.CallUnary(ctx, req)
|
|
}
|
|
|
|
// SendSignal calls process.Process.SendSignal.
|
|
func (c *processClient) SendSignal(ctx context.Context, req *connect.Request[spec.SendSignalRequest]) (*connect.Response[spec.SendSignalResponse], error) {
|
|
return c.sendSignal.CallUnary(ctx, req)
|
|
}
|
|
|
|
// CloseStdin calls process.Process.CloseStdin.
|
|
func (c *processClient) CloseStdin(ctx context.Context, req *connect.Request[spec.CloseStdinRequest]) (*connect.Response[spec.CloseStdinResponse], error) {
|
|
return c.closeStdin.CallUnary(ctx, req)
|
|
}
|
|
|
|
// ProcessHandler is an implementation of the process.Process service.
|
|
type ProcessHandler interface {
|
|
List(context.Context, *connect.Request[spec.ListRequest]) (*connect.Response[spec.ListResponse], error)
|
|
Connect(context.Context, *connect.Request[spec.ConnectRequest], *connect.ServerStream[spec.ConnectResponse]) error
|
|
Start(context.Context, *connect.Request[spec.StartRequest], *connect.ServerStream[spec.StartResponse]) error
|
|
Update(context.Context, *connect.Request[spec.UpdateRequest]) (*connect.Response[spec.UpdateResponse], error)
|
|
// Client input stream ensures ordering of messages
|
|
StreamInput(context.Context, *connect.ClientStream[spec.StreamInputRequest]) (*connect.Response[spec.StreamInputResponse], error)
|
|
SendInput(context.Context, *connect.Request[spec.SendInputRequest]) (*connect.Response[spec.SendInputResponse], error)
|
|
SendSignal(context.Context, *connect.Request[spec.SendSignalRequest]) (*connect.Response[spec.SendSignalResponse], error)
|
|
// Close stdin to signal EOF to the process.
|
|
// Only works for non-PTY processes. For PTY, send Ctrl+D (0x04) instead.
|
|
CloseStdin(context.Context, *connect.Request[spec.CloseStdinRequest]) (*connect.Response[spec.CloseStdinResponse], error)
|
|
}
|
|
|
|
// NewProcessHandler builds an HTTP handler from the service implementation. It returns the path on
|
|
// which to mount the handler and the handler itself.
|
|
//
|
|
// By default, handlers support the Connect, gRPC, and gRPC-Web protocols with the binary Protobuf
|
|
// and JSON codecs. They also support gzip compression.
|
|
func NewProcessHandler(svc ProcessHandler, opts ...connect.HandlerOption) (string, http.Handler) {
|
|
processMethods := spec.File_process_proto.Services().ByName("Process").Methods()
|
|
processListHandler := connect.NewUnaryHandler(
|
|
ProcessListProcedure,
|
|
svc.List,
|
|
connect.WithSchema(processMethods.ByName("List")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processConnectHandler := connect.NewServerStreamHandler(
|
|
ProcessConnectProcedure,
|
|
svc.Connect,
|
|
connect.WithSchema(processMethods.ByName("Connect")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processStartHandler := connect.NewServerStreamHandler(
|
|
ProcessStartProcedure,
|
|
svc.Start,
|
|
connect.WithSchema(processMethods.ByName("Start")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processUpdateHandler := connect.NewUnaryHandler(
|
|
ProcessUpdateProcedure,
|
|
svc.Update,
|
|
connect.WithSchema(processMethods.ByName("Update")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processStreamInputHandler := connect.NewClientStreamHandler(
|
|
ProcessStreamInputProcedure,
|
|
svc.StreamInput,
|
|
connect.WithSchema(processMethods.ByName("StreamInput")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processSendInputHandler := connect.NewUnaryHandler(
|
|
ProcessSendInputProcedure,
|
|
svc.SendInput,
|
|
connect.WithSchema(processMethods.ByName("SendInput")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processSendSignalHandler := connect.NewUnaryHandler(
|
|
ProcessSendSignalProcedure,
|
|
svc.SendSignal,
|
|
connect.WithSchema(processMethods.ByName("SendSignal")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
processCloseStdinHandler := connect.NewUnaryHandler(
|
|
ProcessCloseStdinProcedure,
|
|
svc.CloseStdin,
|
|
connect.WithSchema(processMethods.ByName("CloseStdin")),
|
|
connect.WithHandlerOptions(opts...),
|
|
)
|
|
return "/process.Process/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.URL.Path {
|
|
case ProcessListProcedure:
|
|
processListHandler.ServeHTTP(w, r)
|
|
case ProcessConnectProcedure:
|
|
processConnectHandler.ServeHTTP(w, r)
|
|
case ProcessStartProcedure:
|
|
processStartHandler.ServeHTTP(w, r)
|
|
case ProcessUpdateProcedure:
|
|
processUpdateHandler.ServeHTTP(w, r)
|
|
case ProcessStreamInputProcedure:
|
|
processStreamInputHandler.ServeHTTP(w, r)
|
|
case ProcessSendInputProcedure:
|
|
processSendInputHandler.ServeHTTP(w, r)
|
|
case ProcessSendSignalProcedure:
|
|
processSendSignalHandler.ServeHTTP(w, r)
|
|
case ProcessCloseStdinProcedure:
|
|
processCloseStdinHandler.ServeHTTP(w, r)
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
})
|
|
}
|
|
|
|
// UnimplementedProcessHandler returns CodeUnimplemented from all methods.
|
|
type UnimplementedProcessHandler struct{}
|
|
|
|
func (UnimplementedProcessHandler) List(context.Context, *connect.Request[spec.ListRequest]) (*connect.Response[spec.ListResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.List is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) Connect(context.Context, *connect.Request[spec.ConnectRequest], *connect.ServerStream[spec.ConnectResponse]) error {
|
|
return connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.Connect is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) Start(context.Context, *connect.Request[spec.StartRequest], *connect.ServerStream[spec.StartResponse]) error {
|
|
return connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.Start is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) Update(context.Context, *connect.Request[spec.UpdateRequest]) (*connect.Response[spec.UpdateResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.Update is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) StreamInput(context.Context, *connect.ClientStream[spec.StreamInputRequest]) (*connect.Response[spec.StreamInputResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.StreamInput is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) SendInput(context.Context, *connect.Request[spec.SendInputRequest]) (*connect.Response[spec.SendInputResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.SendInput is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) SendSignal(context.Context, *connect.Request[spec.SendSignalRequest]) (*connect.Response[spec.SendSignalResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.SendSignal is not implemented"))
|
|
}
|
|
|
|
func (UnimplementedProcessHandler) CloseStdin(context.Context, *connect.Request[spec.CloseStdinRequest]) (*connect.Response[spec.CloseStdinResponse], error) {
|
|
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("process.Process.CloseStdin is not implemented"))
|
|
}
|