mirror of
				https://github.com/aykhans/dodo.git
				synced 2025-11-04 06:29:58 +00:00 
			
		
		
		
	Compare commits
	
		
			36 Commits
		
	
	
		
			v0.5.2
			...
			ff09a3365e
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| ff09a3365e | |||
| c83246abe4 | |||
| efc4c62001 | |||
| 
						 | 
					f567774e3a | ||
| d2bd60e3ff | |||
| 0fe782c768 | |||
| b9dd14a588 | |||
| 18b1c7dae3 | |||
| 77ddcf722a | |||
| 048cb68898 | |||
| fd5e87ee68 | |||
| 61b9ce943b | |||
| 6fceb0094b | |||
| 186683fce5 | |||
| 04e5b5f3eb | |||
| 2bebf09adb | |||
| ebacebff16 | |||
| 575ca2913a | |||
| 4686e26ede | |||
| bd33b2c6a2 | |||
| 3068e0acae | |||
| 7a40bbbd3c | |||
| 
						 | 
					43c0bd2138 | ||
| fddeec1791 | |||
| 
						 | 
					6f5fa1860c | ||
| 4480aed0f0 | |||
| 
						 | 
					f5144ac4be | ||
| 4a1051e08c | |||
| 
						 | 
					2371986909 | ||
| e9af46084d | |||
| cdaa63152f | |||
| 16791421c7 | |||
| c90cfd8090 | |||
| 
						 | 
					1aaea8e437 | ||
| c44a0c978a | |||
| 
						 | 
					1ace7be2b4 | 
@@ -49,7 +49,7 @@ dodo -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
				
			|||||||
```
 | 
					```
 | 
				
			||||||
With Docker:
 | 
					With Docker:
 | 
				
			||||||
```sh
 | 
					```sh
 | 
				
			||||||
docker run --rm aykhans/dodo -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
					docker run --rm -i aykhans/dodo -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 2. JSON config file
 | 
					### 2. JSON config file
 | 
				
			||||||
@@ -65,7 +65,7 @@ You can find an example config structure in the [config.json](https://github.com
 | 
				
			|||||||
    "params": {},
 | 
					    "params": {},
 | 
				
			||||||
    "headers": {},
 | 
					    "headers": {},
 | 
				
			||||||
    "cookies": {},
 | 
					    "cookies": {},
 | 
				
			||||||
    "body": [""],
 | 
					    "body": [],
 | 
				
			||||||
    "proxies": [
 | 
					    "proxies": [
 | 
				
			||||||
        {
 | 
					        {
 | 
				
			||||||
            "url": "http://example.com:8080",
 | 
					            "url": "http://example.com:8080",
 | 
				
			||||||
@@ -85,7 +85,7 @@ dodo -c /path/config.json
 | 
				
			|||||||
```
 | 
					```
 | 
				
			||||||
With Docker:
 | 
					With Docker:
 | 
				
			||||||
```sh
 | 
					```sh
 | 
				
			||||||
docker run --rm -v ./path/config.json:/dodo/config.json -i aykhans/dodo
 | 
					docker run --rm -i -v ./path/config.json:/dodo/config.json aykhans/dodo
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 3. Both (CLI & JSON)
 | 
					### 3. Both (CLI & JSON)
 | 
				
			||||||
@@ -96,7 +96,7 @@ dodo -c /path/config.json -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
				
			|||||||
```
 | 
					```
 | 
				
			||||||
With Docker:
 | 
					With Docker:
 | 
				
			||||||
```sh
 | 
					```sh
 | 
				
			||||||
docker run --rm -v ./path/config.json:/dodo/config.json -i aykhans/dodo -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
					docker run --rm -i -v ./path/config.json:/dodo/config.json aykhans/dodo -u https://example.com -m GET -d 10 -r 1000 -t 2000
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## CLI and JSON Config Parameters
 | 
					## CLI and JSON Config Parameters
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,14 +1,14 @@
 | 
				
			|||||||
{
 | 
					{
 | 
				
			||||||
    "method": "GET",
 | 
					    "method": "GET",
 | 
				
			||||||
    "url": "https://example.com",
 | 
					    "url": "https://example.com",
 | 
				
			||||||
    "no_proxy_check": true,
 | 
					    "no_proxy_check": false,
 | 
				
			||||||
    "timeout": 10000,
 | 
					    "timeout": 10000,
 | 
				
			||||||
    "dodos_count": 50,
 | 
					    "dodos_count": 1,
 | 
				
			||||||
    "request_count": 1000,
 | 
					    "request_count": 1,
 | 
				
			||||||
    "params": {},
 | 
					    "params": {},
 | 
				
			||||||
    "headers": {},
 | 
					    "headers": {},
 | 
				
			||||||
    "cookies": {},
 | 
					    "cookies": {},
 | 
				
			||||||
    "body": [""],
 | 
					    "body": [],
 | 
				
			||||||
    "proxies": [
 | 
					    "proxies": [
 | 
				
			||||||
        {
 | 
					        {
 | 
				
			||||||
            "url": "http://example.com:8080",
 | 
					            "url": "http://example.com:8080",
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,23 +1,24 @@
 | 
				
			|||||||
package config
 | 
					package config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"net/url"
 | 
						"net/url"
 | 
				
			||||||
	"os"
 | 
						"os"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						. "github.com/aykhans/dodo/types"
 | 
				
			||||||
	"github.com/aykhans/dodo/utils"
 | 
						"github.com/aykhans/dodo/utils"
 | 
				
			||||||
	"github.com/jedib0t/go-pretty/v6/table"
 | 
						"github.com/jedib0t/go-pretty/v6/table"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
	VERSION                 string = "0.5.2"
 | 
						VERSION                 string = "0.5.4"
 | 
				
			||||||
	DefaultUserAgent        string = "Dodo/" + VERSION
 | 
						DefaultUserAgent        string = "Dodo/" + VERSION
 | 
				
			||||||
	ProxyCheckURL           string = "https://www.google.com"
 | 
						ProxyCheckURL           string = "https://www.google.com"
 | 
				
			||||||
	DefaultMethod           string = "GET"
 | 
						DefaultMethod           string = "GET"
 | 
				
			||||||
	DefaultTimeout          uint32 = 10000 // Milliseconds (10 seconds)
 | 
						DefaultTimeout          uint32 = 10000 // Milliseconds (10 seconds)
 | 
				
			||||||
	DefaultDodosCount       uint   = 1
 | 
						DefaultDodosCount       uint   = 1
 | 
				
			||||||
	DefaultRequestCount     uint   = 1000
 | 
						DefaultRequestCount     uint   = 1
 | 
				
			||||||
	MaxDodosCountForProxies uint   = 20 // Max dodos count for proxy check
 | 
						MaxDodosCountForProxies uint   = 20 // Max dodos count for proxy check
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -41,7 +42,18 @@ func (config *RequestConfig) Print() {
 | 
				
			|||||||
	t.SetOutputMirror(os.Stdout)
 | 
						t.SetOutputMirror(os.Stdout)
 | 
				
			||||||
	t.SetStyle(table.StyleLight)
 | 
						t.SetStyle(table.StyleLight)
 | 
				
			||||||
	t.SetColumnConfigs([]table.ColumnConfig{
 | 
						t.SetColumnConfigs([]table.ColumnConfig{
 | 
				
			||||||
		{Number: 2, WidthMax: 50},
 | 
							{
 | 
				
			||||||
 | 
								Number: 2,
 | 
				
			||||||
 | 
								WidthMaxEnforcer: func(col string, maxLen int) string {
 | 
				
			||||||
 | 
									lines := strings.Split(col, "\n")
 | 
				
			||||||
 | 
									for i, line := range lines {
 | 
				
			||||||
 | 
										if len(line) > maxLen {
 | 
				
			||||||
 | 
											lines[i] = line[:maxLen-3] + "..."
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return strings.Join(lines, "\n")
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								WidthMax: 50},
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	newHeaders := make(map[string][]string)
 | 
						newHeaders := make(map[string][]string)
 | 
				
			||||||
@@ -55,23 +67,23 @@ func (config *RequestConfig) Print() {
 | 
				
			|||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"URL", config.URL})
 | 
						t.AppendRow(table.Row{"URL", config.URL})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Timeout", fmt.Sprintf("%dms", config.Timeout/time.Millisecond)})
 | 
						t.AppendRow(table.Row{"Timeout", config.Timeout})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Dodos", config.DodosCount})
 | 
						t.AppendRow(table.Row{"Dodos", config.DodosCount})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Requests", config.RequestCount})
 | 
						t.AppendRow(table.Row{"Requests", config.RequestCount})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Params", utils.MarshalJSON(config.Params, 3)})
 | 
						t.AppendRow(table.Row{"Params", string(utils.PrettyJSONMarshal(config.Params, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Headers", utils.MarshalJSON(newHeaders, 3)})
 | 
						t.AppendRow(table.Row{"Headers", string(utils.PrettyJSONMarshal(newHeaders, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Cookies", utils.MarshalJSON(config.Cookies, 3)})
 | 
						t.AppendRow(table.Row{"Cookies", string(utils.PrettyJSONMarshal(config.Cookies, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Proxies Count", len(config.Proxies)})
 | 
						t.AppendRow(table.Row{"Proxies Count", string(utils.PrettyJSONMarshal(config.Proxies, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Proxy Check", !config.NoProxyCheck})
 | 
						t.AppendRow(table.Row{"Proxy Check", !config.NoProxyCheck})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Body", utils.MarshalJSON(config.Body, 3)})
 | 
						t.AppendRow(table.Row{"Body", string(utils.PrettyJSONMarshal(config.Body, 3, "", "  "))})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	t.Render()
 | 
						t.Render()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -97,7 +109,7 @@ type Config struct {
 | 
				
			|||||||
	Timeout      uint32       `json:"timeout" validate:"gte=1,lte=100000"`
 | 
						Timeout      uint32       `json:"timeout" validate:"gte=1,lte=100000"`
 | 
				
			||||||
	DodosCount   uint         `json:"dodos_count" validate:"gte=1"`
 | 
						DodosCount   uint         `json:"dodos_count" validate:"gte=1"`
 | 
				
			||||||
	RequestCount uint         `json:"request_count" validation_name:"request-count" validate:"gte=1"`
 | 
						RequestCount uint         `json:"request_count" validation_name:"request-count" validate:"gte=1"`
 | 
				
			||||||
	NoProxyCheck utils.Option[bool] `json:"no_proxy_check"`
 | 
						NoProxyCheck Option[bool] `json:"no_proxy_check"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func NewConfig(
 | 
					func NewConfig(
 | 
				
			||||||
@@ -105,10 +117,10 @@ func NewConfig(
 | 
				
			|||||||
	timeout uint32,
 | 
						timeout uint32,
 | 
				
			||||||
	dodosCount uint,
 | 
						dodosCount uint,
 | 
				
			||||||
	requestCount uint,
 | 
						requestCount uint,
 | 
				
			||||||
	noProxyCheck utils.Option[bool],
 | 
						noProxyCheck Option[bool],
 | 
				
			||||||
) *Config {
 | 
					) *Config {
 | 
				
			||||||
	if noProxyCheck == nil {
 | 
						if noProxyCheck == nil {
 | 
				
			||||||
		noProxyCheck = utils.NewNoneOption[bool]()
 | 
							noProxyCheck = NewNoneOption[bool]()
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return &Config{
 | 
						return &Config{
 | 
				
			||||||
@@ -155,7 +167,7 @@ func (config *Config) SetDefaults() {
 | 
				
			|||||||
		config.RequestCount = DefaultRequestCount
 | 
							config.RequestCount = DefaultRequestCount
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if config.NoProxyCheck.IsNone() {
 | 
						if config.NoProxyCheck.IsNone() {
 | 
				
			||||||
		config.NoProxyCheck = utils.NewOption(false)
 | 
							config.NoProxyCheck = NewOption(false)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -208,13 +220,13 @@ func (config *JSONConfig) MergeConfigs(newConfig *JSONConfig) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
type CLIConfig struct {
 | 
					type CLIConfig struct {
 | 
				
			||||||
	*Config
 | 
						*Config
 | 
				
			||||||
	Yes        bool   `json:"yes" validate:"omitempty"`
 | 
						Yes        Option[bool] `json:"yes" validate:"omitempty"`
 | 
				
			||||||
	ConfigFile string       `validation_name:"config-file" validate:"omitempty,filepath"`
 | 
						ConfigFile string       `validation_name:"config-file" validate:"omitempty,filepath"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func NewCLIConfig(
 | 
					func NewCLIConfig(
 | 
				
			||||||
	config *Config,
 | 
						config *Config,
 | 
				
			||||||
	yes bool,
 | 
						yes Option[bool],
 | 
				
			||||||
	configFile string,
 | 
						configFile string,
 | 
				
			||||||
) *CLIConfig {
 | 
					) *CLIConfig {
 | 
				
			||||||
	return &CLIConfig{
 | 
						return &CLIConfig{
 | 
				
			||||||
@@ -227,4 +239,7 @@ func (config *CLIConfig) MergeConfigs(newConfig *CLIConfig) {
 | 
				
			|||||||
	if newConfig.ConfigFile != "" {
 | 
						if newConfig.ConfigFile != "" {
 | 
				
			||||||
		config.ConfigFile = newConfig.ConfigFile
 | 
							config.ConfigFile = newConfig.ConfigFile
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						if !newConfig.Yes.IsNone() {
 | 
				
			||||||
 | 
							config.Yes = newConfig.Yes
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -19,10 +19,6 @@ func OSErrorFormater(err error) error {
 | 
				
			|||||||
	return ErrInvalidFile
 | 
						return ErrInvalidFile
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func CobraErrorFormater(err error) error {
 | 
					 | 
				
			||||||
	return err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func shortenNamespace(namespace string) string {
 | 
					func shortenNamespace(namespace string) string {
 | 
				
			||||||
	return namespace[strings.Index(namespace, ".")+1:]
 | 
						return namespace[strings.Index(namespace, ".")+1:]
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										17
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										17
									
								
								go.mod
									
									
									
									
									
								
							@@ -4,11 +4,9 @@ go 1.23.2
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
	github.com/go-playground/validator/v10 v10.23.0
 | 
						github.com/go-playground/validator/v10 v10.23.0
 | 
				
			||||||
	github.com/jedib0t/go-pretty/v6 v6.6.2
 | 
						github.com/jedib0t/go-pretty/v6 v6.6.5
 | 
				
			||||||
	github.com/spf13/cobra v1.8.1
 | 
						github.com/valyala/fasthttp v1.58.0
 | 
				
			||||||
	github.com/spf13/pflag v1.0.5
 | 
						golang.org/x/net v0.33.0
 | 
				
			||||||
	github.com/valyala/fasthttp v1.57.0
 | 
					 | 
				
			||||||
	golang.org/x/net v0.31.0
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
@@ -16,14 +14,13 @@ require (
 | 
				
			|||||||
	github.com/gabriel-vasile/mimetype v1.4.4 // indirect
 | 
						github.com/gabriel-vasile/mimetype v1.4.4 // indirect
 | 
				
			||||||
	github.com/go-playground/locales v0.14.1 // indirect
 | 
						github.com/go-playground/locales v0.14.1 // indirect
 | 
				
			||||||
	github.com/go-playground/universal-translator v0.18.1 // indirect
 | 
						github.com/go-playground/universal-translator v0.18.1 // indirect
 | 
				
			||||||
	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 | 
					 | 
				
			||||||
	github.com/klauspost/compress v1.17.11 // indirect
 | 
						github.com/klauspost/compress v1.17.11 // indirect
 | 
				
			||||||
	github.com/leodido/go-urn v1.4.0 // indirect
 | 
						github.com/leodido/go-urn v1.4.0 // indirect
 | 
				
			||||||
	github.com/mattn/go-runewidth v0.0.15 // indirect
 | 
						github.com/mattn/go-runewidth v0.0.15 // indirect
 | 
				
			||||||
	github.com/rivo/uniseg v0.4.7 // indirect
 | 
						github.com/rivo/uniseg v0.4.7 // indirect
 | 
				
			||||||
	github.com/valyala/bytebufferpool v1.0.0 // indirect
 | 
						github.com/valyala/bytebufferpool v1.0.0 // indirect
 | 
				
			||||||
	golang.org/x/crypto v0.29.0 // indirect
 | 
						golang.org/x/crypto v0.31.0 // indirect
 | 
				
			||||||
	golang.org/x/sys v0.27.0 // indirect
 | 
						golang.org/x/sys v0.28.0 // indirect
 | 
				
			||||||
	golang.org/x/term v0.26.0 // indirect
 | 
						golang.org/x/term v0.27.0 // indirect
 | 
				
			||||||
	golang.org/x/text v0.20.0 // indirect
 | 
						golang.org/x/text v0.21.0 // indirect
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										37
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										37
									
								
								go.sum
									
									
									
									
									
								
							@@ -1,6 +1,5 @@
 | 
				
			|||||||
github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
 | 
					github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
 | 
				
			||||||
github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
 | 
					github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
 | 
				
			||||||
github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
 | 
					 | 
				
			||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 | 
					github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 | 
				
			||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 | 
					github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 | 
				
			||||||
github.com/gabriel-vasile/mimetype v1.4.4 h1:QjV6pZ7/XZ7ryI2KuyeEDE8wnh7fHP9YnQy+R0LnH8I=
 | 
					github.com/gabriel-vasile/mimetype v1.4.4 h1:QjV6pZ7/XZ7ryI2KuyeEDE8wnh7fHP9YnQy+R0LnH8I=
 | 
				
			||||||
@@ -13,10 +12,8 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
 | 
				
			|||||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 | 
					github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 | 
				
			||||||
github.com/go-playground/validator/v10 v10.23.0 h1:/PwmTwZhS0dPkav3cdK9kV1FsAmrL8sThn8IHr/sO+o=
 | 
					github.com/go-playground/validator/v10 v10.23.0 h1:/PwmTwZhS0dPkav3cdK9kV1FsAmrL8sThn8IHr/sO+o=
 | 
				
			||||||
github.com/go-playground/validator/v10 v10.23.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 | 
					github.com/go-playground/validator/v10 v10.23.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 | 
				
			||||||
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
 | 
					github.com/jedib0t/go-pretty/v6 v6.6.5 h1:9PgMJOVBedpgYLI56jQRJYqngxYAAzfEUua+3NgSqAo=
 | 
				
			||||||
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 | 
					github.com/jedib0t/go-pretty/v6 v6.6.5/go.mod h1:Uq/HrbhuFty5WSVNfjpQQe47x16RwVGXIveNGEyGtHs=
 | 
				
			||||||
github.com/jedib0t/go-pretty/v6 v6.6.2 h1:27bLj3nRODzaiA7tPIxy9UVWHoPspFfME9XxgwiiNsM=
 | 
					 | 
				
			||||||
github.com/jedib0t/go-pretty/v6 v6.6.2/go.mod h1:zbn98qrYlh95FIhwwsbIip0LYpwSG8SUOScs+v9/t0E=
 | 
					 | 
				
			||||||
github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc=
 | 
					github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc=
 | 
				
			||||||
github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
 | 
					github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
 | 
				
			||||||
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
 | 
					github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
 | 
				
			||||||
@@ -28,29 +25,23 @@ github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZN
 | 
				
			|||||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 | 
					github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 | 
				
			||||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
 | 
					github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
 | 
				
			||||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 | 
					github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 | 
				
			||||||
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 | 
					 | 
				
			||||||
github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
 | 
					 | 
				
			||||||
github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
 | 
					 | 
				
			||||||
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
 | 
					 | 
				
			||||||
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 | 
					 | 
				
			||||||
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
 | 
					github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
 | 
				
			||||||
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 | 
					github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 | 
				
			||||||
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 | 
					github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 | 
				
			||||||
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
 | 
					github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
 | 
				
			||||||
github.com/valyala/fasthttp v1.57.0 h1:Xw8SjWGEP/+wAAgyy5XTvgrWlOD1+TxbbvNADYCm1Tg=
 | 
					github.com/valyala/fasthttp v1.58.0 h1:GGB2dWxSbEprU9j0iMJHgdKYJVDyjrOwF9RE59PbRuE=
 | 
				
			||||||
github.com/valyala/fasthttp v1.57.0/go.mod h1:h6ZBaPRlzpZ6O3H5t2gEk1Qi33+TmLvfwgLLp0t9CpE=
 | 
					github.com/valyala/fasthttp v1.58.0/go.mod h1:SYXvHHaFp7QZHGKSHmoMipInhrI5StHrhDTYVEjK/Kw=
 | 
				
			||||||
github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
 | 
					github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
 | 
				
			||||||
github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
 | 
					github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
 | 
				
			||||||
golang.org/x/crypto v0.29.0 h1:L5SG1JTTXupVV3n6sUqMTeWbjAyfPwoda2DLX8J8FrQ=
 | 
					golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
 | 
				
			||||||
golang.org/x/crypto v0.29.0/go.mod h1:+F4F4N5hv6v38hfeYwTdx20oUvLLc+QfrE9Ax9HtgRg=
 | 
					golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
 | 
				
			||||||
golang.org/x/net v0.31.0 h1:68CPQngjLL0r2AlUKiSxtQFKvzRVbnzLwMUn5SzcLHo=
 | 
					golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
 | 
				
			||||||
golang.org/x/net v0.31.0/go.mod h1:P4fl1q7dY2hnZFxEk4pPSkDHF+QqjitcnDjUQyMM+pM=
 | 
					golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
 | 
				
			||||||
golang.org/x/sys v0.27.0 h1:wBqf8DvsY9Y/2P8gAfPDEYNuS30J4lPHJxXSb/nJZ+s=
 | 
					golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
 | 
				
			||||||
golang.org/x/sys v0.27.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
					golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
				
			||||||
golang.org/x/term v0.26.0 h1:WEQa6V3Gja/BhNxg540hBip/kkaYtRg3cxg4oXSw4AU=
 | 
					golang.org/x/term v0.27.0 h1:WP60Sv1nlK1T6SupCHbXzSaN0b9wUmsPoRS9b61A23Q=
 | 
				
			||||||
golang.org/x/term v0.26.0/go.mod h1:Si5m1o57C5nBNQo5z1iq+XDijt21BDBDp2bK0QI8e3E=
 | 
					golang.org/x/term v0.27.0/go.mod h1:iMsnZpn0cago0GOrHO2+Y7u7JPn5AylBrcoWkElMTSM=
 | 
				
			||||||
golang.org/x/text v0.20.0 h1:gK/Kv2otX8gz+wn7Rmb3vT96ZwuoxnQlY+HlJVj7Qug=
 | 
					golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
 | 
				
			||||||
golang.org/x/text v0.20.0/go.mod h1:D4IsuqiFMhST5bX19pQ9ikHC2GsaKyk/oF+pn3ducp4=
 | 
					golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
 | 
				
			||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 | 
					 | 
				
			||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 | 
					gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 | 
				
			||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 | 
					gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										12
									
								
								main.go
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								main.go
									
									
									
									
									
								
							@@ -27,9 +27,13 @@ func main() {
 | 
				
			|||||||
	)
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	cliConf, err := readers.CLIConfigReader()
 | 
						cliConf, err := readers.CLIConfigReader()
 | 
				
			||||||
	if err != nil || cliConf == nil {
 | 
						if err != nil {
 | 
				
			||||||
 | 
							utils.PrintAndExit(err.Error())
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if cliConf == nil {
 | 
				
			||||||
		os.Exit(0)
 | 
							os.Exit(0)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if err := validator.StructPartial(cliConf, "ConfigFile"); err != nil {
 | 
						if err := validator.StructPartial(cliConf, "ConfigFile"); err != nil {
 | 
				
			||||||
		utils.PrintErrAndExit(
 | 
							utils.PrintErrAndExit(
 | 
				
			||||||
			customerrors.ValidationErrorsFormater(
 | 
								customerrors.ValidationErrorsFormater(
 | 
				
			||||||
@@ -82,11 +86,11 @@ func main() {
 | 
				
			|||||||
		Cookies:      jsonConf.Cookies,
 | 
							Cookies:      jsonConf.Cookies,
 | 
				
			||||||
		Proxies:      jsonConf.Proxies,
 | 
							Proxies:      jsonConf.Proxies,
 | 
				
			||||||
		Body:         jsonConf.Body,
 | 
							Body:         jsonConf.Body,
 | 
				
			||||||
		Yes:          cliConf.Yes,
 | 
							Yes:          cliConf.Yes.ValueOr(false),
 | 
				
			||||||
		NoProxyCheck: conf.NoProxyCheck.ValueOrPanic(),
 | 
							NoProxyCheck: conf.NoProxyCheck.ValueOr(false),
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	requestConf.Print()
 | 
						requestConf.Print()
 | 
				
			||||||
	if !cliConf.Yes {
 | 
						if !cliConf.Yes.ValueOr(false) {
 | 
				
			||||||
		response := readers.CLIYesOrNoReader("Do you want to continue?", true)
 | 
							response := readers.CLIYesOrNoReader("Do you want to continue?", true)
 | 
				
			||||||
		if !response {
 | 
							if !response {
 | 
				
			||||||
			utils.PrintAndExit("Exiting...")
 | 
								utils.PrintAndExit("Exiting...")
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										149
									
								
								readers/cli.go
									
									
									
									
									
								
							
							
						
						
									
										149
									
								
								readers/cli.go
									
									
									
									
									
								
							@@ -1,69 +1,128 @@
 | 
				
			|||||||
package readers
 | 
					package readers
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
 | 
						"flag"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/aykhans/dodo/config"
 | 
						"github.com/aykhans/dodo/config"
 | 
				
			||||||
	"github.com/aykhans/dodo/custom_errors"
 | 
						. "github.com/aykhans/dodo/types"
 | 
				
			||||||
	"github.com/aykhans/dodo/utils"
 | 
						"github.com/aykhans/dodo/utils"
 | 
				
			||||||
	"github.com/spf13/cobra"
 | 
					 | 
				
			||||||
	"github.com/spf13/pflag"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func CLIConfigReader() (*config.CLIConfig, error) {
 | 
					const usageText = `Usage:
 | 
				
			||||||
	var (
 | 
					  dodo [flags]
 | 
				
			||||||
		returnNil    = false
 | 
					
 | 
				
			||||||
		cliConfig    = config.NewCLIConfig(config.NewConfig("", 0, 0, 0, nil), false, "")
 | 
					Examples:
 | 
				
			||||||
		dodosCount   uint
 | 
					
 | 
				
			||||||
		requestCount uint
 | 
					Simple usage only with URL:
 | 
				
			||||||
		timeout      uint32
 | 
					 | 
				
			||||||
		noProxyCheck bool
 | 
					 | 
				
			||||||
		rootCmd      = &cobra.Command{
 | 
					 | 
				
			||||||
			Use: "dodo [flags]",
 | 
					 | 
				
			||||||
			Example: `  Simple usage only with URL:
 | 
					 | 
				
			||||||
  dodo -u https://example.com
 | 
					  dodo -u https://example.com
 | 
				
			||||||
 | 
					
 | 
				
			||||||
  Simple usage with config file:
 | 
					Simple usage with config file:
 | 
				
			||||||
  dodo -c /path/to/config/file/config.json
 | 
					  dodo -c /path/to/config/file/config.json
 | 
				
			||||||
 | 
					
 | 
				
			||||||
  Usage with all flags:
 | 
					Usage with all flags:
 | 
				
			||||||
    dodo -c /path/to/config/file/config.json -u https://example.com -m POST -d 10 -r 1000 -t 2000 --no-proxy-check -y`,
 | 
					  dodo -c /path/to/config/file/config.json -u https://example.com -m POST -d 10 -r 1000 -t 2000 --no-proxy-check -y
 | 
				
			||||||
			Run:           func(cmd *cobra.Command, args []string) {},
 | 
					 | 
				
			||||||
			SilenceErrors: true,
 | 
					 | 
				
			||||||
			SilenceUsage:  true,
 | 
					 | 
				
			||||||
			Version:       config.VERSION,
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	rootCmd.Flags().StringVarP(&cliConfig.ConfigFile, "config-file", "c", "", "Path to the config file")
 | 
					Flags:
 | 
				
			||||||
	rootCmd.Flags().BoolVarP(&cliConfig.Yes, "yes", "y", false, "Answer yes to all questions")
 | 
					  -h, --help                 help for dodo
 | 
				
			||||||
	rootCmd.Flags().StringVarP(&cliConfig.Method, "method", "m", "", fmt.Sprintf("HTTP Method (default %s)", config.DefaultMethod))
 | 
					  -v, --version              version for dodo
 | 
				
			||||||
	rootCmd.Flags().StringVarP(&cliConfig.URL, "url", "u", "", "URL for stress testing")
 | 
					  -c, --config-file string   Path to the config file
 | 
				
			||||||
	rootCmd.Flags().UintVarP(&dodosCount, "dodos-count", "d", config.DefaultDodosCount, "Number of dodos(threads)")
 | 
					  -d, --dodos-count uint     Number of dodos(threads) (default %d)
 | 
				
			||||||
	rootCmd.Flags().UintVarP(&requestCount, "request-count", "r", config.DefaultRequestCount, "Number of total requests")
 | 
					  -m, --method string        HTTP Method (default %s)
 | 
				
			||||||
	rootCmd.Flags().Uint32VarP(&timeout, "timeout", "t", config.DefaultTimeout, "Timeout for each request in milliseconds")
 | 
					  -r, --request-count uint   Number of total requests (default %d)
 | 
				
			||||||
	rootCmd.Flags().BoolVar(&noProxyCheck, "no-proxy-check", false, "Do not check for proxies")
 | 
					  -t, --timeout uint32       Timeout for each request in milliseconds (default %d)
 | 
				
			||||||
	if err := rootCmd.Execute(); err != nil {
 | 
					  -u, --url string           URL for stress testing
 | 
				
			||||||
		utils.PrintErr(err)
 | 
					      --no-proxy-check bool  Do not check for proxies (default false)
 | 
				
			||||||
		rootCmd.Println(rootCmd.UsageString())
 | 
					  -y, --yes bool             Answer yes to all questions (default false)`
 | 
				
			||||||
		return nil, customerrors.CobraErrorFormater(err)
 | 
					
 | 
				
			||||||
 | 
					func CLIConfigReader() (*config.CLIConfig, error) {
 | 
				
			||||||
 | 
						flag.Usage = func() {
 | 
				
			||||||
 | 
							fmt.Printf(
 | 
				
			||||||
 | 
								usageText+"\n",
 | 
				
			||||||
 | 
								config.DefaultDodosCount,
 | 
				
			||||||
 | 
								config.DefaultMethod,
 | 
				
			||||||
 | 
								config.DefaultRequestCount,
 | 
				
			||||||
 | 
								config.DefaultTimeout,
 | 
				
			||||||
 | 
							)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	rootCmd.Flags().Visit(func(f *pflag.Flag) {
 | 
					
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							cliConfig          = config.NewCLIConfig(config.NewConfig("", 0, 0, 0, nil), NewOption(false), "")
 | 
				
			||||||
 | 
							configFile         = ""
 | 
				
			||||||
 | 
							yes                = false
 | 
				
			||||||
 | 
							method             = ""
 | 
				
			||||||
 | 
							url                = ""
 | 
				
			||||||
 | 
							dodosCount    uint = 0
 | 
				
			||||||
 | 
							requestsCount uint = 0
 | 
				
			||||||
 | 
							timeout       uint = 0
 | 
				
			||||||
 | 
							noProxyCheck  bool = false
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
						{
 | 
				
			||||||
 | 
							flag.Bool("version", false, "Prints the version of the program")
 | 
				
			||||||
 | 
							flag.Bool("v", false, "Prints the version of the program")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.StringVar(&configFile, "config-file", "", "Path to the configuration file")
 | 
				
			||||||
 | 
							flag.StringVar(&configFile, "c", "", "Path to the configuration file")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.BoolVar(&yes, "yes", false, "Answer yes to all questions")
 | 
				
			||||||
 | 
							flag.BoolVar(&yes, "y", false, "Answer yes to all questions")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.StringVar(&method, "method", "", "HTTP Method")
 | 
				
			||||||
 | 
							flag.StringVar(&method, "m", "", "HTTP Method")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.StringVar(&url, "url", "", "URL to send the request")
 | 
				
			||||||
 | 
							flag.StringVar(&url, "u", "", "URL to send the request")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.UintVar(&dodosCount, "dodos-count", 0, "Number of dodos(threads)")
 | 
				
			||||||
 | 
							flag.UintVar(&dodosCount, "d", 0, "Number of dodos(threads)")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.UintVar(&requestsCount, "requests-count", 0, "Number of total requests")
 | 
				
			||||||
 | 
							flag.UintVar(&requestsCount, "r", 0, "Number of total requests")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.UintVar(&timeout, "timeout", 0, "Timeout for each request in milliseconds")
 | 
				
			||||||
 | 
							flag.UintVar(&timeout, "t", 0, "Timeout for each request in milliseconds")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							flag.BoolVar(&noProxyCheck, "no-proxy-check", false, "Do not check for active proxies")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						flag.Parse()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						args := flag.Args()
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("unexpected arguments: %v", strings.Join(args, ", "))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						returnNil := false
 | 
				
			||||||
 | 
						flag.Visit(func(f *flag.Flag) {
 | 
				
			||||||
		switch f.Name {
 | 
							switch f.Name {
 | 
				
			||||||
		case "help":
 | 
							case "version", "v":
 | 
				
			||||||
 | 
								fmt.Printf("dodo version %s\n", config.VERSION)
 | 
				
			||||||
			returnNil = true
 | 
								returnNil = true
 | 
				
			||||||
		case "version":
 | 
							case "config-file", "c":
 | 
				
			||||||
			returnNil = true
 | 
								cliConfig.ConfigFile = configFile
 | 
				
			||||||
		case "dodos-count":
 | 
							case "yes", "y":
 | 
				
			||||||
 | 
								cliConfig.Yes.SetValue(yes)
 | 
				
			||||||
 | 
							case "method", "m":
 | 
				
			||||||
 | 
								cliConfig.Method = method
 | 
				
			||||||
 | 
							case "url", "u":
 | 
				
			||||||
 | 
								cliConfig.URL = url
 | 
				
			||||||
 | 
							case "dodos-count", "d":
 | 
				
			||||||
			cliConfig.DodosCount = dodosCount
 | 
								cliConfig.DodosCount = dodosCount
 | 
				
			||||||
		case "request-count":
 | 
							case "requests-count", "r":
 | 
				
			||||||
			cliConfig.RequestCount = requestCount
 | 
								cliConfig.RequestCount = requestsCount
 | 
				
			||||||
		case "timeout":
 | 
							case "timeout", "t":
 | 
				
			||||||
			cliConfig.Timeout = timeout
 | 
								var maxUint32 uint = 4294967295
 | 
				
			||||||
 | 
								if timeout > maxUint32 {
 | 
				
			||||||
 | 
									utils.PrintfC(utils.Colors.Yellow, "timeout value is too large, setting to %d\n", maxUint32)
 | 
				
			||||||
 | 
									timeout = maxUint32
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								cliConfig.Timeout = uint32(timeout)
 | 
				
			||||||
		case "no-proxy-check":
 | 
							case "no-proxy-check":
 | 
				
			||||||
			cliConfig.NoProxyCheck = utils.NewOption(noProxyCheck)
 | 
								cliConfig.NoProxyCheck.SetValue(noProxyCheck)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if returnNil {
 | 
						if returnNil {
 | 
				
			||||||
		return nil, nil
 | 
							return nil, nil
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,85 +1,108 @@
 | 
				
			|||||||
package requests
 | 
					package requests
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"os"
 | 
						"os"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						. "github.com/aykhans/dodo/types"
 | 
				
			||||||
	"github.com/aykhans/dodo/utils"
 | 
						"github.com/aykhans/dodo/utils"
 | 
				
			||||||
	"github.com/jedib0t/go-pretty/v6/table"
 | 
						"github.com/jedib0t/go-pretty/v6/table"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Response struct {
 | 
					type Response struct {
 | 
				
			||||||
	StatusCode int
 | 
						Response string
 | 
				
			||||||
	Error      error
 | 
					 | 
				
			||||||
	Time     time.Duration
 | 
						Time     time.Duration
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Responses []*Response
 | 
					type Responses []*Response
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Print prints the responses in a tabular format, including information such as
 | 
					// Print prints the responses in a tabular format, including information such as
 | 
				
			||||||
// response count, minimum time, maximum time, and average time.
 | 
					// response count, minimum time, maximum time, average time, and latency percentiles.
 | 
				
			||||||
func (respones Responses) Print() {
 | 
					func (responses Responses) Print() {
 | 
				
			||||||
	var (
 | 
						total := struct {
 | 
				
			||||||
		totalMinDuration time.Duration = respones[0].Time
 | 
							Count int
 | 
				
			||||||
		totalMaxDuration time.Duration = respones[0].Time
 | 
							Min   time.Duration
 | 
				
			||||||
		totalDuration    time.Duration
 | 
							Max   time.Duration
 | 
				
			||||||
		totalCount       int = len(respones)
 | 
							Sum   time.Duration
 | 
				
			||||||
	)
 | 
							P90   time.Duration
 | 
				
			||||||
	mergedResponses := make(map[string][]time.Duration)
 | 
							P95   time.Duration
 | 
				
			||||||
 | 
							P99   time.Duration
 | 
				
			||||||
	for _, response := range respones {
 | 
						}{
 | 
				
			||||||
		if response.Time < totalMinDuration {
 | 
							Count: len(responses),
 | 
				
			||||||
			totalMinDuration = response.Time
 | 
							Min:   responses[0].Time,
 | 
				
			||||||
 | 
							Max:   responses[0].Time,
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
		if response.Time > totalMaxDuration {
 | 
						mergedResponses := make(map[string]Durations)
 | 
				
			||||||
			totalMaxDuration = response.Time
 | 
						var allDurations Durations
 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		totalDuration += response.Time
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if response.Error != nil {
 | 
						for _, response := range responses {
 | 
				
			||||||
			mergedResponses[response.Error.Error()] = append(
 | 
							if response.Time < total.Min {
 | 
				
			||||||
				mergedResponses[response.Error.Error()],
 | 
								total.Min = response.Time
 | 
				
			||||||
				response.Time,
 | 
							}
 | 
				
			||||||
			)
 | 
							if response.Time > total.Max {
 | 
				
			||||||
		} else {
 | 
								total.Max = response.Time
 | 
				
			||||||
			mergedResponses[fmt.Sprintf("%d", response.StatusCode)] = append(
 | 
							}
 | 
				
			||||||
				mergedResponses[fmt.Sprintf("%d", response.StatusCode)],
 | 
							total.Sum += response.Time
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							mergedResponses[response.Response] = append(
 | 
				
			||||||
 | 
								mergedResponses[response.Response],
 | 
				
			||||||
			response.Time,
 | 
								response.Time,
 | 
				
			||||||
		)
 | 
							)
 | 
				
			||||||
 | 
							allDurations = append(allDurations, response.Time)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	}
 | 
						allDurations.Sort()
 | 
				
			||||||
 | 
						allDurationsLenAsFloat := float64(len(allDurations) - 1)
 | 
				
			||||||
 | 
						total.P90 = allDurations[int(0.90*allDurationsLenAsFloat)]
 | 
				
			||||||
 | 
						total.P95 = allDurations[int(0.95*allDurationsLenAsFloat)]
 | 
				
			||||||
 | 
						total.P99 = allDurations[int(0.99*allDurationsLenAsFloat)]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	t := table.NewWriter()
 | 
						t := table.NewWriter()
 | 
				
			||||||
	t.SetOutputMirror(os.Stdout)
 | 
						t.SetOutputMirror(os.Stdout)
 | 
				
			||||||
	t.SetStyle(table.StyleLight)
 | 
						t.SetStyle(table.StyleLight)
 | 
				
			||||||
	t.SetColumnConfigs([]table.ColumnConfig{
 | 
						t.SetColumnConfigs([]table.ColumnConfig{
 | 
				
			||||||
		{Number: 1, WidthMax: 80},
 | 
							{Number: 1, WidthMax: 40},
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					 | 
				
			||||||
	t.AppendHeader(table.Row{
 | 
						t.AppendHeader(table.Row{
 | 
				
			||||||
		"Response",
 | 
							"Response",
 | 
				
			||||||
		"Count",
 | 
							"Count",
 | 
				
			||||||
		"Min Time",
 | 
							"Min",
 | 
				
			||||||
		"Max Time",
 | 
							"Max",
 | 
				
			||||||
		"Average Time",
 | 
							"Average",
 | 
				
			||||||
 | 
							"P90",
 | 
				
			||||||
 | 
							"P95",
 | 
				
			||||||
 | 
							"P99",
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var roundPrecision int64 = 4
 | 
				
			||||||
	for key, durations := range mergedResponses {
 | 
						for key, durations := range mergedResponses {
 | 
				
			||||||
 | 
							durations.Sort()
 | 
				
			||||||
 | 
							durationsLen := len(durations)
 | 
				
			||||||
 | 
							durationsLenAsFloat := float64(durationsLen - 1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		t.AppendRow(table.Row{
 | 
							t.AppendRow(table.Row{
 | 
				
			||||||
			key,
 | 
								key,
 | 
				
			||||||
			len(durations),
 | 
								durationsLen,
 | 
				
			||||||
			utils.MinDuration(durations...),
 | 
								utils.DurationRoundBy(*durations.First(), roundPrecision),
 | 
				
			||||||
			utils.MaxDuration(durations...),
 | 
								utils.DurationRoundBy(*durations.Last(), roundPrecision),
 | 
				
			||||||
			utils.AvgDuration(durations...),
 | 
								utils.DurationRoundBy(durations.Avg(), roundPrecision),
 | 
				
			||||||
 | 
								utils.DurationRoundBy(durations[int(0.90*durationsLenAsFloat)], roundPrecision),
 | 
				
			||||||
 | 
								utils.DurationRoundBy(durations[int(0.95*durationsLenAsFloat)], roundPrecision),
 | 
				
			||||||
 | 
								utils.DurationRoundBy(durations[int(0.99*durationsLenAsFloat)], roundPrecision),
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
		t.AppendSeparator()
 | 
							t.AppendSeparator()
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(mergedResponses) > 1 {
 | 
				
			||||||
		t.AppendRow(table.Row{
 | 
							t.AppendRow(table.Row{
 | 
				
			||||||
			"Total",
 | 
								"Total",
 | 
				
			||||||
		totalCount,
 | 
								total.Count,
 | 
				
			||||||
		totalMinDuration,
 | 
								utils.DurationRoundBy(total.Min, roundPrecision),
 | 
				
			||||||
		totalMaxDuration,
 | 
								utils.DurationRoundBy(total.Max, roundPrecision),
 | 
				
			||||||
		totalDuration / time.Duration(totalCount),
 | 
								utils.DurationRoundBy(total.Sum/time.Duration(total.Count), roundPrecision), // Average
 | 
				
			||||||
 | 
								utils.DurationRoundBy(total.P90, roundPrecision),
 | 
				
			||||||
 | 
								utils.DurationRoundBy(total.P95, roundPrecision),
 | 
				
			||||||
 | 
								utils.DurationRoundBy(total.P99, roundPrecision),
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	t.Render()
 | 
						t.Render()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,6 +2,7 @@ package requests
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"context"
 | 
						"context"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
	"sync"
 | 
						"sync"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -142,8 +143,7 @@ func sendRequest(
 | 
				
			|||||||
					return
 | 
										return
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
				*responseData = append(*responseData, &Response{
 | 
									*responseData = append(*responseData, &Response{
 | 
				
			||||||
					StatusCode: 0,
 | 
										Response: err.Error(),
 | 
				
			||||||
					Error:      err,
 | 
					 | 
				
			||||||
					Time:     completedTime,
 | 
										Time:     completedTime,
 | 
				
			||||||
				})
 | 
									})
 | 
				
			||||||
				increase <- 1
 | 
									increase <- 1
 | 
				
			||||||
@@ -151,8 +151,7 @@ func sendRequest(
 | 
				
			|||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
			*responseData = append(*responseData, &Response{
 | 
								*responseData = append(*responseData, &Response{
 | 
				
			||||||
				StatusCode: response.StatusCode(),
 | 
									Response: strconv.Itoa(response.StatusCode()),
 | 
				
			||||||
				Error:      nil,
 | 
					 | 
				
			||||||
				Time:     completedTime,
 | 
									Time:     completedTime,
 | 
				
			||||||
			})
 | 
								})
 | 
				
			||||||
			increase <- 1
 | 
								increase <- 1
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										41
									
								
								types/durations.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								types/durations.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,41 @@
 | 
				
			|||||||
 | 
					package types
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"sort"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Durations []time.Duration
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d Durations) Sort(ascending ...bool) {
 | 
				
			||||||
 | 
						// If ascending is provided and is false, sort in descending order
 | 
				
			||||||
 | 
						if len(ascending) > 0 && ascending[0] == false {
 | 
				
			||||||
 | 
							sort.Slice(d, func(i, j int) bool {
 | 
				
			||||||
 | 
								return d[i] > d[j]
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						} else { // Otherwise, sort in ascending order
 | 
				
			||||||
 | 
							sort.Slice(d, func(i, j int) bool {
 | 
				
			||||||
 | 
								return d[i] < d[j]
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d Durations) First() *time.Duration {
 | 
				
			||||||
 | 
						return &d[0]
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d Durations) Last() *time.Duration {
 | 
				
			||||||
 | 
						return &d[len(d)-1]
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d Durations) Sum() time.Duration {
 | 
				
			||||||
 | 
						sum := time.Duration(0)
 | 
				
			||||||
 | 
						for _, duration := range d {
 | 
				
			||||||
 | 
							sum += duration
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return sum
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d Durations) Avg() time.Duration {
 | 
				
			||||||
 | 
						return d.Sum() / time.Duration(len(d))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -1,4 +1,4 @@
 | 
				
			|||||||
package utils
 | 
					package types
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"encoding/json"
 | 
						"encoding/json"
 | 
				
			||||||
@@ -14,6 +14,8 @@ type Option[T NonNilT] interface {
 | 
				
			|||||||
	ValueOrErr() (T, error)
 | 
						ValueOrErr() (T, error)
 | 
				
			||||||
	ValueOr(def T) T
 | 
						ValueOr(def T) T
 | 
				
			||||||
	ValueOrPanic() T
 | 
						ValueOrPanic() T
 | 
				
			||||||
 | 
						SetValue(value T)
 | 
				
			||||||
 | 
						SetNone()
 | 
				
			||||||
	UnmarshalJSON(data []byte) error
 | 
						UnmarshalJSON(data []byte) error
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
							
								
								
									
										102
									
								
								utils/convert.go
									
									
									
									
									
								
							
							
						
						
									
										102
									
								
								utils/convert.go
									
									
									
									
									
								
							@@ -6,49 +6,77 @@ import (
 | 
				
			|||||||
	"reflect"
 | 
						"reflect"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func MarshalJSON(v any, maxSliceSize uint) string {
 | 
					type TruncatedMarshaller struct {
 | 
				
			||||||
	rv := reflect.ValueOf(v)
 | 
						Value    interface{}
 | 
				
			||||||
	if rv.Kind() == reflect.Slice && rv.Len() == 0 {
 | 
						MaxItems int
 | 
				
			||||||
		return "[]"
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	data, err := json.MarshalIndent(truncateLists(v, int(maxSliceSize)), "", "  ")
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return "{}"
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return string(data)
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func truncateLists(v interface{}, maxItems int) interface{} {
 | 
					func (t TruncatedMarshaller) MarshalJSON() ([]byte, error) {
 | 
				
			||||||
	rv := reflect.ValueOf(v)
 | 
						val := reflect.ValueOf(t.Value)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	switch rv.Kind() {
 | 
						if val.Kind() != reflect.Slice && val.Kind() != reflect.Array {
 | 
				
			||||||
	case reflect.Slice, reflect.Array:
 | 
							return json.Marshal(t.Value)
 | 
				
			||||||
		if rv.Len() > maxItems {
 | 
					 | 
				
			||||||
			newSlice := reflect.MakeSlice(rv.Type(), maxItems, maxItems)
 | 
					 | 
				
			||||||
			reflect.Copy(newSlice, rv.Slice(0, maxItems))
 | 
					 | 
				
			||||||
			newSlice = reflect.Append(newSlice, reflect.ValueOf(fmt.Sprintf("...(%d more)", rv.Len()-maxItems)))
 | 
					 | 
				
			||||||
			return newSlice.Interface()
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						length := val.Len()
 | 
				
			||||||
 | 
						if length <= t.MaxItems {
 | 
				
			||||||
 | 
							return json.Marshal(t.Value)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						truncated := make([]interface{}, t.MaxItems+1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for i := 0; i < t.MaxItems; i++ {
 | 
				
			||||||
 | 
							truncated[i] = val.Index(i).Interface()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						remaining := length - t.MaxItems
 | 
				
			||||||
 | 
						truncated[t.MaxItems] = fmt.Sprintf("+%d", remaining)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return json.Marshal(truncated)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func PrettyJSONMarshal(v interface{}, maxItems int, prefix, indent string) []byte {
 | 
				
			||||||
 | 
						truncated := processValue(v, maxItems)
 | 
				
			||||||
 | 
						d, _ := json.MarshalIndent(truncated, prefix, indent)
 | 
				
			||||||
 | 
						return d
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func processValue(v interface{}, maxItems int) interface{} {
 | 
				
			||||||
 | 
						val := reflect.ValueOf(v)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch val.Kind() {
 | 
				
			||||||
	case reflect.Map:
 | 
						case reflect.Map:
 | 
				
			||||||
		newMap := reflect.MakeMap(rv.Type())
 | 
							newMap := make(map[string]interface{})
 | 
				
			||||||
		for _, key := range rv.MapKeys() {
 | 
							iter := val.MapRange()
 | 
				
			||||||
			newMap.SetMapIndex(key, reflect.ValueOf(truncateLists(rv.MapIndex(key).Interface(), maxItems)))
 | 
							for iter.Next() {
 | 
				
			||||||
		}
 | 
								k := iter.Key().String()
 | 
				
			||||||
		return newMap.Interface()
 | 
								newMap[k] = processValue(iter.Value().Interface(), maxItems)
 | 
				
			||||||
	case reflect.Struct:
 | 
					 | 
				
			||||||
		newStruct := reflect.New(rv.Type()).Elem()
 | 
					 | 
				
			||||||
		for i := 0; i < rv.NumField(); i++ {
 | 
					 | 
				
			||||||
			newStruct.Field(i).Set(reflect.ValueOf(truncateLists(rv.Field(i).Interface(), maxItems)))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		return newStruct.Interface()
 | 
					 | 
				
			||||||
	case reflect.Ptr:
 | 
					 | 
				
			||||||
		if rv.IsNil() {
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		return truncateLists(rv.Elem().Interface(), maxItems)
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							return newMap
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case reflect.Slice, reflect.Array:
 | 
				
			||||||
 | 
							return TruncatedMarshaller{Value: v, MaxItems: maxItems}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case reflect.Struct:
 | 
				
			||||||
 | 
							newMap := make(map[string]interface{})
 | 
				
			||||||
 | 
							t := val.Type()
 | 
				
			||||||
 | 
							for i := 0; i < t.NumField(); i++ {
 | 
				
			||||||
 | 
								field := t.Field(i)
 | 
				
			||||||
 | 
								if field.IsExported() {
 | 
				
			||||||
 | 
									jsonTag := field.Tag.Get("json")
 | 
				
			||||||
 | 
									if jsonTag == "-" {
 | 
				
			||||||
 | 
										continue
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									fieldName := field.Name
 | 
				
			||||||
 | 
									if jsonTag != "" {
 | 
				
			||||||
 | 
										fieldName = jsonTag
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									newMap[fieldName] = processValue(val.Field(i).Interface(), maxItems)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return newMap
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
		return v
 | 
							return v
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										21
									
								
								utils/int.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								utils/int.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
				
			|||||||
 | 
					package utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Number interface {
 | 
				
			||||||
 | 
						int | int8 | int16 | int32 | int64
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func NumLen[T Number](n T) T {
 | 
				
			||||||
 | 
						if n < 0 {
 | 
				
			||||||
 | 
							n = -n
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if n == 0 {
 | 
				
			||||||
 | 
							return 1
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var count T = 0
 | 
				
			||||||
 | 
						for n > 0 {
 | 
				
			||||||
 | 
							n /= 10
 | 
				
			||||||
 | 
							count++
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return count
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -2,30 +2,13 @@ package utils
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import "time"
 | 
					import "time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func MinDuration(durations ...time.Duration) time.Duration {
 | 
					func DurationRoundBy(duration time.Duration, n int64) time.Duration {
 | 
				
			||||||
	min := durations[0]
 | 
						if durationLen := NumLen(duration.Nanoseconds()); durationLen > n {
 | 
				
			||||||
	for _, d := range durations {
 | 
							roundNum := 1
 | 
				
			||||||
		if d < min {
 | 
							for range durationLen - n {
 | 
				
			||||||
			min = d
 | 
								roundNum *= 10
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							return duration.Round(time.Duration(roundNum))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return min
 | 
						return duration
 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func MaxDuration(durations ...time.Duration) time.Duration {
 | 
					 | 
				
			||||||
	max := durations[0]
 | 
					 | 
				
			||||||
	for _, d := range durations {
 | 
					 | 
				
			||||||
		if d > max {
 | 
					 | 
				
			||||||
			max = d
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	return max
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func AvgDuration(durations ...time.Duration) time.Duration {
 | 
					 | 
				
			||||||
	total := time.Duration(0)
 | 
					 | 
				
			||||||
	for _, d := range durations {
 | 
					 | 
				
			||||||
		total += d
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	return total / time.Duration(len(durations))
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user