import feeder apis

.gitignore

@@ -0,0 +1,3 @@
+.vscode/
+knoxAutoPolicy
+

autodiscovery/networkPolicyDiscovery.go

@@ -1,7 +1,6 @@
 package autodiscovery
 
 import (
-	"fmt"
 	"math/rand"
 	"reflect"
 	"sort"
@@ -842,7 +841,7 @@ func groupingDst(perGroupedSrcMergedDst map[string][]DstMerged, conGroups []type
 // ======================= //
 
 // GenerateNetworkPolicies Function
-func GenerateNetworkPolicies(networkLogs []types.NetworkLog, microName string, services []types.K8sService, conGroups []types.ContainerGroup) []types.NetworkPolicy {
+func GenerateNetworkPolicies(microName string, networkLogs []types.NetworkLog, services []types.K8sService, conGroups []types.ContainerGroup) []types.NetworkPolicy {
 	networkLogs = filterLogs(networkLogs, microName)
 
 	// step 0: update exposed ports (k8s service, docker-compose portbinding)
@@ -871,11 +870,3 @@ func GenerateNetworkPolicies(networkLogs []types.NetworkLog, microName string, s
 
 	return policies
 }
-
-// ======================= //
-// == Policy Generation == //
-// ======================= //
-
-func TestGenerateNetworkPolicies() {
-	fmt.Println("TestGenerateNetworkPolicies")
-}

dbase/dbase.go

@@ -14,7 +14,14 @@ func Conn() (db *sql.DB) {
 	dbUser := os.Getenv("NETWORKFLOW_DB_USER")
 	dbPass := os.Getenv("NETWORKFLOW_DB_PASS")
 	dbName := os.Getenv("NETWORKFLOW_DB_NAME")
-	db, err := sql.Open(dbDriver, dbUser+":"+dbPass+"@/"+dbName)
+
+	// for test
+	dbDriver = "mysql"
+	dbUser = "root"
+	dbPass = "password"
+	dbName = "flow_management"
+
+	db, err := sql.Open(dbDriver, dbUser+":"+dbPass+"@tcp(localhost:3306)/"+dbName)
 	if err != nil {
 		panic(err.Error())
 	}

go.mod

@@ -18,5 +18,5 @@ require (
 	golang.org/x/sys v0.0.0-20201009025420-dfb3f7c4e634 // indirect
 	google.golang.org/genproto v0.0.0-20201009135657-4d944d34d83c // indirect
 	google.golang.org/grpc v1.33.0
-	google.golang.org/protobuf v1.25.0 // indirect
+	google.golang.org/protobuf v1.25.0
 )

main.go

@@ -4,22 +4,28 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
-	"fmt"
 	"io"
 	"os"
 	"strings"
 
-	autopol "github.com/seungsoo-lee/knoxAutoPolicy/autodiscovery"
 	"github.com/seungsoo-lee/knoxAutoPolicy/dbase"
 
+	pb "github.com/cilium/cilium/api/v1/flow"
+
 	"github.com/cilium/cilium/api/v1/observer"
 	"google.golang.org/grpc"
+	"google.golang.org/protobuf/types/known/timestamppb"
 )
 
 //ConnectHubbleRelay function
 func ConnectHubbleRelay() *grpc.ClientConn {
 	url := os.Getenv("HUBBLE_URL")
 	port := os.Getenv("HUBBLE_PORT")
+
+	// for test
+	url = "10.104.6.225"
+	port = "80"
+
 	addr := url + ":" + port
 	conn, err := grpc.Dial(addr, grpc.WithInsecure())
 	if err != nil {
@@ -315,12 +321,18 @@ func StartFeederService(ctx context.Context) error {
 
 	client := observer.NewObserverClient(conn)
 
+	wl := []*pb.FlowFilter{
+		{
+			SourceIp: []string{"10.0.1.222"},
+		},
+	}
+
 	req := &observer.GetFlowsRequest{
 		Number:    20,
 		Follow:    true,
-		Whitelist: nil,
+		Whitelist: wl,
 		Blacklist: nil,
-		Since:     nil,
+		Since:     timestamppb.Now(),
 		Until:     nil,
 	}
 
@@ -333,6 +345,7 @@ func StartFeederService(ctx context.Context) error {
 			if err != nil {
 				return errors.New("can't receive network flow")
 			}
+
 			err = insertFlow(res)
 			if err != nil {
 				return err
@@ -344,8 +357,5 @@ func StartFeederService(ctx context.Context) error {
 }
 
 func main() {
-	autopol.TestGenerateNetworkPolicies()
-	url := os.Getenv("HUBBLE_URL")
-	port := os.Getenv("HUBBLE_PORT")
-	fmt.Println(url, port)
+	StartFeederService(context.Background())
 }