yindo/posthog
1
0
Fork 0
mirror of https://github.com/BillyOutlast/posthog.git synced 2026-02-04 03:01:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat(managed migrations): adding pause / resume to managed migration (#41182)

Browse Source
This commit is contained in:
Christian Rafferty
2025-11-11 03:53:55 -05:00
committed by GitHub
parent b33a30d8f2
commit e01a25cd13
3 changed files with 55 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
posthog/api/batch_imports.py
View File

@@ -427,7 +427,7 @@ class BatchImportViewSet(TeamAndOrgViewSetMixin, viewsets.ModelViewSet):
return Response(response_serializer.data, status=status.HTTP_201_CREATED)
@action(methods=["POST"], detail=True)
def pause(self, request: Request, pk=None) -> Response:
def pause(self, request: Request, **kwargs) -> Response:
"""Pause a running batch import."""
batch_import = self.get_object()
@@ -444,7 +444,7 @@ class BatchImportViewSet(TeamAndOrgViewSetMixin, viewsets.ModelViewSet):
return Response({"status": "paused"})
@action(methods=["POST"], detail=True)
def resume(self, request: Request, pk=None) -> Response:
def resume(self, request: Request, **kwargs) -> Response:
"""Resume a paused batch import."""
batch_import = self.get_object()

31
products/managed_migrations/frontend/ManagedMigration.tsx
View File

@@ -214,6 +214,7 @@ export function ManagedMigration(): JSX.Element {
export function ManagedMigrations(): JSX.Element {
const { managedMigrationId, migrations, migrationsLoading } = useValues(managedMigrationLogic)
const { pauseMigration, resumeMigration } = useActions(managedMigrationLogic)
const calculateProgress = (migration: ManagedMigration): { progress: number; completed: number; total: number } => {
if (migration.state?.parts && Array.isArray(migration.state.parts)) {
@@ -399,6 +400,36 @@ export function ManagedMigrations(): JSX.Element {
dataIndex: 'status_message',
render: (_: any, migration: ManagedMigration) => migration.status_message || '-',
},
{
title: 'Actions',
key: 'actions',
render: (_: any, migration: ManagedMigration) => {
if (migration.status === 'running') {
return (
<LemonButton
type="secondary"
size="small"
onClick={() => pauseMigration(migration.id)}
loading={migrationsLoading}
>
Pause
</LemonButton>
)
} else if (migration.status === 'paused') {
return (
<LemonButton
type="primary"
size="small"
onClick={() => resumeMigration(migration.id)}
loading={migrationsLoading}
>
Resume
</LemonButton>
)
}
return null
},
},
]}
emptyState="No migrations found. Create a new migration to get started."
/>

22
products/managed_migrations/frontend/managedMigrationLogic.ts
View File

@@ -57,6 +57,8 @@ export const managedMigrationLogic = kea<managedMigrationLogicType>([
}),
actions({
editManagedMigration: (id: string | null) => ({ id }),
pauseMigration: (id: string) => ({ id }),
resumeMigration: (id: string) => ({ id }),
startPolling: true,
stopPolling: true,
}),
@@ -179,6 +181,26 @@ export const managedMigrationLogic = kea<managedMigrationLogicType>([
lemonToast.error('Failed to create migration. Please try again.')
}
},
pauseMigration: async ({ id }) => {
try {
const projectId = ApiConfig.getCurrentProjectId()
await api.create(`api/projects/${projectId}/managed_migrations/${id}/pause/`)
lemonToast.success('Migration paused successfully')
actions.loadMigrations()
} catch (error: any) {
lemonToast.error(error?.message || 'Failed to pause migration')
}
},
resumeMigration: async ({ id }) => {
try {
const projectId = ApiConfig.getCurrentProjectId()
await api.create(`api/projects/${projectId}/managed_migrations/${id}/resume/`)
lemonToast.success('Migration resumed successfully')
actions.loadMigrations()
} catch (error: any) {
lemonToast.error(error?.message || 'Failed to resume migration')
}
},
loadMigrationsSuccess: () => {
const hasRunningMigrations = values.migrations.some(
(migration: ManagedMigration) => migration.status === 'running'