extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-01-22 22:09:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

zrok test loop -> zrok test loop public (#237)

Browse Source
This commit is contained in:
Michael Quigley 2023-02-16 14:50:07 -05:00
parent 21a019ae4f
commit bca3133de5
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
2 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/zrok/main.go
View File

@ -19,6 +19,7 @@ func init() {
adminCmd.AddCommand(adminDeleteCmd)
adminCmd.AddCommand(adminListCmd)
adminCmd.AddCommand(adminUpdateCmd)
testCmd.AddCommand(loopCmd)
rootCmd.AddCommand(adminCmd)
rootCmd.AddCommand(configCmd)
rootCmd.AddCommand(shareCmd)
@ -72,6 +73,12 @@ var configCmd = &cobra.Command{
Short: "Configure your zrok environment",
}
var loopCmd = &cobra.Command{
Use: "loopback",
Aliases: []string{"loop"},
Short: "Loopback testing utilities",
}
var shareCmd = &cobra.Command{
Use: "share",
Short: "Create backend access for shares",

22
cmd/zrok/loop.go → cmd/zrok/testLoopPublic.go
View File

@ -28,10 +28,10 @@ import (
)
func init() {
testCmd.AddCommand(newLoopCmd().cmd)
loopCmd.AddCommand(newTestLoopPublicCommand().cmd)
}
type loopCmd struct {
type testLoopPublicCommand struct {
cmd *cobra.Command
loopers int
iterations int
@ -46,13 +46,13 @@ type loopCmd struct {
frontendSelection []string
}
func newLoopCmd() *loopCmd {
func newTestLoopPublicCommand() *testLoopPublicCommand {
cmd := &cobra.Command{
Use: "loop",
Short: "Start a loop agent",
Use: "public",
Short: "Start a loop agent testing public proxy shares",
Args: cobra.ExactArgs(0),
}
r := &loopCmd{cmd: cmd}
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")
@ -68,10 +68,10 @@ func newLoopCmd() *loopCmd {
return r
}
func (r *loopCmd) run(_ *cobra.Command, _ []string) {
func (cmd *testLoopPublicCommand) run(_ *cobra.Command, _ []string) {
var loopers []*looper
for i := 0; i < r.loopers; i++ {
l := newLooper(i, r)
for i := 0; i < cmd.loopers; i++ {
l := newLooper(i, cmd)
loopers = append(loopers, l)
go l.run()
}
@ -105,7 +105,7 @@ func (r *loopCmd) run(_ *cobra.Command, _ []string) {
type looper struct {
id int
cmd *loopCmd
cmd *testLoopPublicCommand
env *zrokdir.Environment
done chan struct{}
listener edge.Listener
@ -122,7 +122,7 @@ type looper struct {
stop bool
}
func newLooper(id int, cmd *loopCmd) *looper {
func newLooper(id int, cmd *testLoopPublicCommand) *looper {
return &looper{
id: id,
cmd: cmd,