Use graphql.HasOperationContext in arangodb assembler (#1659)

Signed-off-by: Ridwan Hoq <[email protected]>
guacsec · Jan 23, 2024 · f393612 · f393612
1 parent 91a9be2
commit f393612
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pkg/assembler/backends/arangodb/pkg.go b/pkg/assembler/backends/arangodb/pkg.go
@@ -433,7 +433,7 @@ func (c *arangoClient) Packages(ctx context.Context, pkgSpec *model.PkgSpec) ([]
 		return []*model.Package{p}, nil
 	}
 
-	if _, ok := ctx.Value("graphql").(graphql.OperationContext); ok {
+	if graphql.HasOperationContext(ctx) {
 		// fields: [type namespaces namespaces.namespace namespaces.names namespaces.names.name namespaces.names.versions
 		// namespaces.names.versions.version namespaces.names.versions.qualifiers namespaces.names.versions.qualifiers.key
 		// namespaces.names.versions.qualifiers.value namespaces.names.versions.subpath]

diff --git a/pkg/assembler/backends/arangodb/src.go b/pkg/assembler/backends/arangodb/src.go
@@ -319,7 +319,7 @@ func (c *arangoClient) Sources(ctx context.Context, sourceSpec *model.SourceSpec
 	}
 
 	// fields: [type namespaces namespaces.namespace namespaces.names namespaces.names.name namespaces.names.tag namespaces.names.commit]
-	if _, ok := ctx.Value("graphql").(graphql.OperationContext); ok {
+	if graphql.HasOperationContext(ctx) {
 		fields := getPreloads(ctx)
 
 		nameRequired := false

diff --git a/pkg/assembler/backends/arangodb/vulnerability.go b/pkg/assembler/backends/arangodb/vulnerability.go
@@ -63,7 +63,7 @@ func (c *arangoClient) Vulnerabilities(ctx context.Context, vulnSpec *model.Vuln
 		return []*model.Vulnerability{p}, nil
 	}
 
-	if _, ok := ctx.Value("graphql").(graphql.OperationContext); ok {
+	if graphql.HasOperationContext(ctx) {
 		// fields: [type vulnerabilityIDs ]
 		fields := getPreloads(ctx)