Merge pull request #46 from kubescape/feature/reduced-timeouts

Adding reduced timeouts

Author: amitschendel

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_cassandra.go

@@ -54,7 +54,7 @@ func (d *CassandraDiscovery) Discover(sessionHandler servicediscovery.ISessionHa
 		Username: Username,
 		Password: Password,
 	}
-	cluster.Timeout = time.Second * 3
+	cluster.Timeout = time.Millisecond * 500
 
 	// Create a session
 	session, err := cluster.CreateSession()

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_etcd.go

@@ -44,7 +44,7 @@ func (d *EtcdDiscovery) Discover(sessionHandler servicediscovery.ISessionHandler
 	zapLogger := zap.NewNop()
 	config := clientv3.Config{
 		Endpoints:   endpoints,
-		DialTimeout: 3 * time.Second,
+		DialTimeout: 500 * time.Millisecond,
 		Logger:      zapLogger,
 		LogConfig: &zap.Config{
 			Level:       zap.NewAtomicLevelAt(zap.ErrorLevel),

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_kafka.go

@@ -2,6 +2,7 @@ package applicationlayerdiscovery
 
 import (
 	"fmt"
+	"time"
 
 	log "github.com/sirupsen/logrus"
 
@@ -47,7 +48,7 @@ func (k *KafkaDiscovery) Discover(sessionHandler servicediscovery.ISessionHandle
 	config := sarama.NewConfig()
 	config.Producer.RequiredAcks = sarama.WaitForAll
 	config.Producer.Retry.Max = 1
-	config.Producer.Timeout = 3
+	config.Producer.Timeout = 500 * time.Millisecond
 	config.Producer.Return.Successes = true
 
 	// Create a new SyncProducer

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_kubeapi.go

@@ -50,7 +50,7 @@ func (d *KubeApiServerDiscovery) Discover(sessionHandler servicediscovery.ISessi
 	}
 
 	// Create an http.Client with the custom transport
-	client := &http.Client{Transport: tr, Timeout: time.Second * 3}
+	client := &http.Client{Transport: tr, Timeout: time.Millisecond * 500}
 
 	// Send a GET request to the Kubernetes API server
 	resp, err := client.Get(url)

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_mongodb.go

@@ -45,7 +45,7 @@ func (d *MongoDBDiscovery) Protocol() string {
 
 func (d *MongoDBDiscovery) Discover(sessionHandler servicediscovery.ISessionHandler, presentationLayerDiscoveryResult servicediscovery.IPresentationDiscoveryResult) (servicediscovery.IApplicationDiscoveryResult, error) {
 	clientOptions := options.Client().ApplyURI(fmt.Sprintf("mongodb://%s:%d", sessionHandler.GetHost(), sessionHandler.GetPort()))
-	connectionTimeout := 3 * time.Second
+	connectionTimeout := 500 * time.Millisecond
 	clientOptions.Timeout = &connectionTimeout
 	ctx := context.Background()
 	client, err := mongo.Connect(ctx, clientOptions)

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_postgresql.go

@@ -43,7 +43,7 @@ func (d *PostgresDiscovery) Protocol() string {
 
 func (d *PostgresDiscovery) Discover(sessionHandler servicediscovery.ISessionHandler, presentationLayerDiscoveryResult servicediscovery.IPresentationDiscoveryResult) (servicediscovery.IApplicationDiscoveryResult, error) {
 	// Set a timeout of 30 ms
-	db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=postgres sslmode=disable connect_timeout=3", sessionHandler.GetHost(), sessionHandler.GetPort()))
+	db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=postgres sslmode=disable connect_timeout=1", sessionHandler.GetHost(), sessionHandler.GetPort()))
 	if err != nil {
 		log.Debugf("Error while connecting to postgresql: %s", err.Error())
 		return &PostgresDiscoveryResult{

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_rabbitmq.go

@@ -44,7 +44,7 @@ func (d *RabbitMQDiscovery) Protocol() string {
 func (d *RabbitMQDiscovery) Discover(sessionHandler servicediscovery.ISessionHandler, presentationLayerDiscoveryResult servicediscovery.IPresentationDiscoveryResult) (servicediscovery.IApplicationDiscoveryResult, error) {
 	connectionString := fmt.Sprintf("amqp://%s:%d", sessionHandler.GetHost(), sessionHandler.GetPort())
 	config := amqp.Config{
-		Dial: amqp.DefaultDial(time.Second * 3),
+		Dial: amqp.DefaultDial(time.Millisecond * 500),
 	}
 	conn, err := amqp.DialConfig(connectionString, config)
 	if err != nil {

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_redis.go

@@ -44,7 +44,7 @@ func (d *RedisDiscovery) Discover(sessionHandler servicediscovery.ISessionHandle
 		Addr:        fmt.Sprintf("%s:%d", sessionHandler.GetHost(), sessionHandler.GetPort()),
 		Password:    "", // No password for now, modify as needed
 		DB:          0,  // Use default DB
-		DialTimeout: 3 * time.Second,
+		DialTimeout: 500 * time.Millisecond,
 		MaxRetries:  1,
 	})