#9 Add part of knn classifier

thegodenage · May 15, 2024 · a4cde35 · a4cde35
1 parent fc691f8
commit a4cde35
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/internal/ddosml/knn.go b/internal/ddosml/knn.go
@@ -5,24 +5,34 @@ import (
 	"fmt"
 	"io"
 	"math"
+	"sync"
 
 	"github.com/cdipaolo/goml/base"
 )
 
-type knnClassifier struct {
+type KnnClassifier struct {
+	knn *knn
+
+	mu sync.Mutex
+}
+
+func NewKNNClassifier() *KnnClassifier {
+	return &KnnClassifier{
+		knn: &knn{},
+	}
 }
 
-func (k *knnClassifier) EnhanceClassifierWithRequest(m *Request) {
+func (k *KnnClassifier) EnhanceClassifierWithRequest(m *Request) {
 	//TODO implement me
 	panic("implement me")
 }
 
-func (k *knnClassifier) IsRequestPotentialDDOS(ctx context.Context, m *Request) bool {
+func (k *KnnClassifier) IsRequestPotentialDDOS(ctx context.Context, m *Request) bool {
 	//TODO implement me
 	panic("implement me")
 }
 
-func (k *knnClassifier) Write(writer io.Writer) error {
+func (k *KnnClassifier) Write(writer io.Writer) error {
 	//TODO implement me
 	panic("implement me")
 }