github/dokku
0
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: add parallelism to certain proxy commands

Browse Source 
Closes #4392
This commit is contained in:
Jose Diaz-Gonzalez
2021-02-19 23:44:41 -05:00
parent 5141360b48
commit 04a96125cc
5 changed files with 103 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
plugins/proxy/proxy.go
View File

@@ -3,9 +3,14 @@ package proxy
import (
"fmt"
"github.com/dokku/dokku/plugins/common"
"github.com/dokku/dokku/plugins/config"
)
// RunInSerial is the default value for whether to run a command in parallel or not
// and defaults to -1 (false)
const RunInSerial = 0
// PortMap is a struct that contains a scheme:host-port:container-port mapping
type PortMap struct {
ContainerPort int
@@ -22,6 +27,46 @@ func (p PortMap) AllowsPersistence() bool {
return p.Scheme == "__internal__"
}
// BuildConfig rebuilds the proxy config for the specified app
func BuildConfig(appName string) error {
return common.PlugnTrigger("proxy-build-config", []string{appName}...)
}
// Disable disables proxy implementations for the specified app
func Disable(appName string) error {
if !IsAppProxyEnabled(appName) {
common.LogInfo1("Proxy is already disable for app")
return nil
}
common.LogInfo1("Disabling proxy for app")
entries := map[string]string{
"DOKKU_DISABLE_PROXY": "1",
}
if err := config.SetMany(appName, entries, false); err != nil {
return err
}
return common.PlugnTrigger("proxy-disable", []string{appName}...)
}
// Enable enables proxy implementations for the specified app
func Enable(appName string) error {
if IsAppProxyEnabled(appName) {
common.LogInfo1("Proxy is already enabled for app")
return nil
}
common.LogInfo1("Enabling proxy for app")
keys := []string{"DOKKU_DISABLE_PROXY"}
if err := config.UnsetMany(appName, keys, false); err != nil {
return err
}
return common.PlugnTrigger("proxy-enable", []string{appName}...)
}
// IsAppProxyEnabled returns true if proxy is enabled; otherwise return false
func IsAppProxyEnabled(appName string) bool {
proxyEnabled := true

12
plugins/proxy/src/subcommands/subcommands.go
View File

@@ -20,19 +20,25 @@ func main() {
switch subcommand {
case "build-config":
args := flag.NewFlagSet("proxy:build-config", flag.ExitOnError)
allApps := args.Bool("all", false, "--all: build-config for all apps")
parallelCount := args.Int("parallel", proxy.RunInSerial, "--parallel: number of apps to build-config for in parallel, -1 to match cpu count")
args.Parse(os.Args[2:])
appName := args.Arg(0)
err = proxy.CommandBuildConfig(appName)
err = proxy.CommandBuildConfig(appName, *allApps, *parallelCount)
case "disable":
args := flag.NewFlagSet("proxy:disable", flag.ExitOnError)
allApps := args.Bool("all", false, "--all: disable proxy for all apps")
parallelCount := args.Int("parallel", proxy.RunInSerial, "--parallel: number of apps to disable proxy for in parallel, -1 to match cpu count")
args.Parse(os.Args[2:])
appName := args.Arg(0)
err = proxy.CommandDisable(appName)
err = proxy.CommandDisable(appName, *allApps, *parallelCount)
case "enable":
args := flag.NewFlagSet("proxy:enable", flag.ExitOnError)
allApps := args.Bool("all", false, "--all: enable proxy for all apps")
parallelCount := args.Int("parallel", proxy.RunInSerial, "--parallel: number of apps to enable proxy for in parallel, -1 to match cpu count")
args.Parse(os.Args[2:])
appName := args.Arg(0)
err = proxy.CommandEnable(appName)
err = proxy.CommandEnable(appName, *allApps, *parallelCount)
case "ports":
args := flag.NewFlagSet("proxy:ports", flag.ExitOnError)
args.Parse(os.Args[2:])

49
plugins/proxy/subcommands.go
View File

@@ -9,55 +9,42 @@ import (
)
// CommandBuildConfig rebuilds config for a given app
func CommandBuildConfig(appName string) error {
func CommandBuildConfig(appName string, allApps bool, parallelCount int) error {
if allApps {
return common.RunCommandAgainstAllApps(BuildConfig, "build-config", parallelCount)
}
if err := common.VerifyAppName(appName); err != nil {
return err
}
return common.PlugnTrigger("proxy-build-config", []string{appName}...)
return BuildConfig(appName)
}
// CommandDisable disables the proxy for app via command line
func CommandDisable(appName string) error {
func CommandDisable(appName string, allApps bool, parallelCount int) error {
if allApps {
return common.RunCommandAgainstAllApps(Disable, "disable", parallelCount)
}
if err := common.VerifyAppName(appName); err != nil {
return err
}
if !IsAppProxyEnabled(appName) {
common.LogInfo1("Proxy is already disable for app")
return nil
}
common.LogInfo1("Disabling proxy for app")
entries := map[string]string{
"DOKKU_DISABLE_PROXY": "1",
}
if err := config.SetMany(appName, entries, false); err != nil {
return err
}
return common.PlugnTrigger("proxy-disable", []string{appName}...)
return Disable(appName)
}
// CommandEnable enables the proxy for app via command line
func CommandEnable(appName string) error {
func CommandEnable(appName string, allApps bool, parallelCount int) error {
if allApps {
return common.RunCommandAgainstAllApps(Enable, "enable", parallelCount)
}
if err := common.VerifyAppName(appName); err != nil {
return err
}
if IsAppProxyEnabled(appName) {
common.LogInfo1("Proxy is already enabled for app")
return nil
}
common.LogInfo1("Enabling proxy for app")
keys := []string{"DOKKU_DISABLE_PROXY"}
if err := config.UnsetMany(appName, keys, false); err != nil {
return err
}
return common.PlugnTrigger("proxy-enable", []string{appName}...)
return Enable(appName)
}
// CommandPorts is a cmd wrapper to list proxy port mappings for an app