1 Commits

Author SHA1 Message Date
159ab66b58 Merge adc5a34891 into 565079fe97 2024-11-23 01:21:32 +04:00
12 changed files with 67 additions and 140 deletions

View File

@ -1,4 +1,4 @@
FROM golang:1.23.2-alpine AS builder FROM golang:1.22.6-alpine AS builder
WORKDIR /dodo WORKDIR /dodo

View File

@ -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://ftp.aykhans.me/web/client/pubshares/hB6VSdCnBCr8gFPeiMuCji/browse?path=%2Fdodo.png"> <img width="30%" height="30%" src="https://raw.githubusercontent.com/aykhans/dodo/main/assets/dodo.png">
</p> </p>
## Installation ## Installation
@ -33,7 +33,7 @@ Follow the steps below to build dodo:
3. **Build the project:** 3. **Build the project:**
```sh ```sh
go build -ldflags "-s -w" -o dodo go build -o dodo
``` ```
This will generate an executable named `dodo` in the project directory. This will generate an executable named `dodo` in the project directory.
@ -58,9 +58,8 @@ You can find an example config structure in the [config.json](https://github.com
{ {
"method": "GET", "method": "GET",
"url": "https://example.com", "url": "https://example.com",
"no_proxy_check": false, "timeout": 10000,
"timeout": 2000, "dodos_count": 50,
"dodos_count": 10,
"request_count": 1000, "request_count": 1000,
"params": {}, "params": {},
"headers": {}, "headers": {},
@ -78,7 +77,7 @@ You can find an example config structure in the [config.json](https://github.com
] ]
} }
``` ```
Send 1000 GET requests to https://example.com with 10 parallel dodos (threads) and a timeout of 2000 milliseconds: Send 1000 GET requests to https://example.com with 5 parallel dodos (threads) and a timeout of 10000 milliseconds:
```sh ```sh
dodo -c /path/config.json dodo -c /path/config.json

BIN
assets/dodo.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 230 KiB

View File

@ -1,7 +1,7 @@
{ {
"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": 50,
"request_count": 1000, "request_count": 1000,

View File

@ -11,7 +11,7 @@ import (
) )
const ( const (
VERSION string = "0.5.2" VERSION string = "0.5.1"
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"
@ -21,6 +21,10 @@ 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
@ -44,12 +48,6 @@ 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()
@ -59,11 +57,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{"Requests", config.RequestCount}) t.AppendRow(table.Row{"Request", 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(newHeaders, 3)}) t.AppendRow(table.Row{"Headers", utils.MarshalJSON(config.Headers, 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()
@ -100,26 +98,6 @@ 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
@ -166,7 +144,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"`
@ -174,21 +152,8 @@ 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
} }
@ -207,23 +172,13 @@ 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
} }

2
go.mod
View File

@ -1,6 +1,6 @@
module github.com/aykhans/dodo module github.com/aykhans/dodo
go 1.23.2 go 1.22.6
require ( require (
github.com/go-playground/validator/v10 v10.23.0 github.com/go-playground/validator/v10 v10.23.0

14
main.go
View File

@ -21,10 +21,8 @@ import (
func main() { func main() {
validator := validation.NewValidator() validator := validation.NewValidator()
conf := config.NewConfig("", 0, 0, 0, nil) conf := config.Config{}
jsonConf := config.NewJSONConfig( jsonConf := config.JSONConfig{}
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 {
@ -53,11 +51,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(
@ -83,7 +81,7 @@ func main() {
Proxies: jsonConf.Proxies, Proxies: jsonConf.Proxies,
Body: jsonConf.Body, Body: jsonConf.Body,
Yes: cliConf.Yes, Yes: cliConf.Yes,
NoProxyCheck: conf.NoProxyCheck.ValueOrPanic(), NoProxyCheck: *conf.NoProxyCheck.ValueOrPanic(),
} }
requestConf.Print() requestConf.Print()
if !cliConf.Yes { if !cliConf.Yes {

View File

@ -13,7 +13,11 @@ import (
func CLIConfigReader() (*config.CLIConfig, error) { func CLIConfigReader() (*config.CLIConfig, error) {
var ( var (
returnNil = false returnNil = false
cliConfig = config.NewCLIConfig(config.NewConfig("", 0, 0, 0, nil), false, "") cliConfig = &config.CLIConfig{
Config: config.Config{
NoProxyCheck: utils.NewNoneOption[bool](),
},
}
dodosCount uint dodosCount uint
requestCount uint requestCount uint
timeout uint32 timeout uint32
@ -27,7 +31,7 @@ func CLIConfigReader() (*config.CLIConfig, error) {
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`,
Run: func(cmd *cobra.Command, args []string) {}, Run: func(cmd *cobra.Command, args []string) {},
SilenceErrors: true, SilenceErrors: true,
SilenceUsage: true, SilenceUsage: true,

View File

@ -6,6 +6,7 @@ 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) {
@ -13,10 +14,11 @@ 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.NewJSONConfig( jsonConf := &config.JSONConfig{
config.NewConfig("", 0, 0, 0, nil), Config: config.Config{
nil, nil, nil, nil, nil, NoProxyCheck: utils.NewNoneOption[bool](),
) },
}
err = json.Unmarshal(data, &jsonConf) err = json.Unmarshal(data, &jsonConf)
if err != nil { if err != nil {

View File

@ -47,15 +47,15 @@ func getClients(
} }
clients = append(clients, &fasthttp.HostClient{ clients = append(clients, &fasthttp.HostClient{
MaxConns: int(maxConns), MaxConns: int(maxConns),
IsTLS: isTLS, IsTLS: isTLS,
Addr: addr, Addr: addr,
Dial: dialFunc, Dial: dialFunc,
MaxIdleConnDuration: timeout, MaxIdleConnDuration: timeout,
MaxConnDuration: timeout, MaxConnDuration: timeout,
WriteTimeout: timeout, WriteTimeout: timeout,
ReadTimeout: timeout, ReadTimeout: timeout,
}, },
) )
} }
return clients return clients

View File

@ -2,6 +2,7 @@ package requests
import ( import (
"context" "context"
"fmt"
"sync" "sync"
"time" "time"
@ -24,6 +25,8 @@ import (
// - Responses: A collection of responses from the executed requests. // - Responses: A collection of responses from the executed requests.
// - error: An error if the operation fails, such as no internet connection or an interrupt. // - error: An error if the operation fails, such as no internet connection or an interrupt.
func Run(ctx context.Context, requestConfig *config.RequestConfig) (Responses, error) { func Run(ctx context.Context, requestConfig *config.RequestConfig) (Responses, error) {
fmt.Println("No Proxy Check:", requestConfig.NoProxyCheck)
checkConnectionCtx, checkConnectionCtxCancel := context.WithTimeout(ctx, 8*time.Second) checkConnectionCtx, checkConnectionCtxCancel := context.WithTimeout(ctx, 8*time.Second)
if !checkConnection(checkConnectionCtx) { if !checkConnection(checkConnectionCtx) {
checkConnectionCtxCancel() checkConnectionCtxCancel()

View File

@ -5,83 +5,49 @@ import (
"errors" "errors"
) )
type NonNilT interface { type Option[T any] struct {
~int | ~float64 | ~string | ~bool
}
type Option[T NonNilT] interface {
IsNone() bool
ValueOrErr() (T, error)
ValueOr(def T) T
ValueOrPanic() T
UnmarshalJSON(data []byte) error
}
// Don't call this struct directly, use NewOption[T] or NewNoneOption[T] instead.
type option[T NonNilT] struct {
// value holds the actual value of the Option if it is not None.
value T value T
// none indicates whether the Option is None (i.e., has no value). none bool
none bool
} }
func (o *option[T]) IsNone() bool { func (o *Option[T]) IsNone() bool {
return o.none return o.none
} }
// If the Option is None, it will return zero value of the type and an error. func (o *Option[T]) ValueOrErr() (*T, error) {
func (o *option[T]) ValueOrErr() (T, error) {
if o.IsNone() { if o.IsNone() {
return o.value, errors.New("Option is None") return nil, errors.New("Option is None")
} }
return o.value, nil return &o.value, nil
} }
// If the Option is None, it will return the default value. func (o *Option[T]) ValueOr(def *T) *T {
func (o *option[T]) ValueOr(def T) T {
if o.IsNone() { if o.IsNone() {
return def return def
} }
return o.value return &o.value
} }
// If the Option is None, it will panic. func (o *Option[T]) ValueOrPanic() *T {
func (o *option[T]) ValueOrPanic() T {
if o.IsNone() { if o.IsNone() {
panic("Option is None") panic("Option is None")
} }
return o.value return &o.value
} }
func (o *option[T]) SetValue(value T) { func (o *Option[T]) UnmarshalJSON(data []byte) error {
o.value = value if string(data) == "null" {
o.none = false o.none = true
}
func (o *option[T]) SetNone() {
var zeroValue T
o.value = zeroValue
o.none = true
}
func (o *option[T]) UnmarshalJSON(data []byte) error {
if string(data) == "null" || len(data) == 0 {
o.SetNone()
return nil return nil
} }
if err := json.Unmarshal(data, &o.value); err != nil {
o.SetNone()
return err
}
o.none = false o.none = false
return nil return json.Unmarshal(data, &o.value)
} }
func NewOption[T NonNilT](value T) *option[T] { func NewOption[T any](value T) Option[T] {
return &option[T]{value: value} return Option[T]{value: value}
} }
func NewNoneOption[T NonNilT]() *option[T] { func NewNoneOption[T any]() Option[T] {
return &option[T]{none: true} return Option[T]{none: true}
} }