mirror of
https://github.com/aykhans/dodo.git
synced 2025-07-01 16:07:49 +00:00
Compare commits
34 Commits
39d366aadb
...
v0.5.3
Author | SHA1 | Date | |
---|---|---|---|
04e5b5f3eb | |||
2bebf09adb | |||
ebacebff16 | |||
575ca2913a | |||
4686e26ede | |||
bd33b2c6a2 | |||
3068e0acae | |||
7a40bbbd3c | |||
43c0bd2138 | |||
fddeec1791 | |||
6f5fa1860c | |||
4480aed0f0 | |||
f5144ac4be | |||
4a1051e08c | |||
2371986909 | |||
e9af46084d | |||
cdaa63152f | |||
16791421c7 | |||
c90cfd8090 | |||
1aaea8e437 | |||
c44a0c978a | |||
1ace7be2b4 | |||
fa55de0cf2 | |||
d392d4a787 | |||
2e6110e54a | |||
48721b4ae7 | |||
fe4a93b394 | |||
0725c1a481 | |||
e3cbe1e9b4 | |||
94211eb018 | |||
565079fe97 | |||
696176e611 | |||
9ea06593f9 | |||
7f7dc07f4c |
86
.github/workflows/publish-docker-image.yml
vendored
Normal file
86
.github/workflows/publish-docker-image.yml
vendored
Normal file
@ -0,0 +1,86 @@
|
|||||||
|
name: publish-docker-image
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
# Match stable and pre versions, such as 'v1.0.0', 'v0.23.0-a', 'v0.23.0-a.2', 'v0.23.0-b', 'v0.23.0-b.3'
|
||||||
|
- "v*.*.*"
|
||||||
|
- "v*.*.*-a"
|
||||||
|
- "v*.*.*-a.*"
|
||||||
|
- "v*.*.*-b"
|
||||||
|
- "v*.*.*-b.*"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build-and-push-stable-image:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
permissions:
|
||||||
|
contents: read
|
||||||
|
packages: write
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Set up QEMU
|
||||||
|
uses: docker/setup-qemu-action@v3
|
||||||
|
|
||||||
|
- name: Extract build args
|
||||||
|
# Extract version number and check if it's an pre version
|
||||||
|
run: |
|
||||||
|
if [[ "${GITHUB_REF_NAME}" =~ ^v[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
|
||||||
|
echo "PRE_RELEASE=false" >> $GITHUB_ENV
|
||||||
|
else
|
||||||
|
echo "PRE_RELEASE=true" >> $GITHUB_ENV
|
||||||
|
fi
|
||||||
|
echo "VERSION=${GITHUB_REF_NAME#v}" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Login to Docker Hub
|
||||||
|
uses: docker/login-action@v3
|
||||||
|
with:
|
||||||
|
username: aykhans
|
||||||
|
password: ${{ secrets.DOCKER_TOKEN }}
|
||||||
|
|
||||||
|
- name: Set up Docker Buildx
|
||||||
|
id: buildx
|
||||||
|
uses: docker/setup-buildx-action@v3
|
||||||
|
with:
|
||||||
|
install: true
|
||||||
|
version: v0.9.1
|
||||||
|
|
||||||
|
# Metadata for stable versions
|
||||||
|
- name: Docker meta for stable
|
||||||
|
id: meta-stable
|
||||||
|
if: env.PRE_RELEASE == 'false'
|
||||||
|
uses: docker/metadata-action@v5
|
||||||
|
with:
|
||||||
|
images: |
|
||||||
|
aykhans/dodo
|
||||||
|
tags: |
|
||||||
|
type=semver,pattern={{version}},value=${{ env.VERSION }}
|
||||||
|
type=raw,value=stable
|
||||||
|
flavor: |
|
||||||
|
latest=true
|
||||||
|
labels: |
|
||||||
|
org.opencontainers.image.version=${{ env.VERSION }}
|
||||||
|
|
||||||
|
# Metadata for pre versions
|
||||||
|
- name: Docker meta for pre
|
||||||
|
id: meta-pre
|
||||||
|
if: env.PRE_RELEASE == 'true'
|
||||||
|
uses: docker/metadata-action@v5
|
||||||
|
with:
|
||||||
|
images: |
|
||||||
|
aykhans/dodo
|
||||||
|
tags: |
|
||||||
|
type=raw,value=${{ env.VERSION }}
|
||||||
|
labels: |
|
||||||
|
org.opencontainers.image.version=${{ env.VERSION }}
|
||||||
|
|
||||||
|
- name: Build and Push
|
||||||
|
id: docker_build
|
||||||
|
uses: docker/build-push-action@v6
|
||||||
|
with:
|
||||||
|
context: ./
|
||||||
|
file: ./Dockerfile
|
||||||
|
platforms: linux/amd64,linux/arm64
|
||||||
|
push: true
|
||||||
|
tags: ${{ steps.meta-stable.outputs.tags || steps.meta-pre.outputs.tags }}
|
||||||
|
labels: ${{ steps.meta-stable.outputs.labels || steps.meta-pre.outputs.labels }}
|
@ -1,4 +1,4 @@
|
|||||||
FROM golang:1.22.6-alpine AS builder
|
FROM golang:1.23.2-alpine AS builder
|
||||||
|
|
||||||
WORKDIR /dodo
|
WORKDIR /dodo
|
||||||
|
|
||||||
|
@ -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
|
||||||
@ -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
|
||||||
@ -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
|
||||||
|
BIN
assets/dodo.png
BIN
assets/dodo.png
Binary file not shown.
Before Width: | Height: | Size: 230 KiB |
@ -6,12 +6,13 @@ import (
|
|||||||
"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"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
VERSION string = "0.5.2"
|
VERSION string = "0.5.3"
|
||||||
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,10 +22,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 +45,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 +60,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()
|
||||||
@ -95,7 +98,27 @@ 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(
|
||||||
|
method string,
|
||||||
|
timeout uint32,
|
||||||
|
dodosCount uint,
|
||||||
|
requestCount uint,
|
||||||
|
noProxyCheck Option[bool],
|
||||||
|
) *Config {
|
||||||
|
if noProxyCheck == nil {
|
||||||
|
noProxyCheck = 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) {
|
||||||
@ -133,7 +156,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)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -144,7 +167,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 +175,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 +208,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
|
||||||
}
|
}
|
||||||
|
16
go.mod
16
go.mod
@ -1,14 +1,14 @@
|
|||||||
module github.com/aykhans/dodo
|
module github.com/aykhans/dodo
|
||||||
|
|
||||||
go 1.22.6
|
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/spf13/cobra v1.8.1
|
||||||
github.com/spf13/pflag v1.0.5
|
github.com/spf13/pflag v1.0.5
|
||||||
github.com/valyala/fasthttp v1.57.0
|
github.com/valyala/fasthttp v1.58.0
|
||||||
golang.org/x/net v0.31.0
|
golang.org/x/net v0.32.0
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@ -22,8 +22,8 @@ require (
|
|||||||
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
|
||||||
)
|
)
|
||||||
|
28
go.sum
28
go.sum
@ -15,8 +15,8 @@ github.com/go-playground/validator/v10 v10.23.0 h1:/PwmTwZhS0dPkav3cdK9kV1FsAmrL
|
|||||||
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/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
||||||
github.com/jedib0t/go-pretty/v6 v6.6.2 h1:27bLj3nRODzaiA7tPIxy9UVWHoPspFfME9XxgwiiNsM=
|
github.com/jedib0t/go-pretty/v6 v6.6.5 h1:9PgMJOVBedpgYLI56jQRJYqngxYAAzfEUua+3NgSqAo=
|
||||||
github.com/jedib0t/go-pretty/v6 v6.6.2/go.mod h1:zbn98qrYlh95FIhwwsbIip0LYpwSG8SUOScs+v9/t0E=
|
github.com/jedib0t/go-pretty/v6 v6.6.5/go.mod h1:Uq/HrbhuFty5WSVNfjpQQe47x16RwVGXIveNGEyGtHs=
|
||||||
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=
|
||||||
@ -37,20 +37,20 @@ github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcU
|
|||||||
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.32.0 h1:ZqPmj8Kzc+Y6e0+skZsuACbx+wzMgo5MQsJh9Qd6aYI=
|
||||||
golang.org/x/net v0.31.0/go.mod h1:P4fl1q7dY2hnZFxEk4pPSkDHF+QqjitcnDjUQyMM+pM=
|
golang.org/x/net v0.32.0/go.mod h1:CwU0IoeOlnQQWJ6ioyFrfRuomB8GKF6KbYXZVyeXNfs=
|
||||||
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/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=
|
||||||
|
14
main.go
14
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(
|
||||||
@ -81,7 +83,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 {
|
||||||
|
@ -5,6 +5,7 @@ import (
|
|||||||
|
|
||||||
"github.com/aykhans/dodo/config"
|
"github.com/aykhans/dodo/config"
|
||||||
"github.com/aykhans/dodo/custom_errors"
|
"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/cobra"
|
||||||
"github.com/spf13/pflag"
|
"github.com/spf13/pflag"
|
||||||
@ -13,11 +14,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
|
||||||
@ -65,7 +62,7 @@ func CLIConfigReader() (*config.CLIConfig, error) {
|
|||||||
case "timeout":
|
case "timeout":
|
||||||
cliConfig.Timeout = timeout
|
cliConfig.Timeout = timeout
|
||||||
case "no-proxy-check":
|
case "no-proxy-check":
|
||||||
cliConfig.NoProxyCheck = utils.NewOption(noProxyCheck)
|
cliConfig.NoProxyCheck = NewOption(noProxyCheck)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
if returnNil {
|
if returnNil {
|
||||||
|
@ -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 {
|
||||||
|
@ -1,85 +1,106 @@
|
|||||||
package requests
|
package requests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"os"
|
"os"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/aykhans/dodo/utils"
|
. "github.com/aykhans/dodo/types"
|
||||||
"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 (respones 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
|
||||||
|
}{
|
||||||
|
Count: len(respones),
|
||||||
|
Min: respones[0].Time,
|
||||||
|
Max: respones[0].Time,
|
||||||
|
}
|
||||||
|
mergedResponses := make(map[string]Durations)
|
||||||
|
var allDurations Durations
|
||||||
|
|
||||||
for _, response := range respones {
|
for _, response := range respones {
|
||||||
if response.Time < totalMinDuration {
|
if response.Time < total.Min {
|
||||||
totalMinDuration = response.Time
|
total.Min = response.Time
|
||||||
}
|
}
|
||||||
if response.Time > totalMaxDuration {
|
if response.Time > total.Max {
|
||||||
totalMaxDuration = response.Time
|
total.Max = response.Time
|
||||||
}
|
}
|
||||||
totalDuration += response.Time
|
total.Sum += response.Time
|
||||||
|
|
||||||
if response.Error != nil {
|
mergedResponses[response.Response] = append(
|
||||||
mergedResponses[response.Error.Error()] = append(
|
mergedResponses[response.Response],
|
||||||
mergedResponses[response.Error.Error()],
|
|
||||||
response.Time,
|
|
||||||
)
|
|
||||||
} else {
|
|
||||||
mergedResponses[fmt.Sprintf("%d", response.StatusCode)] = append(
|
|
||||||
mergedResponses[fmt.Sprintf("%d", response.StatusCode)],
|
|
||||||
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 Time",
|
||||||
"Max Time",
|
"Max Time",
|
||||||
"Average Time",
|
"Average Time",
|
||||||
|
"P90",
|
||||||
|
"P95",
|
||||||
|
"P99",
|
||||||
})
|
})
|
||||||
|
|
||||||
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...),
|
durations.First(),
|
||||||
utils.MaxDuration(durations...),
|
durations.Last(),
|
||||||
utils.AvgDuration(durations...),
|
durations.Avg(),
|
||||||
|
durations[int(0.90*durationsLenAsFloat)],
|
||||||
|
durations[int(0.95*durationsLenAsFloat)],
|
||||||
|
durations[int(0.99*durationsLenAsFloat)],
|
||||||
})
|
})
|
||||||
t.AppendSeparator()
|
t.AppendSeparator()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if len(mergedResponses) > 1 {
|
||||||
t.AppendRow(table.Row{
|
t.AppendRow(table.Row{
|
||||||
"Total",
|
"Total",
|
||||||
totalCount,
|
total.Count,
|
||||||
totalMinDuration,
|
total.Min,
|
||||||
totalMaxDuration,
|
total.Max,
|
||||||
totalDuration / time.Duration(totalCount),
|
total.Sum / time.Duration(total.Count), // Average
|
||||||
|
total.P90,
|
||||||
|
total.P95,
|
||||||
|
total.P99,
|
||||||
})
|
})
|
||||||
|
}
|
||||||
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))
|
||||||
|
}
|
87
types/option.go
Normal file
87
types/option.go
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
package types
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NonNilT interface {
|
||||||
|
~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
|
||||||
|
// none indicates whether the Option is None (i.e., has no value).
|
||||||
|
none bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *option[T]) IsNone() bool {
|
||||||
|
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) {
|
||||||
|
if o.IsNone() {
|
||||||
|
return o.value, errors.New("Option is None")
|
||||||
|
}
|
||||||
|
return o.value, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the Option is None, it will return the default value.
|
||||||
|
func (o *option[T]) ValueOr(def T) T {
|
||||||
|
if o.IsNone() {
|
||||||
|
return def
|
||||||
|
}
|
||||||
|
return o.value
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the Option is None, it will panic.
|
||||||
|
func (o *option[T]) ValueOrPanic() T {
|
||||||
|
if o.IsNone() {
|
||||||
|
panic("Option is None")
|
||||||
|
}
|
||||||
|
return o.value
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *option[T]) SetValue(value T) {
|
||||||
|
o.value = value
|
||||||
|
o.none = false
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := json.Unmarshal(data, &o.value); err != nil {
|
||||||
|
o.SetNone()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
o.none = false
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewOption[T NonNilT](value T) *option[T] {
|
||||||
|
return &option[T]{value: value}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNoneOption[T NonNilT]() *option[T] {
|
||||||
|
return &option[T]{none: true}
|
||||||
|
}
|
@ -1,31 +0,0 @@
|
|||||||
package utils
|
|
||||||
|
|
||||||
import "time"
|
|
||||||
|
|
||||||
func MinDuration(durations ...time.Duration) time.Duration {
|
|
||||||
min := durations[0]
|
|
||||||
for _, d := range durations {
|
|
||||||
if d < min {
|
|
||||||
min = d
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return min
|
|
||||||
}
|
|
||||||
|
|
||||||
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))
|
|
||||||
}
|
|
@ -1,71 +0,0 @@
|
|||||||
package utils
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
type NonNilConcrete interface {
|
|
||||||
~int | ~float64 | ~string | ~bool
|
|
||||||
}
|
|
||||||
|
|
||||||
type Option[T NonNilConcrete] 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 NonNilConcrete] struct {
|
|
||||||
// value holds the actual value of the Option if it is not None.
|
|
||||||
value T
|
|
||||||
// none indicates whether the Option is None (i.e., has no value).
|
|
||||||
none bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (o *option[T]) IsNone() bool {
|
|
||||||
return o.none
|
|
||||||
}
|
|
||||||
|
|
||||||
// The returned value can be nil, if the Option is None, it will return nil and an error.
|
|
||||||
func (o *option[T]) ValueOrErr() (*T, error) {
|
|
||||||
if o.IsNone() {
|
|
||||||
return nil, errors.New("Option is None")
|
|
||||||
}
|
|
||||||
return &o.value, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// The returned value can't be nil, if the Option is None, it will return the default value.
|
|
||||||
func (o *option[T]) ValueOr(def *T) *T {
|
|
||||||
if o.IsNone() {
|
|
||||||
return def
|
|
||||||
}
|
|
||||||
return &o.value
|
|
||||||
}
|
|
||||||
|
|
||||||
// The returned value can't be nil, if the Option is None, it will panic.
|
|
||||||
func (o *option[T]) ValueOrPanic() *T {
|
|
||||||
if o.IsNone() {
|
|
||||||
panic("Option is None")
|
|
||||||
}
|
|
||||||
return &o.value
|
|
||||||
}
|
|
||||||
|
|
||||||
func (o *option[T]) UnmarshalJSON(data []byte) error {
|
|
||||||
if string(data) == "null" {
|
|
||||||
o.none = true
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
o.none = false
|
|
||||||
return json.Unmarshal(data, &o.value)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewOption[T NonNilConcrete](value T) *option[T] {
|
|
||||||
return &option[T]{value: value}
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewNoneOption[T NonNilConcrete]() *option[T] {
|
|
||||||
return &option[T]{none: true}
|
|
||||||
}
|
|
Reference in New Issue
Block a user