From bfe5b5b4b6d09995da7f7c3f6f6cdb14295fa9d0 Mon Sep 17 00:00:00 2001 From: Chris Bednarski Date: Tue, 18 Aug 2015 15:21:11 -0700 Subject: [PATCH] Change if/else logic to switch --- command/plugin.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/command/plugin.go b/command/plugin.go index 60776fc54..91a8fe2b4 100644 --- a/command/plugin.go +++ b/command/plugin.go @@ -126,21 +126,22 @@ func (c *PluginCommand) Run(args []string) int { return 1 } - if pluginType == "builder" { + switch pluginType { + case "builder": builder, found := Builders[pluginName] if !found { c.Ui.Error(fmt.Sprintf("Could not load builder: %s", pluginName)) return 1 } server.RegisterBuilder(builder) - } else if pluginType == "provisioner" { + case "provisioner": provisioner, found := Provisioners[pluginName] if !found { c.Ui.Error(fmt.Sprintf("Could not load provisioner: %s", pluginName)) return 1 } server.RegisterProvisioner(provisioner) - } else if pluginType == "post-processor" { + case "post-processor": postProcessor, found := PostProcessors[pluginName] if !found { c.Ui.Error(fmt.Sprintf("Could not load post-processor: %s", pluginName))