11import { useMemo } from 'react' ;
2- import { useMutation , useQuery , useQueryClient } from '@tanstack/react-query' ;
2+ import { useMutation , useQueryClient } from '@tanstack/react-query' ;
33
44import lms from '@src/data/services/lms' ;
55import { sortAlphaAsc } from '@src/data/formatUtils' ;
66import { filtersSnapshot } from '@src/data/filtersSnapshot' ;
77import { useGradebookUi } from '@src/data/gradebookUiContext' ;
88import { useCourseIdWithGate } from '@src/data/apiHook' ;
99import { useCourseId } from '@src/data/courseIdContext' ;
10+ import { useGradebookQuery } from '@src/data/query' ;
1011import {
1112 trackGradesDisplayed ,
1213 trackGradeOverrideSucceeded ,
@@ -19,7 +20,7 @@ import {
1920import { bulkOperationHistoryQueryKeys } from '@src/components/BulkManagementHistoryView/data/queryKeys' ;
2021
2122import { getGradeOverrideHistory , getGrades , getGradesPage } from './api' ;
22- import { gradeOverrideHistoryQueryKeys , gradesQueryKeys } from './queryKeys' ;
23+ import { gradeMutationKeys , gradeOverrideHistoryQueryKeys , gradesQueryKeys } from './queryKeys' ;
2324import { buildGradesFetchParams } from './utils' ;
2425
2526/**
@@ -45,7 +46,7 @@ export const useGrades = (
4546 courseId : string ,
4647 gradesPageEndpoint : string | null ,
4748 { enabled = true } : { enabled ?: boolean } = { } ,
48- ) => useQuery ( {
49+ ) => useGradebookQuery ( {
4950 queryKey : [ ...gradesQueryKeys . byCourse ( courseId ) , gradesPageEndpoint ?? 'base' ] ,
5051 queryFn : async ( ) => {
5152 const data = gradesPageEndpoint
@@ -143,7 +144,7 @@ export const useGradesData = (): GradesData => {
143144 * is not worth retrying.
144145 */
145146export const useGradeOverrideHistory = ( subsectionId ?: string , userId ?: string | number ) => (
146- useQuery ( {
147+ useGradebookQuery ( {
147148 queryKey : gradeOverrideHistoryQueryKeys . byCell ( subsectionId ?? null , userId ?? null ) ,
148149 queryFn : ( ) => getGradeOverrideHistory ( subsectionId as string , userId as string | number ) ,
149150 enabled : ! ! subsectionId && userId !== undefined && userId !== null ,
@@ -187,6 +188,7 @@ export const useUpdateGrades = () => {
187188 const courseId = useCourseId ( ) ;
188189 const { setShowSuccess, setGradesPageEndpoint, modalState } = useGradebookUi ( ) ;
189190 const mutation = useMutation ( {
191+ mutationKey : gradeMutationKeys . updateGrades ,
190192 mutationFn : ( updateData : GradeOverrideUpdate [ ] ) => lms . api . updateGradebookData ( courseId , updateData ) ,
191193 onSuccess : ( { data } , updateData ) => {
192194 trackGradeOverrideSucceeded ( courseId , data ) ;
@@ -238,6 +240,7 @@ export const useSubmitImportGradesButtonData = () => {
238240 setCsvUploadErrors,
239241 } = useGradebookUi ( ) ;
240242 const mutation = useMutation < unknown , CsvUploadError , FormData > ( {
243+ mutationKey : gradeMutationKeys . uploadGradeCsv ,
241244 mutationFn : ( formData ) => lms . api . uploadGradeCsv ( courseId , formData ) ,
242245 onMutate : ( ) => {
243246 resetCsvUpload ( ) ;
0 commit comments