GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-07 00:39:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support hashicorp plugin in gPRC mode

Browse Source
This commit is contained in:
debugtalk
2022-03-04 16:13:38 +08:00
parent c01e8dbfc7
commit 192f95c315
9 changed files with 677 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
plugin/inner/hashicorp_plugin.go
View File

@@ -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