mirror of
https://github.com/aykhans/dodo.git
synced 2025-04-22 11:35:57 +00:00
Compare commits
3 Commits
0725c1a481
...
2e6110e54a
Author | SHA1 | Date | |
---|---|---|---|
2e6110e54a | |||
48721b4ae7 | |||
fe4a93b394 |
@ -1,6 +1,6 @@
|
|||||||
<h1 align="center">Dodo is a simple and easy-to-use HTTP benchmarking tool.</h1>
|
<h1 align="center">Dodo is a simple and easy-to-use HTTP benchmarking tool.</h1>
|
||||||
<p align="center">
|
<p align="center">
|
||||||
<img width="30%" height="30%" src="https://raw.githubusercontent.com/aykhans/dodo/main/assets/dodo.png">
|
<img width="30%" height="30%" src="https://ftp.aykhans.me/web/client/pubshares/hB6VSdCnBCr8gFPeiMuCji/browse?path=%2Fdodo.png">
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
BIN
assets/dodo.png
BIN
assets/dodo.png
Binary file not shown.
Before Width: | Height: | Size: 230 KiB |
@ -21,10 +21,6 @@ const (
|
|||||||
MaxDodosCountForProxies uint = 20 // Max dodos count for proxy check
|
MaxDodosCountForProxies uint = 20 // Max dodos count for proxy check
|
||||||
)
|
)
|
||||||
|
|
||||||
type IConfig interface {
|
|
||||||
MergeConfigs(newConfig IConfig) IConfig
|
|
||||||
}
|
|
||||||
|
|
||||||
type RequestConfig struct {
|
type RequestConfig struct {
|
||||||
Method string
|
Method string
|
||||||
URL *url.URL
|
URL *url.URL
|
||||||
@ -48,6 +44,12 @@ func (config *RequestConfig) Print() {
|
|||||||
{Number: 2, WidthMax: 50},
|
{Number: 2, WidthMax: 50},
|
||||||
})
|
})
|
||||||
|
|
||||||
|
newHeaders := make(map[string][]string)
|
||||||
|
newHeaders["User-Agent"] = []string{DefaultUserAgent}
|
||||||
|
for k, v := range config.Headers {
|
||||||
|
newHeaders[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
t.AppendHeader(table.Row{"Request Configuration"})
|
t.AppendHeader(table.Row{"Request Configuration"})
|
||||||
t.AppendRow(table.Row{"Method", config.Method})
|
t.AppendRow(table.Row{"Method", config.Method})
|
||||||
t.AppendSeparator()
|
t.AppendSeparator()
|
||||||
@ -57,11 +59,11 @@ func (config *RequestConfig) Print() {
|
|||||||
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{"Request", 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", utils.MarshalJSON(config.Params, 3)})
|
||||||
t.AppendSeparator()
|
t.AppendSeparator()
|
||||||
t.AppendRow(table.Row{"Headers", utils.MarshalJSON(config.Headers, 3)})
|
t.AppendRow(table.Row{"Headers", utils.MarshalJSON(newHeaders, 3)})
|
||||||
t.AppendSeparator()
|
t.AppendSeparator()
|
||||||
t.AppendRow(table.Row{"Cookies", utils.MarshalJSON(config.Cookies, 3)})
|
t.AppendRow(table.Row{"Cookies", utils.MarshalJSON(config.Cookies, 3)})
|
||||||
t.AppendSeparator()
|
t.AppendSeparator()
|
||||||
@ -98,6 +100,26 @@ type Config struct {
|
|||||||
NoProxyCheck utils.Option[bool] `json:"no_proxy_check"`
|
NoProxyCheck utils.Option[bool] `json:"no_proxy_check"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewConfig(
|
||||||
|
method string,
|
||||||
|
timeout uint32,
|
||||||
|
dodosCount uint,
|
||||||
|
requestCount uint,
|
||||||
|
noProxyCheck utils.Option[bool],
|
||||||
|
) *Config {
|
||||||
|
if noProxyCheck == nil {
|
||||||
|
noProxyCheck = utils.NewNoneOption[bool]()
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Config{
|
||||||
|
Method: method,
|
||||||
|
Timeout: timeout,
|
||||||
|
DodosCount: dodosCount,
|
||||||
|
RequestCount: requestCount,
|
||||||
|
NoProxyCheck: noProxyCheck,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (config *Config) MergeConfigs(newConfig *Config) {
|
func (config *Config) MergeConfigs(newConfig *Config) {
|
||||||
if newConfig.Method != "" {
|
if newConfig.Method != "" {
|
||||||
config.Method = newConfig.Method
|
config.Method = newConfig.Method
|
||||||
@ -144,7 +166,7 @@ type Proxy struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type JSONConfig struct {
|
type JSONConfig struct {
|
||||||
Config
|
*Config
|
||||||
Params map[string][]string `json:"params"`
|
Params map[string][]string `json:"params"`
|
||||||
Headers map[string][]string `json:"headers"`
|
Headers map[string][]string `json:"headers"`
|
||||||
Cookies map[string][]string `json:"cookies"`
|
Cookies map[string][]string `json:"cookies"`
|
||||||
@ -152,8 +174,21 @@ type JSONConfig struct {
|
|||||||
Body []string `json:"body"`
|
Body []string `json:"body"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewJSONConfig(
|
||||||
|
config *Config,
|
||||||
|
params map[string][]string,
|
||||||
|
headers map[string][]string,
|
||||||
|
cookies map[string][]string,
|
||||||
|
proxies []Proxy,
|
||||||
|
body []string,
|
||||||
|
) *JSONConfig {
|
||||||
|
return &JSONConfig{
|
||||||
|
config, params, headers, cookies, proxies, body,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (config *JSONConfig) MergeConfigs(newConfig *JSONConfig) {
|
func (config *JSONConfig) MergeConfigs(newConfig *JSONConfig) {
|
||||||
config.Config.MergeConfigs(&newConfig.Config)
|
config.Config.MergeConfigs(newConfig.Config)
|
||||||
if len(newConfig.Params) != 0 {
|
if len(newConfig.Params) != 0 {
|
||||||
config.Params = newConfig.Params
|
config.Params = newConfig.Params
|
||||||
}
|
}
|
||||||
@ -172,13 +207,23 @@ func (config *JSONConfig) MergeConfigs(newConfig *JSONConfig) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type CLIConfig struct {
|
type CLIConfig struct {
|
||||||
Config
|
*Config
|
||||||
Yes bool `json:"yes" validate:"omitempty"`
|
Yes 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(
|
||||||
|
config *Config,
|
||||||
|
yes bool,
|
||||||
|
configFile string,
|
||||||
|
) *CLIConfig {
|
||||||
|
return &CLIConfig{
|
||||||
|
config, yes, configFile,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (config *CLIConfig) MergeConfigs(newConfig *CLIConfig) {
|
func (config *CLIConfig) MergeConfigs(newConfig *CLIConfig) {
|
||||||
config.Config.MergeConfigs(&newConfig.Config)
|
config.Config.MergeConfigs(newConfig.Config)
|
||||||
if newConfig.ConfigFile != "" {
|
if newConfig.ConfigFile != "" {
|
||||||
config.ConfigFile = newConfig.ConfigFile
|
config.ConfigFile = newConfig.ConfigFile
|
||||||
}
|
}
|
||||||
|
12
main.go
12
main.go
@ -21,8 +21,10 @@ import (
|
|||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
validator := validation.NewValidator()
|
validator := validation.NewValidator()
|
||||||
conf := config.Config{}
|
conf := config.NewConfig("", 0, 0, 0, nil)
|
||||||
jsonConf := config.JSONConfig{}
|
jsonConf := config.NewJSONConfig(
|
||||||
|
config.NewConfig("", 0, 0, 0, nil), nil, nil, nil, nil, nil,
|
||||||
|
)
|
||||||
|
|
||||||
cliConf, err := readers.CLIConfigReader()
|
cliConf, err := readers.CLIConfigReader()
|
||||||
if err != nil || cliConf == nil {
|
if err != nil || cliConf == nil {
|
||||||
@ -51,11 +53,11 @@ func main() {
|
|||||||
),
|
),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
jsonConf = *jsonConfNew
|
jsonConf = jsonConfNew
|
||||||
conf.MergeConfigs(&jsonConf.Config)
|
conf.MergeConfigs(jsonConf.Config)
|
||||||
}
|
}
|
||||||
|
|
||||||
conf.MergeConfigs(&cliConf.Config)
|
conf.MergeConfigs(cliConf.Config)
|
||||||
conf.SetDefaults()
|
conf.SetDefaults()
|
||||||
if err := validator.Struct(conf); err != nil {
|
if err := validator.Struct(conf); err != nil {
|
||||||
utils.PrintErrAndExit(
|
utils.PrintErrAndExit(
|
||||||
|
@ -13,11 +13,7 @@ import (
|
|||||||
func CLIConfigReader() (*config.CLIConfig, error) {
|
func CLIConfigReader() (*config.CLIConfig, error) {
|
||||||
var (
|
var (
|
||||||
returnNil = false
|
returnNil = false
|
||||||
cliConfig = &config.CLIConfig{
|
cliConfig = config.NewCLIConfig(config.NewConfig("", 0, 0, 0, nil), false, "")
|
||||||
Config: config.Config{
|
|
||||||
NoProxyCheck: utils.NewNoneOption[bool](),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
dodosCount uint
|
dodosCount uint
|
||||||
requestCount uint
|
requestCount uint
|
||||||
timeout uint32
|
timeout uint32
|
||||||
|
@ -6,7 +6,6 @@ import (
|
|||||||
|
|
||||||
"github.com/aykhans/dodo/config"
|
"github.com/aykhans/dodo/config"
|
||||||
customerrors "github.com/aykhans/dodo/custom_errors"
|
customerrors "github.com/aykhans/dodo/custom_errors"
|
||||||
"github.com/aykhans/dodo/utils"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func JSONConfigReader(filePath string) (*config.JSONConfig, error) {
|
func JSONConfigReader(filePath string) (*config.JSONConfig, error) {
|
||||||
@ -14,11 +13,10 @@ func JSONConfigReader(filePath string) (*config.JSONConfig, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, customerrors.OSErrorFormater(err)
|
return nil, customerrors.OSErrorFormater(err)
|
||||||
}
|
}
|
||||||
jsonConf := &config.JSONConfig{
|
jsonConf := config.NewJSONConfig(
|
||||||
Config: config.Config{
|
config.NewConfig("", 0, 0, 0, nil),
|
||||||
NoProxyCheck: utils.NewNoneOption[bool](),
|
nil, nil, nil, nil, nil,
|
||||||
},
|
)
|
||||||
}
|
|
||||||
err = json.Unmarshal(data, &jsonConf)
|
err = json.Unmarshal(data, &jsonConf)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user