mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-08 08:54:29 +01:00
ace81155a4
This commit adds a new gRPC function Status to CacheService. VMCache server will reply the status of VMCache server. Factory destroy will call gRPC Status to get the status of VMCache server and output it when VMCache is enabled. Fixes: #1395 Signed-off-by: Hui Zhu <teawater@hyper.sh>
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
// Copyright (c) 2019 HyperHQ Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// grpccache implements base vm factory that get base vm from grpc
|
|
|
|
package grpccache
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
types "github.com/gogo/protobuf/types"
|
|
pb "github.com/kata-containers/runtime/protocols/cache"
|
|
vc "github.com/kata-containers/runtime/virtcontainers"
|
|
"github.com/kata-containers/runtime/virtcontainers/factory/base"
|
|
"github.com/pkg/errors"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type grpccache struct {
|
|
conn *grpc.ClientConn
|
|
config *vc.VMConfig
|
|
}
|
|
|
|
// New returns a new direct vm factory.
|
|
func New(ctx context.Context, endpoint string) (base.FactoryBase, error) {
|
|
conn, err := grpc.Dial(fmt.Sprintf("unix://%s", endpoint), grpc.WithInsecure())
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to connect %q", endpoint)
|
|
}
|
|
|
|
jConfig, err := pb.NewCacheServiceClient(conn).Config(ctx, &types.Empty{})
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to Config")
|
|
}
|
|
|
|
config, err := vc.GrpcToVMConfig(jConfig)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to convert JSON to VMConfig")
|
|
}
|
|
|
|
return &grpccache{conn: conn, config: config}, nil
|
|
}
|
|
|
|
// Config returns the direct factory's configuration.
|
|
func (g *grpccache) Config() vc.VMConfig {
|
|
return *g.config
|
|
}
|
|
|
|
// GetBaseVM create a new VM directly.
|
|
func (g *grpccache) GetBaseVM(ctx context.Context, config vc.VMConfig) (*vc.VM, error) {
|
|
defer g.conn.Close()
|
|
gVM, err := pb.NewCacheServiceClient(g.conn).GetBaseVM(ctx, &types.Empty{})
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to GetBaseVM")
|
|
}
|
|
return vc.NewVMFromGrpc(ctx, gVM, *g.config)
|
|
}
|
|
|
|
// CloseFactory closes the direct vm factory.
|
|
func (g *grpccache) CloseFactory(ctx context.Context) {
|
|
}
|
|
|
|
// GetVMStatus is not supported
|
|
func (g *grpccache) GetVMStatus() []*pb.GrpcVMStatus {
|
|
panic("ERROR: package grpccache does not support GetVMStatus")
|
|
}
|