Skip to content

Commit f3a5174

Browse files
authored
Add new telemetry parameters (AST-113145) (#1306)
1 parent 76ff26e commit f3a5174

File tree

4 files changed

+16
-1
lines changed

4 files changed

+16
-1
lines changed

internal/commands/telemetry.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ func telemetryAISubCommand(telemetryAIWrapper wrappers.TelemetryWrapper) *cobra.
4040
telemetryAICmd.PersistentFlags().String(params.SubTypeFlag, "", "Sub Type")
4141
telemetryAICmd.PersistentFlags().String(params.EngineFlag, "", "Engine")
4242
telemetryAICmd.PersistentFlags().String(params.AgentFlag, "", "Agent")
43+
telemetryAICmd.PersistentFlags().String(params.ScanTypeFlag, "", "Scan Type")
44+
telemetryAICmd.PersistentFlags().String(params.StatusFlag, "", "Status")
45+
telemetryAICmd.PersistentFlags().Int(params.TotalCountFlag, 0, "Total Count")
4346

4447
return telemetryAICmd
4548
}
@@ -52,6 +55,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm
5255
subType, _ := cmd.Flags().GetString("sub-type")
5356
agent, _ := cmd.Flags().GetString("agent")
5457
engine, _ := cmd.Flags().GetString("engine")
58+
scanType, _ := cmd.Flags().GetString("scan-type")
59+
status, _ := cmd.Flags().GetString("status")
60+
totalCount, _ := cmd.Flags().GetInt("total-count")
5561

5662
err := telemetryWrapper.SendAIDataToLog(&wrappers.DataForAITelemetry{
5763
AIProvider: aiProvider,
@@ -60,6 +66,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm
6066
SubType: subType,
6167
Agent: agent,
6268
Engine: engine,
69+
ScanType: scanType,
70+
Status: status,
71+
TotalCount: totalCount,
6372
})
6473

6574
if err != nil {

internal/params/flags.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ const (
99
TypeFlag = "type"
1010
SubTypeFlag = "sub-type"
1111
EngineFlag = "engine"
12+
StatusFlag = "status"
13+
scanTypeFlag = "scan-type"
14+
TotalCountFlag = "total-count"
1215
OriginFlag = "origin"
1316
AgentFlagUsage = "Scan origin name"
1417
ApplicationName = "application-name"

internal/wrappers/realtime-scanner-http.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ func (r RealtimeScannerHTTPWrapper) ScanPackages(packages *RealtimeScannerPackag
3434
}
3535

3636
fn := func() (*http.Response, error) {
37-
return SendHTTPRequest(http.MethodPost, fmt.Sprint(r.path, "/analyze-manifest"), bytes.NewBuffer(jsonBytes), true, clientTimeout)
37+
return SendHTTPRequest(http.MethodPost, fmt.Sprint(r.path, "/scan/packages"), bytes.NewBuffer(jsonBytes), true, clientTimeout)
3838
}
3939
resp, err := retryHTTPRequest(fn, retryAttempts, retryDelay*time.Millisecond)
4040
if err != nil {

internal/wrappers/telemetry.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@ type DataForAITelemetry struct {
77
SubType string `json:"subType"`
88
Agent string `json:"agent"`
99
Engine string `json:"engine"`
10+
ScanType string `json:"scanType"`
11+
Status string `json:"status"`
12+
TotalCount int `json:"totalCount"`
1013
}
1114

1215
type TelemetryWrapper interface {

0 commit comments

Comments
 (0)