IntenseWebs/opentofu
3
0
Fork 0
mirror of https://github.com/opentofu/opentofu.git synced 2025-02-25 18:45:20 -06:00
Code Issues Packages Projects Releases Wiki Activity

plugin: Register v5 plugins only if present

Browse Source 
Previously we would construct both provisioner and the provider objects if
either callback was set, but this is incorrect because a plugin should
actually set only one of these at a time, depending on what kind of plugin
it is.
This commit is contained in:
Martin Atkins 2018-08-24 16:53:05 -07:00
parent a6b5980a4f
commit 5559d21855
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
plugin/serve.go
View File

@ -60,10 +60,10 @@ func Serve(opts *ServeOpts) {
} }
} }
if opts.GRPCProvisionerFunc == nil && opts.ProvisionerFunc != nil { if opts.GRPCProvisionerFunc == nil && opts.ProvisionerFunc != nil {
provider := grpcplugin.NewGRPCProvisionerServerShim(opts.ProvisionerFunc()) provisioner := grpcplugin.NewGRPCProvisionerServerShim(opts.ProvisionerFunc())
if provider != nil { if provisioner != nil {
opts.GRPCProvisionerFunc = func() proto.ProvisionerServer { opts.GRPCProvisionerFunc = func() proto.ProvisionerServer {
return provider return provisioner
} }
} }
} }
@ -77,7 +77,7 @@ func Serve(opts *ServeOpts) {
// pluginMap returns the legacy map[string]plugin.Plugin to use for configuring // pluginMap returns the legacy map[string]plugin.Plugin to use for configuring
// a plugin server or client. // a plugin server or client.
func pluginMap(opts *ServeOpts) map[string]plugin.Plugin { func legacyPluginMap(opts *ServeOpts) map[string]plugin.Plugin {
return map[string]plugin.Plugin{ return map[string]plugin.Plugin{
"provider": &ResourceProviderPlugin{ "provider": &ResourceProviderPlugin{
ResourceProvider: opts.ProviderFunc, ResourceProvider: opts.ProviderFunc,
@ -93,18 +93,21 @@ func pluginSet(opts *ServeOpts) map[int]plugin.PluginSet {
// The oldest version is returned in when executed by a legacy go-plugin // The oldest version is returned in when executed by a legacy go-plugin
// client. // client.
plugins := map[int]plugin.PluginSet{ plugins := map[int]plugin.PluginSet{
4: pluginMap(opts), 4: legacyPluginMap(opts),
} }
// add the new protocol versions if they're configured // add the new protocol versions if they're configured
if opts.GRPCProviderFunc != nil || opts.GRPCProvisionerFunc != nil { if opts.GRPCProviderFunc != nil || opts.GRPCProvisionerFunc != nil {
plugins[5] = plugin.PluginSet{ plugins[5] = plugin.PluginSet{}
"provider": &GRPCProviderPlugin{ if opts.GRPCProviderFunc != nil {
plugins[5]["provider"] = &GRPCProviderPlugin{
GRPCProvider: opts.GRPCProviderFunc, GRPCProvider: opts.GRPCProviderFunc,
}, }
"provisioner": &GRPCProvisionerPlugin{ }
if opts.GRPCProvisionerFunc != nil {
plugins[5]["provisioner"] = &GRPCProvisionerPlugin{
GRPCProvisioner: opts.GRPCProvisionerFunc, GRPCProvisioner: opts.GRPCProvisionerFunc,
}, }
} }
} }
return plugins return plugins