aboutsummaryrefslogtreecommitdiff
path: root/attack_result.go
diff options
context:
space:
mode:
authorShulhan <ms@kilabit.info>2021-03-21 18:34:30 +0700
committerShulhan <ms@kilabit.info>2021-03-21 18:34:30 +0700
commit5828835bcd18eff070789fd82401480c244bbc27 (patch)
tree0b030e1f037684a9f5e81d62d8db52dd6ce69532 /attack_result.go
parentab00b51a1353c7f17697e027061b4c20509a7a80 (diff)
downloadgorankusu-5828835bcd18eff070789fd82401480c244bbc27.tar.xz
all: load pass attack results and implement function to get attack result
When the service started, it will load all previous attack results from directory Environment.ResultsDir. It will only scan the file name and append it to HttpTarget.Results due to the size and time to load one of them can take time. Through the web interface, user can click "Show" button to load the result and display it on the screen.
Diffstat (limited to 'attack_result.go')
-rw-r--r--attack_result.go30
1 files changed, 16 insertions, 14 deletions
diff --git a/attack_result.go b/attack_result.go
index 1f93126..112f2ca 100644
--- a/attack_result.go
+++ b/attack_result.go
@@ -33,10 +33,10 @@ const (
type AttackResult struct {
mtx sync.Mutex
- TargetID string // TargetID the ID of HTTP target which own the result.
- Name string // Name of output file without path.
- TextReport []byte // TextReport the result reported as text.
- HistReport []byte // HistReport the result reported as histogram text.
+ HttpTargetID string // ID of HTTP target which own the result.
+ Name string // Name of output file without path.
+ TextReport []byte // TextReport the result reported as text.
+ HistReport []byte // HistReport the result reported as histogram text.
fullpath string
fout *os.File
@@ -52,9 +52,9 @@ func newAttackResult(env *Environment, rr *RunRequest) (
ar *AttackResult, err error,
) {
ar = &AttackResult{
- TargetID: rr.HttpTarget.ID,
- metrics: &vegeta.Metrics{},
- hist: &vegeta.Histogram{},
+ HttpTargetID: rr.HttpTarget.ID,
+ metrics: &vegeta.Metrics{},
+ hist: &vegeta.Histogram{},
}
ar.Name = fmt.Sprintf("%s.%s.%s.%dx%s.%s.bin",
@@ -102,14 +102,14 @@ func (ar *AttackResult) cancel() {
if ar.fout != nil {
err := ar.fout.Close()
if err != nil {
- mlog.Errf("AttackResult.cancel %s: %s\n", ar.TargetID, err)
+ mlog.Errf("AttackResult.cancel %s: %s\n", ar.HttpTargetID, err)
}
ar.fout = nil
if len(ar.fullpath) > 0 {
err = os.Remove(ar.fullpath)
if err != nil {
- mlog.Errf("AttackResult.cancel %s: %s\n", ar.TargetID, err)
+ mlog.Errf("AttackResult.cancel %s: %s\n", ar.HttpTargetID, err)
}
}
}
@@ -129,7 +129,7 @@ func (ar *AttackResult) finish() (err error) {
if ar.fout != nil {
err = ar.fout.Close()
if err != nil {
- return fmt.Errorf("%s: %w", ar.TargetID, err)
+ return fmt.Errorf("%s: %w", ar.HttpTargetID, err)
}
ar.fout = nil
}
@@ -157,19 +157,21 @@ func (ar *AttackResult) finish() (err error) {
return nil
}
-func (ar *AttackResult) init(path string) (err error) {
- ar.fullpath = filepath.Join(path, ar.Name)
+func (ar *AttackResult) load() (err error) {
+ if ar.TextReport != nil && ar.HistReport != nil {
+ return nil
+ }
result, err := ioutil.ReadFile(ar.fullpath)
if err != nil {
return err
}
- dec := vegeta.NewDecoder(bytes.NewReader(result))
-
ar.metrics = &vegeta.Metrics{}
ar.hist = &vegeta.Histogram{}
+ dec := vegeta.NewDecoder(bytes.NewReader(result))
+
err = ar.hist.Buckets.UnmarshalText([]byte(histogramBuckets))
if err != nil {
return err