Merge pull request #395 from viveksahu26/feat/add_timeout_flag

add timeout flag for cutomization(default to 60s)

Author: riteshnoronha

cmd/dtrackScore.go

@@ -83,6 +83,8 @@ func extractArgs(cmd *cobra.Command, args []string) (*engine.DtParams, error) {
 		params.ProjectIDs = append(params.ProjectIDs, argID)
 	}
 
+	params.Timeout, _ = cmd.Flags().GetInt("timeout")
+
 	return params, nil
 }
 
@@ -108,4 +110,5 @@ func init() {
 	dtrackScoreCmd.Flags().BoolP("basic", "b", false, "results in single line format")
 
 	dtrackScoreCmd.Flags().BoolP("tag-project-with-score", "t", false, "tag project with sbomqs score")
+	dtrackScoreCmd.Flags().IntP("timeout", "i", 60, "Timeout in seconds for Dependency-Track API requests")
 }

pkg/engine/dtrack.go

@@ -19,6 +19,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"time"
 
 	dtrack "github.com/DependencyTrack/client-go"
 	"github.com/google/uuid"
@@ -39,6 +40,7 @@ type DtParams struct {
 	Detailed bool
 
 	TagProjectWithScore bool
+	Timeout             int // handle cutom timeout
 }
 
 func DtrackScore(ctx context.Context, dtP *DtParams) error {
@@ -47,8 +49,12 @@ func DtrackScore(ctx context.Context, dtP *DtParams) error {
 
 	log.Debugf("Config: %+v", dtP)
 
+	timeout := time.Duration(dtP.Timeout) * time.Second
+
+	log.Debug("Timeout set to: ", timeout)
+
 	dTrackClient, err := dtrack.NewClient(dtP.URL,
-		dtrack.WithAPIKey(dtP.APIKey), dtrack.WithDebug(false))
+		dtrack.WithAPIKey(dtP.APIKey), dtrack.WithTimeout(timeout), dtrack.WithDebug(false))
 	if err != nil {
 		log.Fatalf("Failed to create Dependency-Track client: %s", err)
 	}