mirror of
https://github.com/Luzifer/webcheck.git
synced 2024-10-18 05:14:24 +00:00
Compare commits
6 commits
019b23873b
...
a8193f3845
Author | SHA1 | Date | |
---|---|---|---|
a8193f3845 | |||
38ba058241 | |||
213eefeef5 | |||
3cec575f37 | |||
dde2fca631 | |||
dcee558b80 |
11 changed files with 523 additions and 178 deletions
55
.github/workflows/test-and-build.yml
vendored
Normal file
55
.github/workflows/test-and-build.yml
vendored
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
---
|
||||||
|
|
||||||
|
name: test-and-build
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: ['*']
|
||||||
|
tags: ['v*']
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: write
|
||||||
|
issues: write
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
test-and-build:
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
shell: bash
|
||||||
|
|
||||||
|
container:
|
||||||
|
image: luzifer/gh-arch-env
|
||||||
|
env:
|
||||||
|
CGO_ENABLED: 0
|
||||||
|
GOPATH: /go
|
||||||
|
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
show-progress: false
|
||||||
|
|
||||||
|
- name: Marking workdir safe
|
||||||
|
run: git config --global --add safe.directory /__w/webcheck/webcheck
|
||||||
|
|
||||||
|
- name: 'Lint and test code'
|
||||||
|
run: |
|
||||||
|
go test -cover -v ./...
|
||||||
|
golangci-lint run ./...
|
||||||
|
|
||||||
|
- name: Build release
|
||||||
|
run: make publish
|
||||||
|
|
||||||
|
- name: Extract changelog
|
||||||
|
run: 'awk "/^#/ && ++c==2{exit}; /^#/f" "History.md" | tail -n +2 >release_changelog.md'
|
||||||
|
|
||||||
|
- name: Release
|
||||||
|
uses: ncipollo/release-action@v1
|
||||||
|
if: startsWith(github.ref, 'refs/tags/')
|
||||||
|
with:
|
||||||
|
artifacts: '.build/*'
|
||||||
|
bodyFile: release_changelog.md
|
||||||
|
draft: false
|
||||||
|
generateReleaseNotes: false
|
||||||
|
|
||||||
|
...
|
174
.golangci.yml
Normal file
174
.golangci.yml
Normal file
|
@ -0,0 +1,174 @@
|
||||||
|
# Derived from https://github.com/golangci/golangci-lint/blob/master/.golangci.example.yml
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
run:
|
||||||
|
# timeout for analysis, e.g. 30s, 5m, default is 1m
|
||||||
|
timeout: 5m
|
||||||
|
# Force readonly modules usage for checking
|
||||||
|
modules-download-mode: readonly
|
||||||
|
|
||||||
|
output:
|
||||||
|
formats:
|
||||||
|
- format: tab
|
||||||
|
path: stdout
|
||||||
|
|
||||||
|
issues:
|
||||||
|
# This disables the included exclude-list in golangci-lint as that
|
||||||
|
# list for example fully hides G304 gosec rule, errcheck, exported
|
||||||
|
# rule of revive and other errors one really wants to see.
|
||||||
|
# Smme detail: https://github.com/golangci/golangci-lint/issues/456
|
||||||
|
exclude-use-default: false
|
||||||
|
# Don't limit the number of shown issues: Report ALL of them
|
||||||
|
max-issues-per-linter: 0
|
||||||
|
max-same-issues: 0
|
||||||
|
|
||||||
|
linters:
|
||||||
|
disable-all: true
|
||||||
|
enable:
|
||||||
|
- asciicheck # Simple linter to check that your code does not contain non-ASCII identifiers [fast: true, auto-fix: false]
|
||||||
|
- bidichk # Checks for dangerous unicode character sequences [fast: true, auto-fix: false]
|
||||||
|
- bodyclose # checks whether HTTP response body is closed successfully [fast: true, auto-fix: false]
|
||||||
|
- containedctx # containedctx is a linter that detects struct contained context.Context field [fast: true, auto-fix: false]
|
||||||
|
- contextcheck # check the function whether use a non-inherited context [fast: false, auto-fix: false]
|
||||||
|
- dogsled # Checks assignments with too many blank identifiers (e.g. x, _, _, _, := f()) [fast: true, auto-fix: false]
|
||||||
|
- durationcheck # check for two durations multiplied together [fast: false, auto-fix: false]
|
||||||
|
- errcheck # Errcheck is a program for checking for unchecked errors in go programs. These unchecked errors can be critical bugs in some cases [fast: false, auto-fix: false]
|
||||||
|
- errchkjson # Checks types passed to the json encoding functions. Reports unsupported types and optionally reports occations, where the check for the returned error can be omitted. [fast: false, auto-fix: false]
|
||||||
|
- exportloopref # checks for pointers to enclosing loop variables [fast: true, auto-fix: false]
|
||||||
|
- forbidigo # Forbids identifiers [fast: true, auto-fix: false]
|
||||||
|
- funlen # Tool for detection of long functions [fast: true, auto-fix: false]
|
||||||
|
- gocognit # Computes and checks the cognitive complexity of functions [fast: true, auto-fix: false]
|
||||||
|
- goconst # Finds repeated strings that could be replaced by a constant [fast: true, auto-fix: false]
|
||||||
|
- gocritic # The most opinionated Go source code linter [fast: true, auto-fix: false]
|
||||||
|
- gocyclo # Computes and checks the cyclomatic complexity of functions [fast: true, auto-fix: false]
|
||||||
|
- godox # Tool for detection of FIXME, TODO and other comment keywords [fast: true, auto-fix: false]
|
||||||
|
- gofmt # Gofmt checks whether code was gofmt-ed. By default this tool runs with -s option to check for code simplification [fast: true, auto-fix: true]
|
||||||
|
- gofumpt # Gofumpt checks whether code was gofumpt-ed. [fast: true, auto-fix: true]
|
||||||
|
- goimports # Goimports does everything that gofmt does. Additionally it checks unused imports [fast: true, auto-fix: true]
|
||||||
|
- gosec # Inspects source code for security problems [fast: true, auto-fix: false]
|
||||||
|
- gosimple # Linter for Go source code that specializes in simplifying a code [fast: true, auto-fix: false]
|
||||||
|
- govet # Vet examines Go source code and reports suspicious constructs, such as Printf calls whose arguments do not align with the format string [fast: true, auto-fix: false]
|
||||||
|
- ineffassign # Detects when assignments to existing variables are not used [fast: true, auto-fix: false]
|
||||||
|
- misspell # Finds commonly misspelled English words in comments [fast: true, auto-fix: true]
|
||||||
|
- mnd # An analyzer to detect magic numbers. [fast: true, auto-fix: false]
|
||||||
|
- nakedret # Finds naked returns in functions greater than a specified function length [fast: true, auto-fix: false]
|
||||||
|
- nilerr # Finds the code that returns nil even if it checks that the error is not nil. [fast: false, auto-fix: false]
|
||||||
|
- nilnil # Checks that there is no simultaneous return of `nil` error and an invalid value. [fast: false, auto-fix: false]
|
||||||
|
- noctx # noctx finds sending http request without context.Context [fast: true, auto-fix: false]
|
||||||
|
- nolintlint # Reports ill-formed or insufficient nolint directives [fast: true, auto-fix: false]
|
||||||
|
- revive # Fast, configurable, extensible, flexible, and beautiful linter for Go. Drop-in replacement of golint. [fast: false, auto-fix: false]
|
||||||
|
- staticcheck # Staticcheck is a go vet on steroids, applying a ton of static analysis checks [fast: true, auto-fix: false]
|
||||||
|
- stylecheck # Stylecheck is a replacement for golint [fast: true, auto-fix: false]
|
||||||
|
- tenv # tenv is analyzer that detects using os.Setenv instead of t.Setenv since Go1.17 [fast: false, auto-fix: false]
|
||||||
|
- typecheck # Like the front-end of a Go compiler, parses and type-checks Go code [fast: true, auto-fix: false]
|
||||||
|
- unconvert # Remove unnecessary type conversions [fast: true, auto-fix: false]
|
||||||
|
- unused # Checks Go code for unused constants, variables, functions and types [fast: false, auto-fix: false]
|
||||||
|
- wastedassign # wastedassign finds wasted assignment statements. [fast: false, auto-fix: false]
|
||||||
|
- wrapcheck # Checks that errors returned from external packages are wrapped [fast: false, auto-fix: false]
|
||||||
|
|
||||||
|
linters-settings:
|
||||||
|
funlen:
|
||||||
|
lines: 100
|
||||||
|
statements: 60
|
||||||
|
|
||||||
|
gocyclo:
|
||||||
|
# minimal code complexity to report, 30 by default (but we recommend 10-20)
|
||||||
|
min-complexity: 15
|
||||||
|
|
||||||
|
gomnd:
|
||||||
|
ignored-functions: 'strconv.(?:Format|Parse)\B+'
|
||||||
|
|
||||||
|
revive:
|
||||||
|
rules:
|
||||||
|
#- name: add-constant # Suggests using constant for magic numbers and string literals
|
||||||
|
# Opinion: Makes sense for strings, not for numbers but checks numbers
|
||||||
|
#- name: argument-limit # Specifies the maximum number of arguments a function can receive | Opinion: Don't need this
|
||||||
|
- name: atomic # Check for common mistaken usages of the `sync/atomic` package
|
||||||
|
- name: banned-characters # Checks banned characters in identifiers
|
||||||
|
arguments:
|
||||||
|
- ';' # Greek question mark
|
||||||
|
- name: bare-return # Warns on bare returns
|
||||||
|
- name: blank-imports # Disallows blank imports
|
||||||
|
- name: bool-literal-in-expr # Suggests removing Boolean literals from logic expressions
|
||||||
|
- name: call-to-gc # Warns on explicit call to the garbage collector
|
||||||
|
#- name: cognitive-complexity # Sets restriction for maximum Cognitive complexity.
|
||||||
|
# There is a dedicated linter for this
|
||||||
|
- name: confusing-naming # Warns on methods with names that differ only by capitalization
|
||||||
|
- name: confusing-results # Suggests to name potentially confusing function results
|
||||||
|
- name: constant-logical-expr # Warns on constant logical expressions
|
||||||
|
- name: context-as-argument # `context.Context` should be the first argument of a function.
|
||||||
|
- name: context-keys-type # Disallows the usage of basic types in `context.WithValue`.
|
||||||
|
#- name: cyclomatic # Sets restriction for maximum Cyclomatic complexity.
|
||||||
|
# There is a dedicated linter for this
|
||||||
|
#- name: datarace # Spots potential dataraces
|
||||||
|
# Is not (yet) available?
|
||||||
|
- name: deep-exit # Looks for program exits in funcs other than `main()` or `init()`
|
||||||
|
- name: defer # Warns on some [defer gotchas](https://blog.learngoprogramming.com/5-gotchas-of-defer-in-go-golang-part-iii-36a1ab3d6ef1)
|
||||||
|
- name: dot-imports # Forbids `.` imports.
|
||||||
|
- name: duplicated-imports # Looks for packages that are imported two or more times
|
||||||
|
- name: early-return # Spots if-then-else statements that can be refactored to simplify code reading
|
||||||
|
- name: empty-block # Warns on empty code blocks
|
||||||
|
- name: empty-lines # Warns when there are heading or trailing newlines in a block
|
||||||
|
- name: errorf # Should replace `errors.New(fmt.Sprintf())` with `fmt.Errorf()`
|
||||||
|
- name: error-naming # Naming of error variables.
|
||||||
|
- name: error-return # The error return parameter should be last.
|
||||||
|
- name: error-strings # Conventions around error strings.
|
||||||
|
- name: exported # Naming and commenting conventions on exported symbols.
|
||||||
|
arguments: ['sayRepetitiveInsteadOfStutters']
|
||||||
|
#- name: file-header # Header which each file should have.
|
||||||
|
# Useless without config, have no config for it
|
||||||
|
- name: flag-parameter # Warns on boolean parameters that create a control coupling
|
||||||
|
#- name: function-length # Warns on functions exceeding the statements or lines max
|
||||||
|
# There is a dedicated linter for this
|
||||||
|
#- name: function-result-limit # Specifies the maximum number of results a function can return
|
||||||
|
# Opinion: Don't need this
|
||||||
|
- name: get-return # Warns on getters that do not yield any result
|
||||||
|
- name: identical-branches # Spots if-then-else statements with identical `then` and `else` branches
|
||||||
|
- name: if-return # Redundant if when returning an error.
|
||||||
|
#- name: imports-blacklist # Disallows importing the specified packages
|
||||||
|
# Useless without config, have no config for it
|
||||||
|
- name: import-shadowing # Spots identifiers that shadow an import
|
||||||
|
- name: increment-decrement # Use `i++` and `i--` instead of `i += 1` and `i -= 1`.
|
||||||
|
- name: indent-error-flow # Prevents redundant else statements.
|
||||||
|
#- name: line-length-limit # Specifies the maximum number of characters in a lined
|
||||||
|
# There is a dedicated linter for this
|
||||||
|
#- name: max-public-structs # The maximum number of public structs in a file.
|
||||||
|
# Opinion: Don't need this
|
||||||
|
- name: modifies-parameter # Warns on assignments to function parameters
|
||||||
|
- name: modifies-value-receiver # Warns on assignments to value-passed method receivers
|
||||||
|
#- name: nested-structs # Warns on structs within structs
|
||||||
|
# Opinion: Don't need this
|
||||||
|
- name: optimize-operands-order # Checks inefficient conditional expressions
|
||||||
|
#- name: package-comments # Package commenting conventions.
|
||||||
|
# Opinion: Don't need this
|
||||||
|
- name: range # Prevents redundant variables when iterating over a collection.
|
||||||
|
- name: range-val-address # Warns if address of range value is used dangerously
|
||||||
|
- name: range-val-in-closure # Warns if range value is used in a closure dispatched as goroutine
|
||||||
|
- name: receiver-naming # Conventions around the naming of receivers.
|
||||||
|
- name: redefines-builtin-id # Warns on redefinitions of builtin identifiers
|
||||||
|
#- name: string-format # Warns on specific string literals that fail one or more user-configured regular expressions
|
||||||
|
# Useless without config, have no config for it
|
||||||
|
- name: string-of-int # Warns on suspicious casts from int to string
|
||||||
|
- name: struct-tag # Checks common struct tags like `json`,`xml`,`yaml`
|
||||||
|
- name: superfluous-else # Prevents redundant else statements (extends indent-error-flow)
|
||||||
|
- name: time-equal # Suggests to use `time.Time.Equal` instead of `==` and `!=` for equality check time.
|
||||||
|
- name: time-naming # Conventions around the naming of time variables.
|
||||||
|
- name: unconditional-recursion # Warns on function calls that will lead to (direct) infinite recursion
|
||||||
|
- name: unexported-naming # Warns on wrongly named un-exported symbols
|
||||||
|
- name: unexported-return # Warns when a public return is from unexported type.
|
||||||
|
- name: unhandled-error # Warns on unhandled errors returned by funcion calls
|
||||||
|
arguments:
|
||||||
|
- "fmt.(Fp|P)rint(f|ln|)"
|
||||||
|
- name: unnecessary-stmt # Suggests removing or simplifying unnecessary statements
|
||||||
|
- name: unreachable-code # Warns on unreachable code
|
||||||
|
- name: unused-parameter # Suggests to rename or remove unused function parameters
|
||||||
|
- name: unused-receiver # Suggests to rename or remove unused method receivers
|
||||||
|
#- name: use-any # Proposes to replace `interface{}` with its alias `any`
|
||||||
|
# Is not (yet) available?
|
||||||
|
- name: useless-break # Warns on useless `break` statements in case clauses
|
||||||
|
- name: var-declaration # Reduces redundancies around variable declaration.
|
||||||
|
- name: var-naming # Naming rules.
|
||||||
|
- name: waitgroup-by-value # Warns on functions taking sync.WaitGroup as a by-value parameter
|
||||||
|
|
||||||
|
...
|
|
@ -1,10 +0,0 @@
|
||||||
---
|
|
||||||
|
|
||||||
image: "reporunner/golang-alpine"
|
|
||||||
checkout_dir: /go/src/github.com/Luzifer/webcheck
|
|
||||||
|
|
||||||
commands:
|
|
||||||
- make publish
|
|
||||||
|
|
||||||
environment:
|
|
||||||
CGO_ENABLED: 0
|
|
|
@ -1,3 +1,9 @@
|
||||||
|
# 0.2.3 / 2024-07-16
|
||||||
|
|
||||||
|
* Fix: Do not panic when no-log is enabled
|
||||||
|
* Deps: Update dependencies
|
||||||
|
* Lint: Resolve linter errors, improve code
|
||||||
|
|
||||||
# 0.2.2 / 2020-07-21
|
# 0.2.2 / 2020-07-21
|
||||||
|
|
||||||
* Update dependencies
|
* Update dependencies
|
||||||
|
|
3
Makefile
3
Makefile
|
@ -1,3 +1,2 @@
|
||||||
publish:
|
publish:
|
||||||
curl -sSLo golang.sh https://raw.githubusercontent.com/Luzifer/github-publish/master/golang.sh
|
bash ./ci/build.sh
|
||||||
bash golang.sh
|
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
[![Go Report Card](https://goreportcard.com/badge/github.com/Luzifer/webcheck)](https://goreportcard.com/report/github.com/Luzifer/webcheck)
|
|
||||||
![](https://badges.fyi/github/license/Luzifer/webcheck)
|
![](https://badges.fyi/github/license/Luzifer/webcheck)
|
||||||
![](https://badges.fyi/github/downloads/Luzifer/webcheck)
|
![](https://badges.fyi/github/downloads/Luzifer/webcheck)
|
||||||
![](https://badges.fyi/github/latest-release/Luzifer/webcheck)
|
![](https://badges.fyi/github/latest-release/Luzifer/webcheck)
|
||||||
|
@ -18,7 +17,7 @@ If the request is marked as a `FAIL` all headers and the response body are writt
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
You either can download a pre-compiled binary for you system from the [Releases](https://github.com/Luzifer/webcheck/releases) section of this repository or if you do have a working Go environment you can just `go get github.com/Luzifer/webcheck` the tool.
|
You either can download a pre-compiled binary for you system from the [Releases](https://github.com/Luzifer/webcheck/releases) section of this repository or if you do have a working Go environment you can just `go install github.com/Luzifer/webcheck@latest` the tool.
|
||||||
|
|
||||||
## Usage
|
## Usage
|
||||||
|
|
||||||
|
@ -27,6 +26,7 @@ $ webcheck --help
|
||||||
Usage of webcheck:
|
Usage of webcheck:
|
||||||
-i, --interval duration Check interval (default 1s)
|
-i, --interval duration Check interval (default 1s)
|
||||||
-l, --log-dir string Directory to log non-matched requests to (default "/tmp/resp-log/")
|
-l, --log-dir string Directory to log non-matched requests to (default "/tmp/resp-log/")
|
||||||
|
--log-level string Log level (debug, info, warn, error, fatal) (default "info")
|
||||||
--log-retention duration When to clean up file from log-dir (default 24h0m0s)
|
--log-retention duration When to clean up file from log-dir (default 24h0m0s)
|
||||||
-m, --match string RegExp to match the response body against to validate it (default ".*")
|
-m, --match string RegExp to match the response body against to validate it (default ".*")
|
||||||
--no-log Disable response body logging
|
--no-log Disable response body logging
|
||||||
|
|
63
ci/build.sh
Normal file
63
ci/build.sh
Normal file
|
@ -0,0 +1,63 @@
|
||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
osarch=(
|
||||||
|
darwin/amd64
|
||||||
|
darwin/arm64
|
||||||
|
linux/amd64
|
||||||
|
linux/arm
|
||||||
|
linux/arm64
|
||||||
|
windows/amd64
|
||||||
|
)
|
||||||
|
|
||||||
|
function go_package() {
|
||||||
|
cd "${4}"
|
||||||
|
|
||||||
|
local outname="${3}"
|
||||||
|
[[ $1 == windows ]] && outname="${3}.exe"
|
||||||
|
|
||||||
|
log "=> Building ${3} for ${1}/${2}..."
|
||||||
|
CGO_ENABLED=0 GOARCH=$2 GOOS=$1 go build \
|
||||||
|
-ldflags "-s -w -X main.version=${version}" \
|
||||||
|
-mod=readonly \
|
||||||
|
-trimpath \
|
||||||
|
-o "${outname}"
|
||||||
|
|
||||||
|
if [[ $1 == linux ]]; then
|
||||||
|
log "=> Packging ${3} as ${3}_${1}_${2}.tgz..."
|
||||||
|
tar -czf "${builddir}/${3}_${1}_${2}.tgz" "${outname}"
|
||||||
|
else
|
||||||
|
log "=> Packging ${3} as ${3}_${1}_${2}.zip..."
|
||||||
|
zip "${builddir}/${3}_${1}_${2}.zip" "${outname}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
rm "${outname}"
|
||||||
|
}
|
||||||
|
|
||||||
|
function go_package_all() {
|
||||||
|
for oa in "${osarch[@]}"; do
|
||||||
|
local os=$(cut -d / -f 1 <<<"${oa}")
|
||||||
|
local arch=$(cut -d / -f 2 <<<"${oa}")
|
||||||
|
(go_package "${os}" "${arch}" "${1}" "${2}")
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
function log() {
|
||||||
|
echo "[$(date +%H:%M:%S)] $@" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
root=$(pwd)
|
||||||
|
builddir="${root}/.build"
|
||||||
|
version="$(git describe --tags --always || echo dev)"
|
||||||
|
|
||||||
|
log "Building version ${version}..."
|
||||||
|
|
||||||
|
log "Resetting output directory..."
|
||||||
|
rm -rf "${builddir}"
|
||||||
|
mkdir -p "${builddir}"
|
||||||
|
|
||||||
|
log "Building Tool..."
|
||||||
|
go_package_all "webcheck" "."
|
||||||
|
|
||||||
|
log "Generating SHA256SUMS file..."
|
||||||
|
(cd "${builddir}" && sha256sum * | tee SHA256SUMS)
|
20
go.mod
20
go.mod
|
@ -1,14 +1,16 @@
|
||||||
module github.com/Luzifer/webcheck
|
module github.com/Luzifer/webcheck
|
||||||
|
|
||||||
go 1.14
|
go 1.22
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/Luzifer/rconfig/v2 v2.2.1
|
github.com/Luzifer/rconfig/v2 v2.5.0
|
||||||
github.com/montanaflynn/stats v0.6.3
|
github.com/montanaflynn/stats v0.7.1
|
||||||
github.com/sirupsen/logrus v1.6.0
|
github.com/sirupsen/logrus v1.9.3
|
||||||
github.com/spf13/pflag v1.0.5
|
)
|
||||||
golang.org/x/crypto v0.0.0-20200709230013-948cd5f35899
|
|
||||||
golang.org/x/sys v0.0.0-20200720211630-cb9d2d5c5666
|
require (
|
||||||
gopkg.in/validator.v2 v2.0.0-20200605151824-2b28d334fa05
|
github.com/spf13/pflag v1.0.5 // indirect
|
||||||
gopkg.in/yaml.v2 v2.3.0
|
golang.org/x/sys v0.22.0 // indirect
|
||||||
|
gopkg.in/validator.v2 v2.0.1 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
)
|
)
|
||||||
|
|
52
go.sum
52
go.sum
|
@ -1,30 +1,32 @@
|
||||||
github.com/Luzifer/rconfig/v2 v2.2.1 h1:zcDdLQlnlzwcBJ8E0WFzOkQE1pCMn3EbX0dFYkeTczg=
|
github.com/Luzifer/rconfig/v2 v2.5.0 h1:zx5lfQbNX3za4VegID97IeY+M+BmfgHxWJTYA94sxok=
|
||||||
github.com/Luzifer/rconfig/v2 v2.2.1/go.mod h1:OKIX0/JRZrPJ/ZXXWklQEFXA6tBfWaljZbW37w+sqBw=
|
github.com/Luzifer/rconfig/v2 v2.5.0/go.mod h1:eGWUPQeCPv/Pr/p0hjmwFgI20uqvwi/Szen69hUzGzU=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
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/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=
|
github.com/kr/pretty v0.2.1 h1:Fmg33tUaq4/8ym9TJN1x7sLJnHVwhP33CNkpYV/7rwI=
|
||||||
github.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
|
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||||
github.com/montanaflynn/stats v0.6.3 h1:F8446DrvIF5V5smZfZ8K9nrmmix0AFgevPdLruGOmzk=
|
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||||
github.com/montanaflynn/stats v0.6.3/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/montanaflynn/stats v0.7.1 h1:etflOAAHORrCC44V+aR6Ftzort912ZU+YLiSTuV8eaE=
|
||||||
|
github.com/montanaflynn/stats v0.7.1/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/sirupsen/logrus v1.6.0 h1:UBcNElsrwanuuMsnGSlYmtmgbb23qDR5dG+6X6Oo89I=
|
github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
|
||||||
github.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=
|
github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
|
||||||
github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
|
|
||||||
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
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/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
golang.org/x/crypto v0.0.0-20200709230013-948cd5f35899/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI=
|
||||||
golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
golang.org/x/sys v0.0.0-20200720211630-cb9d2d5c5666 h1:gVCS+QOncANNPlmlO1AhlU3oxs4V9z+gTtPwIk3p2N8=
|
|
||||||
golang.org/x/sys v0.0.0-20200720211630-cb9d2d5c5666/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
|
||||||
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/validator.v2 v2.0.0-20180514200540-135c24b11c19/go.mod h1:o4V0GXN9/CAmCsvJ0oXYZvrZOe7syiDZSN1GWGZTGzc=
|
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
|
||||||
gopkg.in/validator.v2 v2.0.0-20200605151824-2b28d334fa05 h1:l9eKDCWy9n7C5NAiQAMvDePh0vyLAweR6LcSUVXFUGg=
|
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
|
||||||
gopkg.in/validator.v2 v2.0.0-20200605151824-2b28d334fa05/go.mod h1:o4V0GXN9/CAmCsvJ0oXYZvrZOe7syiDZSN1GWGZTGzc=
|
gopkg.in/validator.v2 v2.0.1 h1:xF0KWyGWXm/LM2G1TrEjqOu4pa6coO9AlWSf3msVfDY=
|
||||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/validator.v2 v2.0.1/go.mod h1:lIUZBlB3Im4s/eYp39Ry/wkR02yOPhZ9IwIRBjuPuG8=
|
||||||
gopkg.in/yaml.v2 v2.3.0 h1:clyUAQHOM3G0M3f5vQj7LuJrETvjVot3Z5el9nffUtU=
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|
187
main.go
187
main.go
|
@ -2,27 +2,23 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
"text/template"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/montanaflynn/stats"
|
"github.com/sirupsen/logrus"
|
||||||
log "github.com/sirupsen/logrus"
|
|
||||||
|
|
||||||
"github.com/Luzifer/rconfig/v2"
|
"github.com/Luzifer/rconfig/v2"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
dateFormat = time.RFC1123
|
cleanupInterval = 10 * time.Second
|
||||||
numHistoricalDurations = 300
|
logFolderPerms = 0o750
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -30,6 +26,7 @@ var (
|
||||||
DisableLog bool `flag:"no-log" default:"false" description:"Disable response body logging"`
|
DisableLog bool `flag:"no-log" default:"false" description:"Disable response body logging"`
|
||||||
Interval time.Duration `flag:"interval,i" default:"1s" description:"Check interval"`
|
Interval time.Duration `flag:"interval,i" default:"1s" description:"Check interval"`
|
||||||
LogDir string `flag:"log-dir,l" default:"/tmp/resp-log/" description:"Directory to log non-matched requests to"`
|
LogDir string `flag:"log-dir,l" default:"/tmp/resp-log/" description:"Directory to log non-matched requests to"`
|
||||||
|
LogLevel string `flag:"log-level" default:"info" description:"Log level (debug, info, warn, error, fatal)"`
|
||||||
LogRetention time.Duration `flag:"log-retention" default:"24h" description:"When to clean up file from log-dir"`
|
LogRetention time.Duration `flag:"log-retention" default:"24h" description:"When to clean up file from log-dir"`
|
||||||
Match string `flag:"match,m" default:".*" description:"RegExp to match the response body against to validate it"`
|
Match string `flag:"match,m" default:".*" description:"RegExp to match the response body against to validate it"`
|
||||||
Timeout time.Duration `flag:"timeout,t" default:"30s" description:"Timeout for the request"`
|
Timeout time.Duration `flag:"timeout,t" default:"30s" description:"Timeout for the request"`
|
||||||
|
@ -40,120 +37,35 @@ var (
|
||||||
version = "dev"
|
version = "dev"
|
||||||
)
|
)
|
||||||
|
|
||||||
type checkStatus uint
|
func initApp() error {
|
||||||
|
|
||||||
func (c checkStatus) String() string {
|
|
||||||
return map[checkStatus]string{
|
|
||||||
statusUnknown: "UNKN",
|
|
||||||
statusFailed: "FAIL",
|
|
||||||
statusOk: "OKAY",
|
|
||||||
}[c]
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
statusUnknown checkStatus = iota
|
|
||||||
statusOk
|
|
||||||
statusFailed
|
|
||||||
)
|
|
||||||
|
|
||||||
type checkResult struct {
|
|
||||||
DumpFile string
|
|
||||||
Durations *ringDuration
|
|
||||||
Message string
|
|
||||||
Start time.Time
|
|
||||||
Status checkStatus
|
|
||||||
|
|
||||||
lock sync.RWMutex
|
|
||||||
lastLineLen int
|
|
||||||
}
|
|
||||||
|
|
||||||
func newCheckResult(status checkStatus, message string, duration time.Duration) *checkResult {
|
|
||||||
r := newRingDuration(numHistoricalDurations)
|
|
||||||
r.SetNext(duration)
|
|
||||||
|
|
||||||
return &checkResult{
|
|
||||||
Durations: r,
|
|
||||||
Message: message,
|
|
||||||
Start: time.Now(),
|
|
||||||
Status: status,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *checkResult) AddDuration(d time.Duration) {
|
|
||||||
c.lock.Lock()
|
|
||||||
defer c.lock.Unlock()
|
|
||||||
|
|
||||||
c.Durations.SetNext(d)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *checkResult) DurationStats() string {
|
|
||||||
c.lock.RLock()
|
|
||||||
defer c.lock.RUnlock()
|
|
||||||
|
|
||||||
var (
|
|
||||||
s = stats.LoadRawData(c.Durations.GetAll())
|
|
||||||
min, avg, max float64
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
if min, err = s.Min(); err != nil {
|
|
||||||
min = 0
|
|
||||||
}
|
|
||||||
if avg, err = s.Median(); err != nil {
|
|
||||||
avg = 0
|
|
||||||
}
|
|
||||||
if max, err = s.Max(); err != nil {
|
|
||||||
max = 0
|
|
||||||
}
|
|
||||||
|
|
||||||
return fmt.Sprintf("%s/%s/%s",
|
|
||||||
time.Duration(min).Round(time.Microsecond).String(),
|
|
||||||
time.Duration(avg).Round(time.Microsecond).String(),
|
|
||||||
time.Duration(max).Round(time.Microsecond).String(),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *checkResult) Equals(r *checkResult) bool {
|
|
||||||
return c.Status == r.Status && c.Message == r.Message
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *checkResult) Print() {
|
|
||||||
tpl := strings.Join([]string{
|
|
||||||
`[{{ .Start.Format "` + dateFormat + `" }}]`,
|
|
||||||
`({{ .Status }})`,
|
|
||||||
`{{ .Message }}`,
|
|
||||||
`({{ .DurationStats }})`,
|
|
||||||
`{{ if ne .DumpFile "" }}(Resp: {{ .DumpFile }}){{ end }}`,
|
|
||||||
}, " ")
|
|
||||||
templ := template.Must(template.New("result").Parse(tpl))
|
|
||||||
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
templ.Execute(buf, c)
|
|
||||||
|
|
||||||
if c.lastLineLen > 0 {
|
|
||||||
fmt.Fprintf(os.Stdout, "\r%s\r", strings.Repeat(" ", c.lastLineLen))
|
|
||||||
}
|
|
||||||
|
|
||||||
c.lastLineLen = buf.Len()
|
|
||||||
buf.WriteTo(os.Stdout)
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
rconfig.AutoEnv(true)
|
rconfig.AutoEnv(true)
|
||||||
if err := rconfig.ParseAndValidate(&cfg); err != nil {
|
if err := rconfig.ParseAndValidate(&cfg); err != nil {
|
||||||
log.Fatalf("Unable to parse commandline options: %s", err)
|
return fmt.Errorf("parsing cli options: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if cfg.VersionAndExit {
|
l, err := logrus.ParseLevel(cfg.LogLevel)
|
||||||
fmt.Printf("webcheck %s\n", version)
|
if err != nil {
|
||||||
os.Exit(0)
|
return fmt.Errorf("parsing log-level: %w", err)
|
||||||
}
|
}
|
||||||
|
logrus.SetLevel(l)
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
http.DefaultClient.Timeout = cfg.Timeout
|
var err error
|
||||||
|
if err = initApp(); err != nil {
|
||||||
|
logrus.WithError(err).Fatal("initializing app")
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.VersionAndExit {
|
||||||
|
logrus.WithField("version", version).Info("webcheck")
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
|
||||||
matcher, err := regexp.Compile(cfg.Match)
|
matcher, err := regexp.Compile(cfg.Match)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.WithError(err).Fatal("Matcher regexp does not compile")
|
logrus.WithError(err).Fatal("compiling matcher RegExp")
|
||||||
}
|
}
|
||||||
|
|
||||||
lastResult := newCheckResult(statusUnknown, "Uninitialized", 0)
|
lastResult := newCheckResult(statusUnknown, "Uninitialized", 0)
|
||||||
|
@ -162,7 +74,7 @@ func main() {
|
||||||
|
|
||||||
for range time.Tick(cfg.Interval) {
|
for range time.Tick(cfg.Interval) {
|
||||||
var (
|
var (
|
||||||
body *bytes.Buffer
|
body io.ReadWriter
|
||||||
result *checkResult
|
result *checkResult
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -173,13 +85,13 @@ func main() {
|
||||||
result = doCheck(cfg.URL, matcher, body)
|
result = doCheck(cfg.URL, matcher, body)
|
||||||
|
|
||||||
if !result.Equals(lastResult) {
|
if !result.Equals(lastResult) {
|
||||||
fmt.Println()
|
fmt.Println() //nolint:forbidigo
|
||||||
lastResult = result
|
lastResult = result
|
||||||
|
|
||||||
if result.Status == statusFailed {
|
if result.Status == statusFailed {
|
||||||
fn, err := dumpRequest(body)
|
fn, err := dumpRequest(body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.WithError(err).Fatal("Could not dump request")
|
logrus.WithError(err).Fatal("logging request")
|
||||||
}
|
}
|
||||||
lastResult.DumpFile = fn
|
lastResult.DumpFile = fn
|
||||||
}
|
}
|
||||||
|
@ -187,12 +99,14 @@ func main() {
|
||||||
lastResult.AddDuration(result.Durations.GetCurrent())
|
lastResult.AddDuration(result.Durations.GetCurrent())
|
||||||
}
|
}
|
||||||
|
|
||||||
lastResult.Print()
|
if err = lastResult.Print(); err != nil {
|
||||||
|
logrus.WithError(err).Fatal("displaying status")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func cleanupLogFiles() {
|
func cleanupLogFiles() {
|
||||||
for range time.Tick(10 * time.Second) {
|
for range time.Tick(cleanupInterval) {
|
||||||
if info, err := os.Stat(cfg.LogDir); err != nil || !info.IsDir() {
|
if info, err := os.Stat(cfg.LogDir); err != nil || !info.IsDir() {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
@ -208,14 +122,17 @@ func cleanupLogFiles() {
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
fmt.Println()
|
fmt.Println() //nolint:forbidigo
|
||||||
log.WithError(err).Error("Could not clean up logs")
|
logrus.WithError(err).Error("cleaning up logs")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func doCheck(url string, match *regexp.Regexp, responseBody io.Writer) *checkResult {
|
func doCheck(url string, match *regexp.Regexp, responseBody io.Writer) *checkResult {
|
||||||
req, _ := http.NewRequest("GET", url, nil)
|
ctx, cancel := context.WithTimeout(context.Background(), cfg.Timeout)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
req, _ := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||||
|
|
||||||
respStart := time.Now()
|
respStart := time.Now()
|
||||||
resp, err := http.DefaultClient.Do(req)
|
resp, err := http.DefaultClient.Do(req)
|
||||||
|
@ -227,7 +144,11 @@ func doCheck(url string, match *regexp.Regexp, responseBody io.Writer) *checkRes
|
||||||
respDuration,
|
respDuration,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer func() {
|
||||||
|
if err := resp.Body.Close(); err != nil {
|
||||||
|
logrus.WithError(err).Error("closing response body (leaked fd)")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
body := new(bytes.Buffer)
|
body := new(bytes.Buffer)
|
||||||
if _, err = io.Copy(body, resp.Body); err != nil {
|
if _, err = io.Copy(body, resp.Body); err != nil {
|
||||||
|
@ -246,8 +167,8 @@ func doCheck(url string, match *regexp.Regexp, responseBody io.Writer) *checkRes
|
||||||
respDuration,
|
respDuration,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
fmt.Fprintln(responseBody)
|
|
||||||
if _, err = responseBody.Write(body.Bytes()); err != nil {
|
if _, err = responseBody.Write(append([]byte{'\n'}, body.Bytes()...)); err != nil {
|
||||||
return newCheckResult(
|
return newCheckResult(
|
||||||
statusFailed,
|
statusFailed,
|
||||||
"Was not able to copy body",
|
"Was not able to copy body",
|
||||||
|
@ -284,17 +205,23 @@ func dumpRequest(body io.Reader) (string, error) {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := os.MkdirAll(cfg.LogDir, 0755); err != nil {
|
if err := os.MkdirAll(cfg.LogDir, logFolderPerms); err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("creating log folder: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
f, err := ioutil.TempFile(cfg.LogDir, "resp")
|
f, err := os.CreateTemp(cfg.LogDir, "resp")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", fmt.Errorf("creating log file: %w", err)
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer func() {
|
||||||
|
if err := f.Close(); err != nil {
|
||||||
|
logrus.WithError(err).Error("closing log file (leaked fd)")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
_, err = io.Copy(f, body)
|
if _, err = io.Copy(f, body); err != nil {
|
||||||
|
return "", fmt.Errorf("copying request body: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
return f.Name(), err
|
return f.Name(), nil
|
||||||
}
|
}
|
||||||
|
|
127
types.go
Normal file
127
types.go
Normal file
|
@ -0,0 +1,127 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/montanaflynn/stats"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
dateFormat = time.RFC1123
|
||||||
|
numHistoricalDurations = 300
|
||||||
|
|
||||||
|
statusTemplateStr = `[{{ .Start.Format "Mon, 02 Jan 2006 15:04:05 MST" }}] ({{ .Status }}) {{ .Message }} ({{ .DurationStats }}){{ if ne .DumpFile "" }} (Resp: {{ .DumpFile }}){{ end }}`
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
statusUnknown checkStatus = iota
|
||||||
|
statusOk
|
||||||
|
statusFailed
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
checkResult struct {
|
||||||
|
DumpFile string
|
||||||
|
Durations *ringDuration
|
||||||
|
Message string
|
||||||
|
Start time.Time
|
||||||
|
Status checkStatus
|
||||||
|
|
||||||
|
lock sync.RWMutex
|
||||||
|
lastLineLen int
|
||||||
|
}
|
||||||
|
|
||||||
|
checkStatus uint
|
||||||
|
)
|
||||||
|
|
||||||
|
var statusTemplate *template.Template
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
var err error
|
||||||
|
if statusTemplate, err = template.New("statusTemplate").Parse(statusTemplateStr); err != nil {
|
||||||
|
panic(fmt.Errorf("parsing status template: %w", err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func newCheckResult(status checkStatus, message string, duration time.Duration) *checkResult {
|
||||||
|
r := newRingDuration(numHistoricalDurations)
|
||||||
|
r.SetNext(duration)
|
||||||
|
|
||||||
|
return &checkResult{
|
||||||
|
Durations: r,
|
||||||
|
Message: message,
|
||||||
|
Start: time.Now(),
|
||||||
|
Status: status,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *checkResult) AddDuration(d time.Duration) {
|
||||||
|
c.lock.Lock()
|
||||||
|
defer c.lock.Unlock()
|
||||||
|
|
||||||
|
c.Durations.SetNext(d)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *checkResult) DurationStats() string {
|
||||||
|
c.lock.RLock()
|
||||||
|
defer c.lock.RUnlock()
|
||||||
|
|
||||||
|
var (
|
||||||
|
s = stats.LoadRawData(c.Durations.GetAll())
|
||||||
|
min, avg, max float64
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
if min, err = s.Min(); err != nil {
|
||||||
|
min = 0
|
||||||
|
}
|
||||||
|
if avg, err = s.Median(); err != nil {
|
||||||
|
avg = 0
|
||||||
|
}
|
||||||
|
if max, err = s.Max(); err != nil {
|
||||||
|
max = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("%s/%s/%s",
|
||||||
|
time.Duration(min).Round(time.Microsecond).String(),
|
||||||
|
time.Duration(avg).Round(time.Microsecond).String(),
|
||||||
|
time.Duration(max).Round(time.Microsecond).String(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *checkResult) Equals(r *checkResult) bool {
|
||||||
|
return c.Status == r.Status && c.Message == r.Message
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *checkResult) Print() (err error) {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
if err = statusTemplate.Execute(buf, c); err != nil {
|
||||||
|
return fmt.Errorf("executing template: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.lastLineLen > 0 {
|
||||||
|
if _, err = fmt.Fprintf(os.Stdout, "\r%s\r", strings.Repeat(" ", c.lastLineLen)); err != nil {
|
||||||
|
return fmt.Errorf("clearing status: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c.lastLineLen = buf.Len()
|
||||||
|
if _, err = buf.WriteTo(os.Stdout); err != nil {
|
||||||
|
return fmt.Errorf("printing status: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c checkStatus) String() string {
|
||||||
|
return map[checkStatus]string{
|
||||||
|
statusUnknown: "UNKN",
|
||||||
|
statusFailed: "FAIL",
|
||||||
|
statusOk: "OKAY",
|
||||||
|
}[c]
|
||||||
|
}
|
Loading…
Reference in a new issue