Files
supabase/apps/studio/data/read-replicas/replica-setup-mutation.ts
Joshen LimandIvan Vasilov bb6349f34f Second round of wrapping RQ errors with handleError (#26428)
* First round of wrapping RQ errors with handleError

* Remove the throw before the handleError usage.

* Make the handling of an API error more versatile. Add logging in Sentry if the error is of unknown type.

* Remove throwing of the handleError function.

* Add return type to the handleError function to be never so that we're sure it always throws.

* Second round of wrapping RQ errors with handleError

* Temp fix in delete credential mutation, and fix loading state

---------

Co-authored-by: Ivan Vasilov <vasilov.ivan@gmail.com>
2024-05-21 15:51:11 +08:00

73 lines
1.9 KiB
TypeScript

import { useMutation, UseMutationOptions, useQueryClient } from '@tanstack/react-query'
import { toast } from 'react-hot-toast'
import { handleError, post } from 'data/fetchers'
import type { ResponseError } from 'types'
import { replicaKeys } from './keys'
export type Region =
| 'us-east-1'
| 'us-west-1'
| 'us-west-2'
| 'ap-southeast-1'
| 'ap-northeast-1'
| 'ap-northeast-2'
| 'ap-southeast-2'
| 'eu-west-1'
| 'eu-west-2'
| 'eu-west-3'
| 'eu-central-1'
| 'ca-central-1'
| 'ap-south-1'
| 'sa-east-1'
export type ReadReplicaSetUpVariables = {
projectRef: string
region: Region
size: string // Not used in API yet, purely for UI atm
}
export async function setUpReadReplica({ projectRef, region }: ReadReplicaSetUpVariables) {
const { data, error } = await post('/v1/projects/{ref}/read-replicas/setup', {
params: {
path: { ref: projectRef },
},
body: {
read_replica_region: region,
},
})
if (error) handleError(error)
return data
}
type ReadReplicaSetUpData = Awaited<ReturnType<typeof setUpReadReplica>>
export const useReadReplicaSetUpMutation = ({
onSuccess,
onError,
...options
}: Omit<
UseMutationOptions<ReadReplicaSetUpData, ResponseError, ReadReplicaSetUpVariables>,
'mutationFn'
> = {}) => {
const queryClient = useQueryClient()
return useMutation<ReadReplicaSetUpData, ResponseError, ReadReplicaSetUpVariables>(
(vars) => setUpReadReplica(vars),
{
async onSuccess(data, variables, context) {
const { projectRef } = variables
await queryClient.invalidateQueries(replicaKeys.list(projectRef))
await onSuccess?.(data, variables, context)
},
async onError(data, variables, context) {
if (onError === undefined) {
toast.error(`Failed to set up read replica: ${data.message}`)
} else {
onError(data, variables, context)
}
},
...options,
}
)
}