Add vars map to QueryRaw

Author: matthewmcneely

client.go

@@ -1,3 +1,8 @@
+/*
+ * SPDX-FileCopyrightText: © Hypermode Inc. <hello@hypermode.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
 package modusgraph
 
 import (
@@ -27,7 +32,7 @@ type Client interface {
 	GetSchema(context.Context) (string, error)
 	DropAll(context.Context) error
 	DropData(context.Context) error
-	QueryRaw(context.Context, string) ([]byte, error)
+	QueryRaw(context.Context, string, map[string]string) ([]byte, error)
 
 	DgraphClient() (*dgo.Dgraph, func(), error)
 }
@@ -288,7 +293,7 @@ func (c client) Query(ctx context.Context, model any) *dg.Query {
 	}
 	defer c.pool.put(client)
 
-	txn := dg.NewTxn(client)
+	txn := dg.NewReadOnlyTxnContext(ctx, client)
 	return txn.Get(model)
 }
 
@@ -342,11 +347,11 @@ func (c client) DropData(ctx context.Context) error {
 	return client.Alter(ctx, &api.Operation{DropOp: api.Operation_DATA})
 }
 
-// QueryRaw implements raw querying (DQL syntax).
-func (c client) QueryRaw(ctx context.Context, q string) ([]byte, error) {
+// QueryRaw implements raw querying (DQL syntax) and optional variables.
+func (c client) QueryRaw(ctx context.Context, q string, vars map[string]string) ([]byte, error) {
 	if c.engine != nil {
 		ns := c.engine.GetDefaultNamespace()
-		resp, err := ns.Query(ctx, q)
+		resp, err := ns.QueryWithVars(ctx, q, vars)
 		if err != nil {
 			return nil, err
 		}
@@ -361,7 +366,7 @@ func (c client) QueryRaw(ctx context.Context, q string) ([]byte, error) {
 	defer c.pool.put(client)
 
 	txn := dg.NewReadOnlyTxnContext(ctx, client)
-	resp, err := txn.Txn().Query(ctx, q)
+	resp, err := txn.Txn().QueryWithVars(ctx, q, vars)
 	if err != nil {
 		return nil, err
 	}

cmd/query/main.go

@@ -97,7 +97,7 @@ func main() {
 	start := time.Now()
 
 	// Execute the query
-	resp, err := client.QueryRaw(ctx, query)
+	resp, err := client.QueryRaw(ctx, query, nil)
 	if err != nil {
 		logger.Error(err, "Query execution failed")
 		os.Exit(1)