Merge pull request #3 from taskiq-python/fix/worker-validation

fix: worker param can be nullable

Author: spikeninja

src/components/tasks-table.vue

@@ -138,7 +138,7 @@ const searchHandler: ((value: string) => void) | undefined =
             />
           </div>
         </TableHead>
-        <TableHead> Worker </TableHead>
+        <TableHead> Broker </TableHead>
       </TableRow>
     </TableHeader>
     <TableBody>

src/server/api/tasks/[id]/started.post.ts

@@ -1,17 +1,17 @@
 import {
   taskRouteParamsSchema,
-  taskStartedRequestSchema,
-} from "../../../schemas/tasks"
-import { tasksRepository } from "../../../repositories/tasks"
-import { envVariables } from "~/server/env"
+  taskStartedRequestSchema
+} from '../../../schemas/tasks'
+import { tasksRepository } from '../../../repositories/tasks'
+import { envVariables } from '~/server/env'
 
 export default defineEventHandler(async (event) => {
-  const accessToken = getRequestHeader(event, "access-token")
+  const accessToken = getRequestHeader(event, 'access-token')
   if (!accessToken || accessToken !== envVariables.taskiqAdminApiToken) {
     throw createError({
       status: 401,
-      statusMessage: "Unauthorized",
-      message: "Invalid access token",
+      statusMessage: 'Unauthorized',
+      message: 'Invalid access token'
     })
   }
   const params = await getValidatedRouterParams(
@@ -24,15 +24,15 @@ export default defineEventHandler(async (event) => {
     finishedAt: null,
     returnValue: null,
     executionTime: null,
-    state: "running",
+    state: 'running',
     args: body.args,
     id: params.id,
     worker: body.worker,
     kwargs: body.kwargs,
     name: body.taskName,
-    startedAt: body.startedAt,
+    startedAt: body.startedAt
   })
   return {
-    success: true,
+    success: true
   }
 })

src/server/repositories/tasks.ts

@@ -71,9 +71,9 @@ class TasksRepository {
   async create(values: {
     id: string
     name: string
-    worker: string
     startedAt: Date
     args: Array<any>
+    worker: string | null
     finishedAt: Date | null
     kwargs: Record<string, any>
     executionTime: number | null

src/server/schemas/tasks.ts

@@ -1,10 +1,10 @@
 import { z } from 'zod'
 
 export const taskStartedRequestSchema = z.object({
-  worker: z.string(),
   taskName: z.string(),
   startedAt: z.coerce.date(),
   args: z.array(z.unknown()),
+  worker: z.string().nullable(),
   kwargs: z.record(z.string(), z.unknown())
 })