extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-06-20 17:58:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

plumbing for the 'zrok test canary' commands (#771)

Browse Source
This commit is contained in:
Michael Quigley 2024-10-30 13:29:25 -04:00
parent 3cf98fc04f
commit 5bee2bb312
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
3 changed files with 59 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
cmd/zrok/main.go
View File

@ -24,7 +24,8 @@ func init() {
adminCmd.AddCommand(adminDeleteCmd)
adminCmd.AddCommand(adminListCmd)
adminCmd.AddCommand(adminUpdateCmd)
testCmd.AddCommand(loopCmd)
testCmd.AddCommand(testCanaryCmd)
testCmd.AddCommand(testLoopCmd)
rootCmd.AddCommand(adminCmd)
rootCmd.AddCommand(configCmd)
rootCmd.AddCommand(modifyCmd)
@ -82,12 +83,6 @@ var configCmd = &cobra.Command{
Short: "Configure your zrok environment",
}
var loopCmd = &cobra.Command{
Use: "loopback",
Aliases: []string{"loop"},
Short: "Loopback testing utilities",
}
var modifyCmd = &cobra.Command{
Use: "modify",
Aliases: []string{"mod"},
@ -101,7 +96,18 @@ var shareCmd = &cobra.Command{
var testCmd = &cobra.Command{
Use: "test",
Short: "Utilities for testing zrok deployments",
Short: "Utilities for testing deployments",
}
var testCanaryCmd = &cobra.Command{
Use: "canary",
Short: "Utilities for performance management",
}
var testLoopCmd = &cobra.Command{
Use: "loopback",
Aliases: []string{"loop"},
Short: "Loopback testing utilities",
}
func main() {

29
cmd/zrok/testCanaryPeriodic.go Normal file
View File

@ -0,0 +1,29 @@
package main
import (
"github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
func init() {
testCanaryCmd.AddCommand(newTestCanaryPeriodicCommand().cmd)
}
type testCanaryPeriodicCommand struct {
cmd *cobra.Command
}
func newTestCanaryPeriodicCommand() *testCanaryPeriodicCommand {
cmd := &cobra.Command{
Use: "periodic",
Short: "Run a periodic canary inspection",
Args: cobra.NoArgs,
}
command := &testCanaryPeriodicCommand{cmd: cmd}
cmd.Run = command.run
return command
}
func (c *testCanaryPeriodicCommand) run(_ *cobra.Command, _ []string) {
logrus.Info("periodic")
}

32
cmd/zrok/testLoopPublic.go
View File

@ -28,7 +28,7 @@ import (
)
func init() {
loopCmd.AddCommand(newTestLoopPublicCommand().cmd)
testLoopCmd.AddCommand(newTestLoopPublicCommand().cmd)
}
type testLoopPublicCommand struct {
@ -50,22 +50,22 @@ func newTestLoopPublicCommand() *testLoopPublicCommand {
cmd := &cobra.Command{
Use: "public",
Short: "Start a loop agent testing public proxy shares",
Args: cobra.ExactArgs(0),
Args: cobra.NoArgs,
}
r := &testLoopPublicCommand{cmd: cmd}
cmd.Run = r.run
cmd.Flags().IntVarP(&r.loopers, "loopers", "l", 1, "Number of current loopers to start")
cmd.Flags().IntVarP(&r.iterations, "iterations", "i", 1, "Number of iterations per looper")
cmd.Flags().IntVarP(&r.statusEvery, "status-every", "E", 100, "Show status every # iterations")
cmd.Flags().IntVarP(&r.timeoutSeconds, "timeout-seconds", "T", 30, "Time out after # seconds when sending http requests")
cmd.Flags().IntVar(&r.minPayload, "min-payload", 64, "Minimum payload size in bytes")
cmd.Flags().IntVar(&r.maxPayload, "max-payload", 10240, "Maximum payload size in bytes")
cmd.Flags().IntVar(&r.minDwellMs, "min-dwell-ms", 1000, "Minimum dwell time in milliseconds")
cmd.Flags().IntVar(&r.maxDwellMs, "max-dwell-ms", 1000, "Maximum dwell time in milliseconds")
cmd.Flags().IntVar(&r.minPacingMs, "min-pacing-ms", 0, "Minimum pacing in milliseconds")
cmd.Flags().IntVar(&r.maxPacingMs, "max-pacing-ms", 0, "Maximum pacing in milliseconds")
cmd.Flags().StringArrayVar(&r.frontendSelection, "frontends", []string{"public"}, "Selected frontends to use for the share")
return r
command := &testLoopPublicCommand{cmd: cmd}
cmd.Run = command.run
cmd.Flags().IntVarP(&command.loopers, "loopers", "l", 1, "Number of current loopers to start")
cmd.Flags().IntVarP(&command.iterations, "iterations", "i", 1, "Number of iterations per looper")
cmd.Flags().IntVarP(&command.statusEvery, "status-every", "E", 100, "Show status every # iterations")
cmd.Flags().IntVarP(&command.timeoutSeconds, "timeout-seconds", "T", 30, "Time out after # seconds when sending http requests")
cmd.Flags().IntVar(&command.minPayload, "min-payload", 64, "Minimum payload size in bytes")
cmd.Flags().IntVar(&command.maxPayload, "max-payload", 10240, "Maximum payload size in bytes")
cmd.Flags().IntVar(&command.minDwellMs, "min-dwell-ms", 1000, "Minimum dwell time in milliseconds")
cmd.Flags().IntVar(&command.maxDwellMs, "max-dwell-ms", 1000, "Maximum dwell time in milliseconds")
cmd.Flags().IntVar(&command.minPacingMs, "min-pacing-ms", 0, "Minimum pacing in milliseconds")
cmd.Flags().IntVar(&command.maxPacingMs, "max-pacing-ms", 0, "Maximum pacing in milliseconds")
cmd.Flags().StringArrayVar(&command.frontendSelection, "frontends", []string{"public"}, "Selected frontends to use for the share")
return command
}
func (cmd *testLoopPublicCommand) run(_ *cobra.Command, _ []string) {