Skip to content

Commit b2585ee

Browse files
committed
post fumpt
1 parent b006267 commit b2585ee

7 files changed

Lines changed: 31 additions & 11 deletions

File tree

internal/api/graphql/graph/baseResolver/vulnerability.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,14 @@ import (
2020
// - Only returns Issues of type Vulnerability
2121
// - Only returns Issues with at least one IssueMatch with status "new"
2222
// - Default ordering is by IssueVariantRating (descending), EarliestTargetRemediationDate (ascending) and IssuePrimaryName (ascending)
23-
func VulnerabilityBaseResolver(app app.Heureka, ctx context.Context, filter *model.VulnerabilityFilter, first *int, after *string, parent *model.NodeParent) (*model.VulnerabilityConnection, error) {
23+
func VulnerabilityBaseResolver(
24+
app app.Heureka,
25+
ctx context.Context,
26+
filter *model.VulnerabilityFilter,
27+
first *int,
28+
after *string,
29+
parent *model.NodeParent,
30+
) (*model.VulnerabilityConnection, error) {
2431
requestedFields := GetPreloads(ctx)
2532
logrus.WithFields(logrus.Fields{
2633
"requestedFields": requestedFields,

internal/api/graphql/graph/model/models.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -469,10 +469,12 @@ func NewVulnerability(issue *entity.Issue, aggregations *entity.IssueAggregation
469469
Name: &issue.PrimaryName,
470470
Description: &issue.Description,
471471
}
472+
472473
if aggregations != nil && !aggregations.EarliestTargetRemediationDate.IsZero() {
473474
date := aggregations.EarliestTargetRemediationDate.Format(time.RFC3339)
474475
v.EarliestTargetRemediationDate = &date
475476
}
477+
476478
return v
477479
}
478480

internal/api/graphql/graph/resolver/vulnerability.go

Lines changed: 1 addition & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/app/issue/issue_handler_test.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,12 +238,14 @@ var _ = Describe("When listing Issues", Label("app", "ListIssues"), func() {
238238
db.On("GetIssues", filter, []entity.Order{}).Return(issues, nil)
239239
db.On("GetAllIssueCursors", filter, []entity.Order{}).Return(cursors, nil)
240240
db.On("CountIssueTypes", filter).Return(issueTypeCounts, nil)
241-
issueHandler = appIssue.NewIssueHandler(handlerContext)
241+
issueHandler = issue.NewIssueHandler(handlerContext)
242242
res, err := issueHandler.ListIssues(filter, options)
243243
Expect(err).To(BeNil(), "no error should be thrown")
244244
Expect(*res.PageInfo.HasNextPage).To(BeEquivalentTo(hasNextPage), "correct hasNextPage indicator")
245245
Expect(len(res.Elements)).To(BeEquivalentTo(resElements))
246-
Expect(len(res.PageInfo.Pages)).To(BeEquivalentTo(int(math.Ceil(float64(dbElements)/float64(pageSize)))), "correct number of pages")
246+
Expect(
247+
len(res.PageInfo.Pages),
248+
).To(BeEquivalentTo(int(math.Ceil(float64(dbElements)/float64(pageSize)))), "correct number of pages")
247249
},
248250
Entry("When pageSize is 1 and the database was returning 2 elements", 1, 2, 1, true),
249251
Entry("When pageSize is 10 and the database was returning 9 elements", 10, 9, 9, false),

internal/database/mariadb/cursor.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,10 @@ func WithIssue(order []entity.Order, issue entity.Issue, ivRating int64, earlies
369369
case entity.IssueVariantRating:
370370
cursors.fields = append(cursors.fields, Field{Name: entity.IssueVariantRating, Value: ivRating, Order: o.Direction})
371371
case entity.IssueEarliestTargetRemediationDate:
372-
cursors.fields = append(cursors.fields, Field{Name: entity.IssueEarliestTargetRemediationDate, Value: earliestTargetRemediationDate, Order: o.Direction})
372+
cursors.fields = append(
373+
cursors.fields,
374+
Field{Name: entity.IssueEarliestTargetRemediationDate, Value: earliestTargetRemediationDate, Order: o.Direction},
375+
)
373376
default:
374377
continue
375378
}

internal/database/mariadb/issue.go

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -182,8 +182,8 @@ func getIssueJoinsExtended(filter *entity.IssueFilter, order []entity.Order, inc
182182
`)
183183
}
184184

185-
if len(filter.ServiceId) > 0 || len(filter.ServiceCCRN) > 0 || len(filter.SupportGroupCCRN) > 0 || filter.AllServices || includeAggJoins {
186-
185+
if len(filter.ServiceId) > 0 || len(filter.ServiceCCRN) > 0 || len(filter.SupportGroupCCRN) > 0 || filter.AllServices ||
186+
includeAggJoins {
187187
joins = fmt.Sprintf("%s\n%s", joins, `
188188
LEFT JOIN ComponentInstance CI ON CI.componentinstance_id = IM.issuematch_component_instance_id
189189
`)
@@ -435,7 +435,8 @@ func (s *SqlDatabase) GetIssuesWithAggregations(
435435

436436
filter = ensureIssueFilter(filter)
437437
joins := getIssueJoinsExtended(filter, order, true)
438-
cursorFields, err := DecodeCursor(filter.Paginated.After)
438+
439+
cursorFields, err := DecodeCursor(filter.After)
439440
if err != nil {
440441
return nil, err
441442
}
@@ -635,9 +636,10 @@ func (s *SqlDatabase) GetAllIssueCursors(
635636
if row.IssueVariantRow != nil {
636637
ivRating = row.RatingNumerical.Int64
637638
}
639+
638640
var etrd time.Time
639641
if row.IssueAggregationsRow != nil {
640-
etrd = GetTimeValue(row.IssueAggregationsRow.EarliestTargetRemediationDate)
642+
etrd = GetTimeValue(row.EarliestTargetRemediationDate)
641643
}
642644

643645
cursor, _ := EncodeCursor(WithIssue(order, issue, ivRating, etrd))
@@ -685,9 +687,10 @@ func (s *SqlDatabase) GetIssues(
685687
if e.IssueVariantRow != nil {
686688
ivRating = e.RatingNumerical.Int64
687689
}
690+
688691
var etrd time.Time
689692
if e.IssueAggregationsRow != nil {
690-
etrd = GetTimeValue(e.IssueAggregationsRow.EarliestTargetRemediationDate)
693+
etrd = GetTimeValue(e.EarliestTargetRemediationDate)
691694
}
692695

693696
cursor, _ := EncodeCursor(WithIssue(order, issue, ivRating, etrd))

internal/database/mariadb/issue_test.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -517,7 +517,9 @@ var _ = Describe("Issue", Label("database", "Issue"), func() {
517517
},
518518
[]entity.Order{},
519519
func(entries []entity.IssueResult) string {
520-
after, _ := mariadb.EncodeCursor(mariadb.WithIssue([]entity.Order{}, *entries[len(entries)-1].Issue, 0, time.Time{}))
520+
after, _ := mariadb.EncodeCursor(
521+
mariadb.WithIssue([]entity.Order{}, *entries[len(entries)-1].Issue, 0, time.Time{}),
522+
)
521523
return after
522524
},
523525
len(seedCollection.IssueRows),
@@ -1283,7 +1285,7 @@ var _ = Describe("Ordering Issues", Label("IssueOrder"), func() {
12831285
// 2. Issue 3 (Feb 17 07:15)
12841286
// 3. Issue 1 (Feb 17 01:10)
12851287

1286-
var idx1, idx2, idx3 int = -1, -1, -1
1288+
idx1, idx2, idx3 := -1, -1, -1
12871289
for i, r := range res {
12881290
if r.Issue.Id == issue1Row.Id.Int64 {
12891291
idx1 = i

0 commit comments

Comments
 (0)