all: set default HTTPTarget Attack if its not set

Previously, the function for Attack need to be coded manually.

This changes introduce new function DefaultHTTPAttack that generate
HTTPAttackHandler based on the HTTPTarget method, request type, and
Params; if AllowAttack is true and Attack is nil.

Implements: https://todo.sr.ht/~shulhan/gorankusu/4

1 files changed, 9 insertions, 6 deletions

diff --git a/http_target.go b/http_target.go
index b62156a..262f8ec 100644
--- a/http_target.go
+++ b/http_target.go
@@ -15,7 +15,6 @@ import (
 	libhttp "github.com/shuLhan/share/lib/http"
 	"github.com/shuLhan/share/lib/mlog"
 	libpath "github.com/shuLhan/share/lib/path"
-	vegeta "github.com/tsenart/vegeta/v12/lib"
 )
 
 // HTTPConvertParams is a handler that will be called inside the Run handler
@@ -26,10 +25,6 @@ type HTTPConvertParams func(target *HTTPTarget) (interface{}, error)
 // send request to run the HTTP target.
 type HTTPRunHandler func(rr *RunRequest) (runres *RunResponse, err error)
 
-// HTTPAttackHandler define the function type that will be called when client
-// send request to attack HTTP target.
-type HTTPAttackHandler func(rr *RunRequest) vegeta.Targeter
-
 // HTTPPreAttackHandler define the function type that will be called before
 // the actual Attack being called.
 type HTTPPreAttackHandler func(rr *RunRequest)
@@ -43,7 +38,10 @@ type HTTPTarget struct {
 
 	Run       HTTPRunHandler       `json:"-"`
 	PreAttack HTTPPreAttackHandler `json:"-"`
-	Attack    HTTPAttackHandler    `json:"-"`
+
+	// Attack define custom handler to generate [vegeta.Attacker].
+	// This field is optional default to [DefaultAttack].
+	Attack HTTPAttackHandler `json:"-"`
 
 	// RequestDumper define the handler to store [http.Request] after
 	// Run into [RunRequest] DumpRequest.
@@ -135,6 +133,11 @@ func (ht *HTTPTarget) init() (err error) {
 	if len(ht.Path) == 0 {
 		ht.Path = "/"
 	}
+
+	if ht.AllowAttack && ht.Attack == nil {
+		ht.Attack = DefaultHTTPAttack()
+	}
+
 	if ht.RequestDumper == nil {
 		ht.RequestDumper = DumpHTTPRequest
 	}