From a069603b4f3b2c9caa2158d7b4087e432cf90668 Mon Sep 17 00:00:00 2001
From: Birk Johansson <birk@dhis2.org>
Date: Mon, 19 Jun 2023 11:20:00 +0200
Subject: [PATCH] fix(types): add generic result type to oncomplete param
 (#1350)

---
 services/data/src/engine/types/Query.ts       | 4 ++--
 services/data/src/react/hooks/useDataQuery.ts | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/services/data/src/engine/types/Query.ts b/services/data/src/engine/types/Query.ts
index 5a79c1c6..e058b53c 100644
--- a/services/data/src/engine/types/Query.ts
+++ b/services/data/src/engine/types/Query.ts
@@ -21,9 +21,9 @@ export interface ResolvedResourceQuery extends ResourceQuery {
 export type Query = Record<string, ResourceQuery>
 export type QueryResult = JsonMap
 
-export interface QueryOptions {
+export interface QueryOptions<TQueryResult = QueryResult> {
     variables?: QueryVariables
-    onComplete?: (data: QueryResult) => void
+    onComplete?: (data: TQueryResult) => void
     onError?: (error: FetchError) => void
     lazy?: boolean
 }
diff --git a/services/data/src/react/hooks/useDataQuery.ts b/services/data/src/react/hooks/useDataQuery.ts
index 9b81a145..ea9cafe2 100644
--- a/services/data/src/react/hooks/useDataQuery.ts
+++ b/services/data/src/react/hooks/useDataQuery.ts
@@ -40,7 +40,7 @@ export const useDataQuery = <TQueryResult = QueryResult>(
         onError: userOnError,
         variables: initialVariables = {},
         lazy: initialLazy = false,
-    }: QueryOptions = {}
+    }: QueryOptions<TQueryResult> = {}
 ): QueryRenderInput<TQueryResult> => {
     const [staticQuery] = useStaticInput<Query>(query, {
         warn: true,