mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
feat: support hashicorp plugin in gPRC mode
This commit is contained in:
@@ -3,6 +3,8 @@ package pluginInternal
|
||||
import "github.com/hashicorp/go-plugin"
|
||||
|
||||
const PluginName = "debugtalk"
|
||||
const RPCPluginName = PluginName + "_rpc"
|
||||
const GRPCPluginName = PluginName + "_grpc"
|
||||
|
||||
// handshakeConfigs are used to just do a basic handshake between
|
||||
// a plugin and host. If the handshake fails, a user friendly error is shown.
|
||||
|
||||
108
plugin/inner/grpc.go
Normal file
108
plugin/inner/grpc.go
Normal file
@@ -0,0 +1,108 @@
|
||||
package pluginInternal
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
|
||||
"github.com/hashicorp/go-plugin"
|
||||
"github.com/pkg/errors"
|
||||
"github.com/rs/zerolog/log"
|
||||
"google.golang.org/grpc"
|
||||
|
||||
"github.com/httprunner/hrp/plugin/proto"
|
||||
)
|
||||
|
||||
// functionGRPCClient runs on the host side, it implements FuncCaller interface
|
||||
type functionGRPCClient struct {
|
||||
client proto.DebugTalkClient
|
||||
}
|
||||
|
||||
func (m *functionGRPCClient) GetNames() ([]string, error) {
|
||||
log.Info().Msg("function GetNames called on host side")
|
||||
resp, err := m.client.GetNames(context.Background(), &proto.Empty{})
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("gRPC call GetNames() failed")
|
||||
return nil, err
|
||||
}
|
||||
return resp.Names, err
|
||||
}
|
||||
|
||||
func (m *functionGRPCClient) Call(funcName string, funcArgs ...interface{}) (interface{}, error) {
|
||||
log.Info().Str("funcName", funcName).Interface("funcArgs", funcArgs).Msg("call function via gRPC")
|
||||
|
||||
req := &proto.CallRequest{
|
||||
Name: funcName,
|
||||
}
|
||||
funcArgBytes, err := json.Marshal(funcArgs)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to marshal Call() funcArgs")
|
||||
}
|
||||
req.Args = funcArgBytes
|
||||
|
||||
response, err := m.client.Call(context.Background(), req)
|
||||
if err != nil {
|
||||
log.Error().Err(err).
|
||||
Str("funcName", funcName).Interface("funcArgs", funcArgs).
|
||||
Msg("gRPC Call() failed")
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var resp interface{}
|
||||
err = json.Unmarshal(response.Value, &resp)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to unmarshal Call() response")
|
||||
}
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// Here is the gRPC server that functionGRPCClient talks to.
|
||||
type functionGRPCServer struct {
|
||||
proto.UnimplementedDebugTalkServer
|
||||
Impl FuncCaller
|
||||
}
|
||||
|
||||
func (m *functionGRPCServer) GetNames(ctx context.Context, req *proto.Empty) (*proto.GetNamesResponse, error) {
|
||||
log.Info().Interface("req", req).Msg("gRPC GetNames() called on plugin side")
|
||||
v, err := m.Impl.GetNames()
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("gRPC GetNames execution failed")
|
||||
return nil, err
|
||||
}
|
||||
return &proto.GetNamesResponse{Names: v}, err
|
||||
}
|
||||
|
||||
func (m *functionGRPCServer) Call(ctx context.Context, req *proto.CallRequest) (*proto.CallResponse, error) {
|
||||
var funcArgs []interface{}
|
||||
if err := json.Unmarshal(req.Args, &funcArgs); err != nil {
|
||||
return nil, errors.Wrap(err, "failed to unmarshal Call() funcArgs")
|
||||
}
|
||||
|
||||
log.Info().Interface("req", req).Msg("gRPC Call() called on plugin side")
|
||||
|
||||
v, err := m.Impl.Call(req.Name, funcArgs...)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Interface("req", req).Msg("gRPC Call() execution failed")
|
||||
return nil, err
|
||||
}
|
||||
|
||||
value, err := json.Marshal(v)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to marshal Call() response")
|
||||
}
|
||||
return &proto.CallResponse{Value: value}, err
|
||||
}
|
||||
|
||||
// HRPPlugin implements hashicorp's plugin.GRPCPlugin.
|
||||
type GRPCPlugin struct {
|
||||
plugin.Plugin
|
||||
Impl FuncCaller
|
||||
}
|
||||
|
||||
func (p *GRPCPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
||||
proto.RegisterDebugTalkServer(s, &functionGRPCServer{Impl: p.Impl})
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *GRPCPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
||||
return &functionGRPCClient{client: proto.NewDebugTalkClient(c)}, nil
|
||||
}
|
||||
@@ -19,8 +19,9 @@ type HashicorpPlugin struct {
|
||||
}
|
||||
|
||||
func (p *HashicorpPlugin) Init(path string) error {
|
||||
pluginName := GRPCPluginName
|
||||
loggerOptions := &hclog.LoggerOptions{
|
||||
Name: PluginName,
|
||||
Name: pluginName,
|
||||
Output: os.Stdout,
|
||||
}
|
||||
if p.logOn {
|
||||
@@ -32,10 +33,15 @@ func (p *HashicorpPlugin) Init(path string) error {
|
||||
client = plugin.NewClient(&plugin.ClientConfig{
|
||||
HandshakeConfig: HandshakeConfig,
|
||||
Plugins: map[string]plugin.Plugin{
|
||||
PluginName: &HRPPlugin{},
|
||||
RPCPluginName: &RPCPlugin{},
|
||||
GRPCPluginName: &GRPCPlugin{},
|
||||
},
|
||||
Cmd: exec.Command(path),
|
||||
Logger: hclog.New(loggerOptions),
|
||||
AllowedProtocols: []plugin.Protocol{
|
||||
plugin.ProtocolNetRPC,
|
||||
plugin.ProtocolGRPC,
|
||||
},
|
||||
})
|
||||
|
||||
// Connect via RPC
|
||||
@@ -46,7 +52,7 @@ func (p *HashicorpPlugin) Init(path string) error {
|
||||
}
|
||||
|
||||
// Request the plugin
|
||||
raw, err := rpcClient.Dispense(PluginName)
|
||||
raw, err := rpcClient.Dispense(pluginName)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("request plugin failed")
|
||||
return err
|
||||
|
||||
@@ -18,12 +18,12 @@ type funcData struct {
|
||||
Args []interface{} // function arguments
|
||||
}
|
||||
|
||||
// functionRPC runs on the host side.
|
||||
type functionRPC struct {
|
||||
// functionRPCClient runs on the host side, it implements FuncCaller interface
|
||||
type functionRPCClient struct {
|
||||
client *rpc.Client
|
||||
}
|
||||
|
||||
func (g *functionRPC) GetNames() ([]string, error) {
|
||||
func (g *functionRPCClient) GetNames() ([]string, error) {
|
||||
var resp []string
|
||||
err := g.client.Call("Plugin.GetNames", new(interface{}), &resp)
|
||||
if err != nil {
|
||||
@@ -34,7 +34,7 @@ func (g *functionRPC) GetNames() ([]string, error) {
|
||||
}
|
||||
|
||||
// host -> plugin
|
||||
func (g *functionRPC) Call(funcName string, funcArgs ...interface{}) (interface{}, error) {
|
||||
func (g *functionRPCClient) Call(funcName string, funcArgs ...interface{}) (interface{}, error) {
|
||||
log.Info().Str("funcName", funcName).Interface("funcArgs", funcArgs).Msg("call function via RPC")
|
||||
f := funcData{
|
||||
Name: funcName,
|
||||
@@ -47,7 +47,7 @@ func (g *functionRPC) Call(funcName string, funcArgs ...interface{}) (interface{
|
||||
if err != nil {
|
||||
log.Error().Err(err).
|
||||
Str("funcName", funcName).Interface("funcArgs", funcArgs).
|
||||
Msg("rpc call Call() failed")
|
||||
Msg("rpc Call() failed")
|
||||
return nil, err
|
||||
}
|
||||
return resp, nil
|
||||
@@ -60,11 +60,11 @@ type functionRPCServer struct {
|
||||
|
||||
// plugin execution
|
||||
func (s *functionRPCServer) GetNames(args interface{}, resp *[]string) error {
|
||||
log.Info().Interface("args", args).Msg("GetNames called on plugin side")
|
||||
log.Info().Interface("args", args).Msg("rpc GetNames() called on plugin side")
|
||||
var err error
|
||||
*resp, err = s.Impl.GetNames()
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("GetNames execution failed")
|
||||
log.Error().Err(err).Msg("rpc GetNames() execution failed")
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
@@ -72,26 +72,26 @@ func (s *functionRPCServer) GetNames(args interface{}, resp *[]string) error {
|
||||
|
||||
// plugin execution
|
||||
func (s *functionRPCServer) Call(args interface{}, resp *interface{}) error {
|
||||
log.Info().Interface("args", args).Msg("function called on plugin side")
|
||||
log.Info().Interface("args", args).Msg("rpc Call() called on plugin side")
|
||||
f := args.(*funcData)
|
||||
var err error
|
||||
*resp, err = s.Impl.Call(f.Name, f.Args...)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Interface("args", args).Msg("function execution failed")
|
||||
log.Error().Err(err).Interface("args", args).Msg("rpc Call() execution failed")
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// HRPPlugin implements hashicorp's plugin.Plugin.
|
||||
type HRPPlugin struct {
|
||||
// RPCPlugin implements hashicorp's plugin.Plugin.
|
||||
type RPCPlugin struct {
|
||||
Impl FuncCaller
|
||||
}
|
||||
|
||||
func (p *HRPPlugin) Server(*plugin.MuxBroker) (interface{}, error) {
|
||||
func (p *RPCPlugin) Server(*plugin.MuxBroker) (interface{}, error) {
|
||||
return &functionRPCServer{Impl: p.Impl}, nil
|
||||
}
|
||||
|
||||
func (HRPPlugin) Client(b *plugin.MuxBroker, c *rpc.Client) (interface{}, error) {
|
||||
return &functionRPC{client: c}, nil
|
||||
func (RPCPlugin) Client(b *plugin.MuxBroker, c *rpc.Client) (interface{}, error) {
|
||||
return &functionRPCClient{client: c}, nil
|
||||
}
|
||||
|
||||
@@ -7,6 +7,7 @@ import (
|
||||
|
||||
hclog "github.com/hashicorp/go-hclog"
|
||||
"github.com/hashicorp/go-plugin"
|
||||
"github.com/rs/zerolog/log"
|
||||
|
||||
pluginInternal "github.com/httprunner/hrp/plugin/inner"
|
||||
)
|
||||
@@ -50,21 +51,48 @@ func Register(funcName string, fn interface{}) {
|
||||
functions[funcName] = reflect.ValueOf(fn)
|
||||
}
|
||||
|
||||
// Serve starts a plugin server process.
|
||||
func Serve() {
|
||||
// Serve starts a plugin server process in RPC mode.
|
||||
func ServeRPC() {
|
||||
log.Info().Msg("start plugin server in RPC mode")
|
||||
funcPlugin := &functionPlugin{
|
||||
logger: hclog.New(&hclog.LoggerOptions{
|
||||
Name: pluginInternal.PluginName,
|
||||
Name: pluginInternal.RPCPluginName,
|
||||
Output: os.Stdout,
|
||||
Level: hclog.Info,
|
||||
}),
|
||||
functions: functions,
|
||||
}
|
||||
var pluginMap = map[string]plugin.Plugin{
|
||||
pluginInternal.PluginName: &pluginInternal.HRPPlugin{Impl: funcPlugin},
|
||||
pluginInternal.RPCPluginName: &pluginInternal.RPCPlugin{Impl: funcPlugin},
|
||||
}
|
||||
// start RPC server
|
||||
plugin.Serve(&plugin.ServeConfig{
|
||||
HandshakeConfig: pluginInternal.HandshakeConfig,
|
||||
Plugins: pluginMap,
|
||||
})
|
||||
}
|
||||
|
||||
// Serve starts a plugin server process in gRPC mode.
|
||||
func ServeGRPC() {
|
||||
log.Info().Msg("start plugin server in gRPC mode")
|
||||
funcPlugin := &functionPlugin{
|
||||
logger: hclog.New(&hclog.LoggerOptions{
|
||||
Name: pluginInternal.GRPCPluginName,
|
||||
Output: os.Stdout,
|
||||
Level: hclog.Info,
|
||||
}),
|
||||
functions: functions,
|
||||
}
|
||||
var pluginMap = map[string]plugin.Plugin{
|
||||
pluginInternal.GRPCPluginName: &pluginInternal.GRPCPlugin{Impl: funcPlugin},
|
||||
}
|
||||
// start gRPC server
|
||||
plugin.Serve(&plugin.ServeConfig{
|
||||
HandshakeConfig: pluginInternal.HandshakeConfig,
|
||||
Plugins: pluginMap,
|
||||
GRPCServer: plugin.DefaultGRPCServer,
|
||||
})
|
||||
}
|
||||
|
||||
// default to run plugin in gRPC mode
|
||||
var Serve = ServeGRPC
|
||||
|
||||
10
plugin/proto/README.md
Normal file
10
plugin/proto/README.md
Normal file
@@ -0,0 +1,10 @@
|
||||
|
||||
## Updating the Protocol
|
||||
|
||||
If you update the protocol buffers file, you can regenerate the file using the following command from the project root directory. You do not need to run this if you're just using the plugin.
|
||||
|
||||
For Go:
|
||||
|
||||
```bash
|
||||
$ protoc --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative plugin/proto/debugtalk.proto
|
||||
```
|
||||
341
plugin/proto/debugtalk.pb.go
Normal file
341
plugin/proto/debugtalk.pb.go
Normal file
@@ -0,0 +1,341 @@
|
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.26.0
|
||||
// protoc v3.19.4
|
||||
// source: plugin/proto/debugtalk.proto
|
||||
|
||||
package proto
|
||||
|
||||
import (
|
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
||||
reflect "reflect"
|
||||
sync "sync"
|
||||
)
|
||||
|
||||
const (
|
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||
)
|
||||
|
||||
type Empty struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
}
|
||||
|
||||
func (x *Empty) Reset() {
|
||||
*x = Empty{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[0]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *Empty) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*Empty) ProtoMessage() {}
|
||||
|
||||
func (x *Empty) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[0]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use Empty.ProtoReflect.Descriptor instead.
|
||||
func (*Empty) Descriptor() ([]byte, []int) {
|
||||
return file_plugin_proto_debugtalk_proto_rawDescGZIP(), []int{0}
|
||||
}
|
||||
|
||||
type GetNamesResponse struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
Names []string `protobuf:"bytes,1,rep,name=names,proto3" json:"names,omitempty"`
|
||||
}
|
||||
|
||||
func (x *GetNamesResponse) Reset() {
|
||||
*x = GetNamesResponse{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[1]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *GetNamesResponse) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*GetNamesResponse) ProtoMessage() {}
|
||||
|
||||
func (x *GetNamesResponse) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[1]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use GetNamesResponse.ProtoReflect.Descriptor instead.
|
||||
func (*GetNamesResponse) Descriptor() ([]byte, []int) {
|
||||
return file_plugin_proto_debugtalk_proto_rawDescGZIP(), []int{1}
|
||||
}
|
||||
|
||||
func (x *GetNamesResponse) GetNames() []string {
|
||||
if x != nil {
|
||||
return x.Names
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type CallRequest struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
|
||||
Args []byte `protobuf:"bytes,2,opt,name=args,proto3" json:"args,omitempty"` // []interface{}
|
||||
}
|
||||
|
||||
func (x *CallRequest) Reset() {
|
||||
*x = CallRequest{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[2]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *CallRequest) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*CallRequest) ProtoMessage() {}
|
||||
|
||||
func (x *CallRequest) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[2]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use CallRequest.ProtoReflect.Descriptor instead.
|
||||
func (*CallRequest) Descriptor() ([]byte, []int) {
|
||||
return file_plugin_proto_debugtalk_proto_rawDescGZIP(), []int{2}
|
||||
}
|
||||
|
||||
func (x *CallRequest) GetName() string {
|
||||
if x != nil {
|
||||
return x.Name
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *CallRequest) GetArgs() []byte {
|
||||
if x != nil {
|
||||
return x.Args
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type CallResponse struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
Value []byte `protobuf:"bytes,1,opt,name=value,proto3" json:"value,omitempty"` // interface{}
|
||||
}
|
||||
|
||||
func (x *CallResponse) Reset() {
|
||||
*x = CallResponse{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[3]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *CallResponse) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*CallResponse) ProtoMessage() {}
|
||||
|
||||
func (x *CallResponse) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_plugin_proto_debugtalk_proto_msgTypes[3]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use CallResponse.ProtoReflect.Descriptor instead.
|
||||
func (*CallResponse) Descriptor() ([]byte, []int) {
|
||||
return file_plugin_proto_debugtalk_proto_rawDescGZIP(), []int{3}
|
||||
}
|
||||
|
||||
func (x *CallResponse) GetValue() []byte {
|
||||
if x != nil {
|
||||
return x.Value
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
var File_plugin_proto_debugtalk_proto protoreflect.FileDescriptor
|
||||
|
||||
var file_plugin_proto_debugtalk_proto_rawDesc = []byte{
|
||||
0x0a, 0x1c, 0x70, 0x6c, 0x75, 0x67, 0x69, 0x6e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x64,
|
||||
0x65, 0x62, 0x75, 0x67, 0x74, 0x61, 0x6c, 0x6b, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x05,
|
||||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x07, 0x0a, 0x05, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x28,
|
||||
0x0a, 0x10, 0x47, 0x65, 0x74, 0x4e, 0x61, 0x6d, 0x65, 0x73, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e,
|
||||
0x73, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x6e, 0x61, 0x6d, 0x65, 0x73, 0x18, 0x01, 0x20, 0x03, 0x28,
|
||||
0x09, 0x52, 0x05, 0x6e, 0x61, 0x6d, 0x65, 0x73, 0x22, 0x35, 0x0a, 0x0b, 0x43, 0x61, 0x6c, 0x6c,
|
||||
0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x12, 0x0a, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x18,
|
||||
0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x61,
|
||||
0x72, 0x67, 0x73, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x61, 0x72, 0x67, 0x73, 0x22,
|
||||
0x24, 0x0a, 0x0c, 0x43, 0x61, 0x6c, 0x6c, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12,
|
||||
0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x05,
|
||||
0x76, 0x61, 0x6c, 0x75, 0x65, 0x32, 0x6f, 0x0a, 0x09, 0x44, 0x65, 0x62, 0x75, 0x67, 0x54, 0x61,
|
||||
0x6c, 0x6b, 0x12, 0x31, 0x0a, 0x08, 0x47, 0x65, 0x74, 0x4e, 0x61, 0x6d, 0x65, 0x73, 0x12, 0x0c,
|
||||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x1a, 0x17, 0x2e, 0x70,
|
||||
0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x47, 0x65, 0x74, 0x4e, 0x61, 0x6d, 0x65, 0x73, 0x52, 0x65, 0x73,
|
||||
0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x2f, 0x0a, 0x04, 0x43, 0x61, 0x6c, 0x6c, 0x12, 0x12, 0x2e,
|
||||
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x43, 0x61, 0x6c, 0x6c, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73,
|
||||
0x74, 0x1a, 0x13, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x43, 0x61, 0x6c, 0x6c, 0x52, 0x65,
|
||||
0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x42, 0x28, 0x5a, 0x26, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62,
|
||||
0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x68, 0x74, 0x74, 0x70, 0x72, 0x75, 0x6e, 0x6e, 0x65, 0x72, 0x2f,
|
||||
0x68, 0x72, 0x70, 0x2f, 0x70, 0x6c, 0x75, 0x67, 0x69, 0x6e, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f,
|
||||
0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
||||
}
|
||||
|
||||
var (
|
||||
file_plugin_proto_debugtalk_proto_rawDescOnce sync.Once
|
||||
file_plugin_proto_debugtalk_proto_rawDescData = file_plugin_proto_debugtalk_proto_rawDesc
|
||||
)
|
||||
|
||||
func file_plugin_proto_debugtalk_proto_rawDescGZIP() []byte {
|
||||
file_plugin_proto_debugtalk_proto_rawDescOnce.Do(func() {
|
||||
file_plugin_proto_debugtalk_proto_rawDescData = protoimpl.X.CompressGZIP(file_plugin_proto_debugtalk_proto_rawDescData)
|
||||
})
|
||||
return file_plugin_proto_debugtalk_proto_rawDescData
|
||||
}
|
||||
|
||||
var file_plugin_proto_debugtalk_proto_msgTypes = make([]protoimpl.MessageInfo, 4)
|
||||
var file_plugin_proto_debugtalk_proto_goTypes = []interface{}{
|
||||
(*Empty)(nil), // 0: proto.Empty
|
||||
(*GetNamesResponse)(nil), // 1: proto.GetNamesResponse
|
||||
(*CallRequest)(nil), // 2: proto.CallRequest
|
||||
(*CallResponse)(nil), // 3: proto.CallResponse
|
||||
}
|
||||
var file_plugin_proto_debugtalk_proto_depIdxs = []int32{
|
||||
0, // 0: proto.DebugTalk.GetNames:input_type -> proto.Empty
|
||||
2, // 1: proto.DebugTalk.Call:input_type -> proto.CallRequest
|
||||
1, // 2: proto.DebugTalk.GetNames:output_type -> proto.GetNamesResponse
|
||||
3, // 3: proto.DebugTalk.Call:output_type -> proto.CallResponse
|
||||
2, // [2:4] is the sub-list for method output_type
|
||||
0, // [0:2] is the sub-list for method input_type
|
||||
0, // [0:0] is the sub-list for extension type_name
|
||||
0, // [0:0] is the sub-list for extension extendee
|
||||
0, // [0:0] is the sub-list for field type_name
|
||||
}
|
||||
|
||||
func init() { file_plugin_proto_debugtalk_proto_init() }
|
||||
func file_plugin_proto_debugtalk_proto_init() {
|
||||
if File_plugin_proto_debugtalk_proto != nil {
|
||||
return
|
||||
}
|
||||
if !protoimpl.UnsafeEnabled {
|
||||
file_plugin_proto_debugtalk_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*Empty); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
file_plugin_proto_debugtalk_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*GetNamesResponse); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
file_plugin_proto_debugtalk_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*CallRequest); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
file_plugin_proto_debugtalk_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*CallResponse); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
type x struct{}
|
||||
out := protoimpl.TypeBuilder{
|
||||
File: protoimpl.DescBuilder{
|
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||
RawDescriptor: file_plugin_proto_debugtalk_proto_rawDesc,
|
||||
NumEnums: 0,
|
||||
NumMessages: 4,
|
||||
NumExtensions: 0,
|
||||
NumServices: 1,
|
||||
},
|
||||
GoTypes: file_plugin_proto_debugtalk_proto_goTypes,
|
||||
DependencyIndexes: file_plugin_proto_debugtalk_proto_depIdxs,
|
||||
MessageInfos: file_plugin_proto_debugtalk_proto_msgTypes,
|
||||
}.Build()
|
||||
File_plugin_proto_debugtalk_proto = out.File
|
||||
file_plugin_proto_debugtalk_proto_rawDesc = nil
|
||||
file_plugin_proto_debugtalk_proto_goTypes = nil
|
||||
file_plugin_proto_debugtalk_proto_depIdxs = nil
|
||||
}
|
||||
24
plugin/proto/debugtalk.proto
Normal file
24
plugin/proto/debugtalk.proto
Normal file
@@ -0,0 +1,24 @@
|
||||
syntax = "proto3";
|
||||
package proto;
|
||||
|
||||
option go_package = "github.com/httprunner/hrp/plugin/proto";
|
||||
|
||||
message Empty {}
|
||||
|
||||
message GetNamesResponse {
|
||||
repeated string names = 1;
|
||||
}
|
||||
|
||||
message CallRequest {
|
||||
string name = 1;
|
||||
bytes args = 2; // []interface{}
|
||||
}
|
||||
|
||||
message CallResponse {
|
||||
bytes value = 1; // interface{}
|
||||
}
|
||||
|
||||
service DebugTalk {
|
||||
rpc GetNames(Empty) returns (GetNamesResponse);
|
||||
rpc Call(CallRequest) returns (CallResponse);
|
||||
}
|
||||
137
plugin/proto/debugtalk_grpc.pb.go
Normal file
137
plugin/proto/debugtalk_grpc.pb.go
Normal file
@@ -0,0 +1,137 @@
|
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
|
||||
package proto
|
||||
|
||||
import (
|
||||
context "context"
|
||||
grpc "google.golang.org/grpc"
|
||||
codes "google.golang.org/grpc/codes"
|
||||
status "google.golang.org/grpc/status"
|
||||
)
|
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.32.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion7
|
||||
|
||||
// DebugTalkClient is the client API for DebugTalk service.
|
||||
//
|
||||
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
||||
type DebugTalkClient interface {
|
||||
GetNames(ctx context.Context, in *Empty, opts ...grpc.CallOption) (*GetNamesResponse, error)
|
||||
Call(ctx context.Context, in *CallRequest, opts ...grpc.CallOption) (*CallResponse, error)
|
||||
}
|
||||
|
||||
type debugTalkClient struct {
|
||||
cc grpc.ClientConnInterface
|
||||
}
|
||||
|
||||
func NewDebugTalkClient(cc grpc.ClientConnInterface) DebugTalkClient {
|
||||
return &debugTalkClient{cc}
|
||||
}
|
||||
|
||||
func (c *debugTalkClient) GetNames(ctx context.Context, in *Empty, opts ...grpc.CallOption) (*GetNamesResponse, error) {
|
||||
out := new(GetNamesResponse)
|
||||
err := c.cc.Invoke(ctx, "/proto.DebugTalk/GetNames", in, out, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *debugTalkClient) Call(ctx context.Context, in *CallRequest, opts ...grpc.CallOption) (*CallResponse, error) {
|
||||
out := new(CallResponse)
|
||||
err := c.cc.Invoke(ctx, "/proto.DebugTalk/Call", in, out, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// DebugTalkServer is the server API for DebugTalk service.
|
||||
// All implementations must embed UnimplementedDebugTalkServer
|
||||
// for forward compatibility
|
||||
type DebugTalkServer interface {
|
||||
GetNames(context.Context, *Empty) (*GetNamesResponse, error)
|
||||
Call(context.Context, *CallRequest) (*CallResponse, error)
|
||||
mustEmbedUnimplementedDebugTalkServer()
|
||||
}
|
||||
|
||||
// UnimplementedDebugTalkServer must be embedded to have forward compatible implementations.
|
||||
type UnimplementedDebugTalkServer struct {
|
||||
}
|
||||
|
||||
func (UnimplementedDebugTalkServer) GetNames(context.Context, *Empty) (*GetNamesResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method GetNames not implemented")
|
||||
}
|
||||
func (UnimplementedDebugTalkServer) Call(context.Context, *CallRequest) (*CallResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method Call not implemented")
|
||||
}
|
||||
func (UnimplementedDebugTalkServer) mustEmbedUnimplementedDebugTalkServer() {}
|
||||
|
||||
// UnsafeDebugTalkServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to DebugTalkServer will
|
||||
// result in compilation errors.
|
||||
type UnsafeDebugTalkServer interface {
|
||||
mustEmbedUnimplementedDebugTalkServer()
|
||||
}
|
||||
|
||||
func RegisterDebugTalkServer(s grpc.ServiceRegistrar, srv DebugTalkServer) {
|
||||
s.RegisterService(&DebugTalk_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
func _DebugTalk_GetNames_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(Empty)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(DebugTalkServer).GetNames(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: "/proto.DebugTalk/GetNames",
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(DebugTalkServer).GetNames(ctx, req.(*Empty))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _DebugTalk_Call_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(CallRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(DebugTalkServer).Call(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: "/proto.DebugTalk/Call",
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(DebugTalkServer).Call(ctx, req.(*CallRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
// DebugTalk_ServiceDesc is the grpc.ServiceDesc for DebugTalk service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
// and not to be introspected or modified (even as a copy)
|
||||
var DebugTalk_ServiceDesc = grpc.ServiceDesc{
|
||||
ServiceName: "proto.DebugTalk",
|
||||
HandlerType: (*DebugTalkServer)(nil),
|
||||
Methods: []grpc.MethodDesc{
|
||||
{
|
||||
MethodName: "GetNames",
|
||||
Handler: _DebugTalk_GetNames_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "Call",
|
||||
Handler: _DebugTalk_Call_Handler,
|
||||
},
|
||||
},
|
||||
Streams: []grpc.StreamDesc{},
|
||||
Metadata: "plugin/proto/debugtalk.proto",
|
||||
}
|
||||
Reference in New Issue
Block a user