#9 Implement interface in the validator

internal/ddosml/forest.go

@@ -30,7 +30,7 @@ func newRandomForestClassifier() *randomForestClassifier {
 	}
 }
 
-// AddNewClassifierModel adds new Request to the randomForestClassifier.
+// EnhanceClassifierWithRequest adds new Request to the randomForestClassifier.
 // The trees are regenerated in order to increase decision success.
 func (r *randomForestClassifier) EnhanceClassifierWithRequest(m *Request) {
 	r.mu.Lock()

internal/ddosml/validator.go

@@ -3,6 +3,7 @@ package ddosml
 import (
 	"context"
 	"io"
+	"net/http"
 )
 
 type Classifier interface {
@@ -30,3 +31,11 @@ func NewMlModelValidator(classifier Classifier) *MLBasedModelValidator {
 		classifier: classifier,
 	}
 }
+
+func (m *MLBasedModelValidator) ValidateRequest(ctx context.Context, req *http.Request) (bool, error) {
+	requestModel := Request{}
+
+	isDDOS := m.classifier.IsRequestPotentialDDOS(ctx, &requestModel)
+
+	return isDDOS, nil
+}