@squiz/job-runner-lib 2.3.0 → 2.5.0

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # Change Log
 
+## 2.5.0
+
+### Minor Changes
+
+- 1c41058: Updated packages
+
+## 2.4.0
+
+### Minor Changes
+
+- 22c2a8f: Updated dependencies.
+
 ## 2.3.0
 
 ### Minor Changes

package/lib/domains/component/errors/JobError.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/domains/job/ParsePartitionKey/ParsePartitionKey.d.ts

@@ -1,2 +1,2 @@
-import { JobExecutionPk, PartitionKey } from '../constants/Repository.constants';
-export declare function parsePartitionKey(type: string): PartitionKey | JobExecutionPk;
+import { ParsePartitionKeyResult } from './ParsePartitionKey.types';
+export declare function parsePartitionKey(type: string): ParsePartitionKeyResult;

package/lib/domains/job/ParsePartitionKey/ParsePartitionKey.types.d.ts

@@ -0,0 +1,5 @@
+import { EntityType } from '../constants/Repository.constants';
+export type ParsePartitionKeyResult = {
+    entityType: EntityType;
+    tenant: string;
+};

package/lib/domains/job/S3Repository/JobDetailsRepository.d.ts

@@ -12,6 +12,5 @@ export declare class JobDetailsRepository extends S3RepositoryBase {
      * We want the file to be created so the job starts/executes rather than remaining in a queued state/causing
      * an error.
      */
-    createJobDetailsFile(jobDetailsFileName: string, jobContext: JobContext | null, jobManifest: JobManifest | null, jobExecution: JobExecution): Promise<void>;
-    static createJobDetailsFileName(tenantId: string, jobExecutionId: string): string;
+    createJobDetailsFile(tenant: string, jobContext: JobContext | null, jobManifest: JobManifest | null, jobExecution: JobExecution): Promise<string>;
 }

package/lib/domains/job/S3Repository/JobDetailsRepository.spec.d.ts

@@ -2,4 +2,4 @@
  * @license
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
-export {};
+import 'aws-sdk-client-mock-jest';

package/lib/domains/job/S3Repository/JobUploadsRepository.d.ts

@@ -0,0 +1,16 @@
+/*!
+ * @license
+ * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
+ */
+import { ListObjectsV2CommandOutput, S3Client } from '@aws-sdk/client-s3';
+import { S3RepositoryBase } from './S3RepositoryBase';
+/**
+ * Repository class for interacting with the job uploads S3 bucket.
+ * @export
+ * @class JobUploadsRepository
+ */
+export declare class JobUploadsRepository extends S3RepositoryBase {
+    protected s3Client: S3Client;
+    constructor(s3Client: S3Client, bucketName: string);
+    listByTenant(tenant: string, continuationToken?: string): Promise<ListObjectsV2CommandOutput>;
+}

package/lib/domains/job/{services/JobQueueService/JobQueueService.spec.d.ts → S3Repository/JobUploadsRepository.spec.d.ts}

@@ -2,4 +2,4 @@
  * @license
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
-export {};
+import 'aws-sdk-client-mock-jest';

package/lib/domains/job/S3Repository/S3RepositoryBase.d.ts

@@ -7,9 +7,9 @@ import 'reflect-metadata';
 /**
  * Repository class for interacting with S3.
  * @export
- * @class S3Repository
+ * @class S3RepositoryBase
  */
-export declare abstract class S3RepositoryBase {
+export declare class S3RepositoryBase {
     protected s3Client: S3Client;
     protected bucketName: string;
     constructor(s3Client: S3Client, bucketName: string);
@@ -18,21 +18,28 @@ export declare abstract class S3RepositoryBase {
      * @param {string} key The S3 object key.
      * @param {PutObjectCommandInputType['Body']} body The body of the S3 object.
      * @return {Promise<void>}
-     * @memberof S3Repository
+     * @memberof S3RepositoryBase
      */
     put(key: string, body: PutObjectCommandInput['Body']): Promise<void>;
+    /**
+     * Delete object from S3.
+     *
+     * @param {string} key The S3 object key.
+     * @memberof S3RepositoryBase
+     */
+    delete(key: string): Promise<void>;
     /**
      * Retrieve object from S3.
      * @param {string} key The S3 object key.
      * @return {Promise<string | null>} The S3 object.
-     * @memberof S3Repository
+     * @memberof S3RepositoryBase
      */
     get(key: string): Promise<string | null>;
     /**
      * Check if the S3 object key exists.
      * @param {string} key The S3 object key.
      * @return {Promise<HeadObjectCommandOutput>}
-     * @memberof S3Repository
+     * @memberof S3RepositoryBase
      */
     check(key: string): Promise<HeadObjectCommandOutput>;
     getS3SignedUrl(key: string): Promise<string>;

package/lib/domains/job/S3Repository/index.d.ts

@@ -1,2 +1,3 @@
 export * from './JobDetailsRepository';
-export * from './S3Repository';
+export * from './JobUploadsRepository';
+export * from './JobUploadsRepository';

package/lib/domains/job/constants/InjectTokens.d.ts

@@ -0,0 +1,8 @@
+export declare const InjectTokens: {
+    Tenant: symbol;
+    DynamoTableName: symbol;
+    JobUploadsBucket: symbol;
+    JobDetailsBucket: symbol;
+    JobStartQueueUrl: symbol;
+    Logger: symbol;
+};

package/lib/domains/job/constants/JobExecutionStatus.constants.d.ts

@@ -4,5 +4,6 @@ export declare enum JOB_EXECUTION_STATUS {
     terminated = "terminated",
     successfullyCompleted = "successfully-completed",
     running = "running",
-    errored = "errored"
+    errored = "errored",
+    taskFailedToStart = "task-failed-to-start"
 }

package/lib/domains/job/constants/Repository.constants.d.ts

@@ -2,9 +2,9 @@
  * @license
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
-export type PartitionKey = PARTITION_KEYS | JobExecutionPk;
-export type JobExecutionPk = string;
-export declare enum PARTITION_KEYS {
+export type EntityType = ENTITY_TYPES | EntityTypeRaw;
+export type EntityTypeRaw = string;
+export declare enum ENTITY_TYPES {
     job = "job",
     manifest = "manifest",
     jobExecution = "jobExecution",

package/lib/domains/job/constants/RetryConfig.constants.d.ts

@@ -0,0 +1,21 @@
+/*!
+ * @license
+ * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
+ */
+/**
+ * Retry configuration with exponential backoff delays in seconds.
+ * Using SQS message timer limits (max 15 minutes = 900 seconds).
+ *
+ * Delays: 30s, 1min, 2min, 5min, 10min, 15min
+ */
+export declare const RETRY_DELAYS_SECONDS: ReadonlyArray<number>;
+/**
+ * Maximum number of retry attempts for container start failures.
+ */
+export declare const MAX_RETRY_ATTEMPTS: number;
+/**
+ * Get the delay in seconds for a given retry attempt.
+ * @param retryAttempt - The current retry attempt (0-indexed)
+ * @returns The delay in seconds for this retry attempt
+ */
+export declare const getRetryDelaySeconds: (retryAttempt: number) => number;

package/lib/domains/job/index.d.ts

@@ -1,18 +1,20 @@
-export * from './core/pagination';
 export * from './core/dynamoClient.mock';
+export * from './core/pagination';
+export * from './constants/InjectTokens';
 export * from './constants/JobExecutionStatus.constants';
 export * from './constants/Repository.constants';
+export * from './constants/RetryConfig.constants';
 export * from './services';
 export * from './utils';
+export * from './model/Base';
 export * from './model/Job';
 export * from './model/JobContext';
 export * from './model/JobExecution';
-export * from './model/Base';
 export * from './model/LogPayload';
 export * from './errors';
 export * from './manifest';
 export * from './ParsePartitionKey';
-export * from './transientProvider';
-export * from './singletonProvider';
 export * from './repository';
 export * from './S3Repository';
+export * from './singletonProvider';
+export * from './transientProvider';

package/lib/domains/job/manifest/JobManifestTypes.d.ts

@@ -4,7 +4,7 @@
  */
 import { AttributeValue } from '@aws-sdk/client-dynamodb';
 import { MANIFEST_MODELS } from '@squiz/dx-json-schema-lib';
-import { PartitionKey } from '../constants/Repository.constants';
+import { EntityType } from '../constants/Repository.constants';
 /**
  * Job manifest interface.
  * @export
@@ -51,10 +51,10 @@ export interface JobManifest {
     concurrency: number;
     /**
      * The DynamoDB record type ('job').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof JobManifest
      */
-    type: PartitionKey;
+    type: EntityType;
 }
 /**
  * The Job Manifest record attributes stored in DynamoDB.
@@ -64,10 +64,10 @@ export interface JobManifest {
 export interface JobManifestRecord {
     /**
      * The partition key ('manifest').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof JobManifestRecord
      */
-    pk: PartitionKey;
+    pk: EntityType;
     /**
      * The sort key ('name-version').
      * @type {string}
@@ -123,4 +123,5 @@ export type CreateJobManifestRequest = Omit<JobManifest, 'type'>;
  * The request body for updating a job (PUT /job-manifest).
  * @export
  */
-export type UpdateJobManifestRequest = Partial<Omit<JobManifest, 'name' | 'type'>>;
+export type UpdateJobManifestRequest = Partial<Omit<JobManifest, 'name' | 'version' | 'type'>>;
+export type JobManifestSk = Pick<JobManifest, 'name' | 'version'>;

package/lib/domains/job/model/Base/Base.d.ts

@@ -1,5 +1,5 @@
 import { AttributeValue } from '@aws-sdk/client-dynamodb';
-import { PartitionKey } from '../../constants/Repository.constants';
+import { EntityType } from '../../constants/Repository.constants';
 /**
  * The base interface for job runner entities.
  * @export
@@ -8,10 +8,10 @@ import { PartitionKey } from '../../constants/Repository.constants';
 export interface BaseItem {
     /**
      * The property representing the corresponding record's partition key.
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof BaseItem
      */
-    type: PartitionKey;
+    type: EntityType;
 }
 /**
  * The base interface representing the attributes of a DynamoDB record.
@@ -21,10 +21,10 @@ export interface BaseItem {
 export interface BaseRecord {
     /**
      * The partition key.
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof BaseRecord
      */
-    pk: PartitionKey;
+    pk: EntityType;
     /**
      * The sort key.
      * @type {string}
@@ -38,3 +38,8 @@ export interface BaseRecord {
  * @interface BaseDto
  */
 export type BaseDto = Record<keyof BaseRecord, AttributeValue>;
+export type ListParameters<TFilters> = {
+    filters?: TFilters;
+    consistentRead?: boolean;
+    limit?: number;
+};

package/lib/domains/job/model/Job/Job.d.ts

@@ -3,7 +3,8 @@
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
 import { AttributeValue } from '@aws-sdk/client-dynamodb';
-import { PartitionKey } from '../../constants/Repository.constants';
+import { EntityType } from '../../constants/Repository.constants';
+import { ListParameters } from '../Base';
 /**
  * Job interface.
  * @export
@@ -30,10 +31,10 @@ export interface Job {
     version: string;
     /**
      * The DynamoDB record type ('job').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof Job
      */
-    type: PartitionKey;
+    type: EntityType;
 }
 /**
  * The Job record attributes stored in DynamoDB.
@@ -43,10 +44,10 @@ export interface Job {
 export interface JobRecord {
     /**
      * The partition key ('job').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof JobRecord
      */
-    pk: PartitionKey;
+    pk: EntityType;
     /**
      * The sort key ('name-version').
      * @type {string}
@@ -77,9 +78,11 @@ export type JobDto = Record<keyof JobRecord, AttributeValue>;
  * The parameters for listing jobs.
  * @export
  */
-export type ListJobParameters = Partial<Pick<Job, 'name' | 'version'>>;
+export type ListJobParameters = ListParameters<Partial<Pick<Job, 'name' | 'version'>>>;
 /**
  * The request body for creating a job (POST /job).
  * @export
  */
 export type CreateJobRequest = Omit<Job, 'type'>;
+export type UpdateJobRequest = Partial<Omit<Job, 'name' | 'version' | 'type'>>;
+export type JobSk = Pick<Job, 'name' | 'version'>;

package/lib/domains/job/model/JobContext/JobContext.d.ts

@@ -3,7 +3,7 @@
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
 import { AttributeValue } from '@aws-sdk/client-dynamodb';
-import { PartitionKey } from '../../constants/Repository.constants';
+import { EntityType } from '../../constants/Repository.constants';
 /**
  * Environment interface.
  * @interface Environment
@@ -38,10 +38,10 @@ export interface JobContext {
     environment: Environment;
     /**
      * The DynamoDB record type ('context').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof JobContext
      */
-    type: PartitionKey;
+    type: EntityType;
 }
 /**
  * The Job Context record attributes stored in DynamoDB.
@@ -51,10 +51,10 @@ export interface JobContext {
 export interface JobContextRecord {
     /**
      * The partition key ('context').
-     * @type {PartitionKey}
+     * @type {EntityType}
      * @memberof JobContextRecord
      */
-    pk: PartitionKey;
+    pk: EntityType;
     /**
      * The description of the job context.
      * @type {string}
@@ -85,3 +85,4 @@ export type CreateJobContextRequest = Omit<JobContext, 'type'>;
  * @export
  */
 export type UpdateJobContextRequest = Partial<Omit<JobContext, 'contextName' | 'type'>>;
+export type JobContextSk = Pick<JobContext, 'contextName'>;

package/lib/domains/job/model/JobExecution/JobExecution.d.ts

@@ -2,9 +2,9 @@
  * @license
  * Copyright Squiz Australia Pty Ltd. All Rights Reserved.
  */
-import { JobExecutionStatus } from '../../constants/JobExecutionStatus.constants';
-import { JobExecutionPk } from '../../constants/Repository.constants';
 import { AttributeValue } from '@aws-sdk/client-dynamodb';
+import { JobExecutionStatus } from '../../constants/JobExecutionStatus.constants';
+import { ListParameters } from '../Base';
 /**
  * JobExecution interface.
  * @export
@@ -73,10 +73,29 @@ export interface JobExecution {
     input: any;
     /**
      * The DynamoDB record type (dynamic, 'jobExecution-<jobName>').
-     * @type {JobExecutionPk}
+     * @type {string}
+     * @memberof JobExecution
+     */
+    type: string;
+    /**
+     * The number of times the container start has been retried.
+     * @type {number}
+     * @memberof JobExecution
+     */
+    retryCount: number;
+    /**
+     * The reason for the last container start failure (if any).
+     * @type {(string | null)}
+     * @memberof JobExecution
+     */
+    lastFailureReason: string | null;
+    /**
+     * ISO timestamp indicating when the execution can be retried.
+     * Used to implement exponential backoff - executions with retryAfter > now should be skipped.
+     * @type {(string | null)}
      * @memberof JobExecution
      */
-    type: JobExecutionPk;
+    retryAfter: string | null;
 }
 /**
  * The JobExecution record attributes stored in DynamoDB.
@@ -85,17 +104,29 @@ export interface JobExecution {
  */
 export interface JobExecutionRecord {
     /**
-     * The partition key (dynamic, 'jobExecution-<jobName>').
-     * @type {JobExecutionPk}
+     * The partition key ('jobExecution~<tenant>').
+     * @type {string}
      * @memberof JobExecutionRecord
      */
-    pk: JobExecutionPk;
+    pk: string;
     /**
-     * The sort key ('timeQueued-id').
+     * The sort key ('<jobName>~<id>').
      * @type {string}
      * @memberof JobExecutionRecord
      */
     sk: string;
+    /**
+     * Secondary sort key ('<status>~<jobName>').
+     * @type {string}
+     * @memberof JobExecutionRecord
+     */
+    lsi1sk: string;
+    /**
+     * Global secondary partition key ('jobExecution~<status>').
+     * @type {string}
+     * @memberof JobExecutionRecord
+     */
+    gsi1pk: string;
     /**
      * The name of the job to execute.
      * @type {string}
@@ -156,6 +187,25 @@ export interface JobExecutionRecord {
      * @memberof JobExecutionRecord
      */
     input: any;
+    /**
+     * The number of times the container start has been retried.
+     * @type {number}
+     * @memberof JobExecutionRecord
+     */
+    retryCount: number;
+    /**
+     * The reason for the last container start failure (if any).
+     * @type {(string | null)}
+     * @memberof JobExecutionRecord
+     */
+    lastFailureReason: string | null;
+    /**
+     * ISO timestamp indicating when the execution can be retried.
+     * Used to implement exponential backoff - executions with retryAfter > now should be skipped.
+     * @type {(string | null)}
+     * @memberof JobExecutionRecord
+     */
+    retryAfter: string | null;
 }
 export type JobExecutionDto = Record<keyof JobExecutionRecord, AttributeValue>;
 /**
@@ -181,9 +231,14 @@ export interface TerminateJobExecutionRequest {
  * The parameters for listing job executions.
  * @export
  */
-export type ListJobExecutionParameters = Partial<Pick<JobExecution, 'jobName' | 'status'>>;
+export type ListJobExecutionParameters = ListParameters<Partial<Pick<JobExecution, 'jobName' | 'status'>>>;
 /**
  * The parameters for updating job executions.
  * @export
  */
 export type UpdateJobExecutionParameters = Partial<Omit<JobExecution, 'id' | 'jobName' | 'timeQueued' | 'type'>>;
+export type JobExecutionSk = Pick<JobExecution, 'jobName' | 'id'>;
+/**
+ * Array of job executions keys by tenant.
+ */
+export type JobExecutionsByTenant = Record<string, Array<JobExecution>>;

package/lib/domains/job/model/LogPayload/LogPayload.d.ts

@@ -1,12 +1,21 @@
-export type LogPayload = {
-    message: string;
-    type?: string;
-    area?: string;
-    req_id?: string;
-    'log-command'?: string;
-    level?: string;
-    'function-name'?: string;
-    'job-name'?: string;
-    'job-version'?: string;
-    timestamp: string;
-};
+import { z } from 'zod';
+export declare enum LogLevel {
+    AUDIT = "AUDIT",
+    INFO = "INFO",
+    WARNING = "WARNING",
+    ERROR = "ERROR"
+}
+export declare const LogPayloadSchema: z.ZodObject<{
+    message: z.ZodString;
+    meta: z.ZodObject<{
+        tenant: z.ZodString;
+        jobName: z.ZodString;
+        level: z.ZodEnum<typeof LogLevel>;
+        timestamp: z.ZodISODateTime;
+        traceid: z.ZodString;
+        tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
+        details: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export type LogPayload = z.infer<typeof LogPayloadSchema>;
+export declare const CONSOLE_LOG_METHOD_LEVELS: Partial<Record<keyof typeof console, LogLevel>>;

package/lib/domains/job/model/LogPayload/index.d.ts

@@ -1 +1,2 @@
 export * from './LogPayload';
+export * from './mocks/LogPayload.mock';

package/lib/domains/job/model/LogPayload/mocks/LogPayload.mock.d.ts

@@ -0,0 +1,6 @@
+import { LogPayload } from '../LogPayload';
+type DeepPartial<T> = {
+    [K in keyof T]?: T[K] extends object ? DeepPartial<T[K]> : T[K];
+};
+export declare const createMockLogPayload: (partialRecord: DeepPartial<LogPayload>) => LogPayload;
+export {};

package/lib/domains/job/repository/AbstractRepository.d.ts

@@ -1,7 +1,6 @@
 import 'reflect-metadata';
 import { AttributeValue, DynamoDBClient } from '@aws-sdk/client-dynamodb';
-import { PartitionKey } from '../constants/Repository.constants';
-import { BaseDto, BaseItem } from '../../job/model/Base/Base';
+import { BaseDto, BaseItem, ListParameters } from '../../job/model/Base/Base';
 /**
  * Generic abstract class for representing repositories interacting with DynamoDB.
  * @export
@@ -10,11 +9,11 @@ import { BaseDto, BaseItem } from '../../job/model/Base/Base';
  * @template T The entity definition.
  * @template U The corresponding DB record attributes for the entity.
  */
-export declare abstract class Repository<T extends BaseItem, U extends BaseDto> {
+export declare abstract class Repository<T extends BaseItem, U extends BaseDto, SK extends Partial<T>> {
     protected dynamoDbClient: DynamoDBClient;
-    protected pk: PartitionKey;
     protected tableName: string;
-    constructor(dynamoDbClient: DynamoDBClient);
+    protected pk: string;
+    constructor(dynamoDbClient: DynamoDBClient, tableName: string, pk: string);
     /**
      * Save item into database.
      * @abstract
@@ -30,22 +29,22 @@ export declare abstract class Repository<T extends BaseItem, U extends BaseDto>
      * @returns {Promise<T | null>} The retrieved record.
      * @memberof Repository
      */
-    findOne(sk: string): Promise<T | null>;
+    findOne(sk: SK): Promise<T | null>;
     /**
      * Get list of records from database.
      * @abstract
      * @returns {Promise<Array<T>>} The retrieved records.
      * @memberof Repository
      */
-    list(): Promise<Array<T>>;
+    list(parameters?: ListParameters<never>): Promise<Array<T>>;
     /**
      * Update record in database.
      * @param {string} sk The sort key used to identify the record.
-     * @param {Partial<T>} input The (partial) record to update.
+     * @param {Partial<T>} input The (partial) record to update (properties used in the SK cannot be updated).
      * @returns {Promise<T | null>} The updated record.
      * @memberof Repository
      */
-    update(sk: string, input: Partial<T>): Promise<T | null>;
+    update(sk: SK, input: Partial<Omit<T, keyof SK>>): Promise<T>;
     /**
      * Delete record from database.
      * @abstract
@@ -53,11 +52,14 @@ export declare abstract class Repository<T extends BaseItem, U extends BaseDto>
      * @returns {void}
      * @memberof Repository
      */
-    delete(sk: string): Promise<void>;
+    delete(sk: T | SK): Promise<void>;
+    batchDelete<TItem extends T | SK>(sk: Array<TItem>): Promise<{
+        unprocessedItems: Array<TItem>;
+    }>;
     /**
      * Convert the DynanomDB attributes of the entity into a DynamoDB item.
      * @abstract
-     * @param {U} attributes The DynamoDB item attributes to convert.
+     * @param {T} item The DynamoDB item attributes to convert.
      * @returns {Record<keyof U, AttributeValue>} The DynamoDB record.
      * @memberof Repository
      */
@@ -70,4 +72,11 @@ export declare abstract class Repository<T extends BaseItem, U extends BaseDto>
      * @memberof Repository
      */
     protected abstract convertToItem(record: U): T;
+    /**
+     * Converts a sort key to a string for querying/putting in Dynamo.
+     * @param {SK} sk
+     * @return {string}
+     * @protected
+     */
+    protected abstract buildSk(sk: T | SK): string;
 }

package/lib/domains/job/repository/JobContextRepository/JobContextRepository.d.ts

@@ -4,12 +4,10 @@
  */
 import { DynamoDBClient } from '@aws-sdk/client-dynamodb';
 import { Repository } from '../AbstractRepository';
-import { CreateJobContextRequest, JobContext, JobContextDto } from '../../model/JobContext/JobContext';
-import { PaginationInfo } from '../../core/pagination/model/PaginationInfo';
-import { PaginationQueryResponse } from '../../core/pagination/model/PaginationQueryResponse';
-export declare class JobContextRepository extends Repository<JobContext, JobContextDto> {
-    protected dynamoDbClient: DynamoDBClient;
-    constructor(dynamoDbClient: DynamoDBClient);
+import { CreateJobContextRequest, JobContext, JobContextDto, JobContextSk, UpdateJobContextRequest } from '../../model/JobContext';
+import { PaginationInfo, PaginationQueryResponse } from '../../core/pagination';
+export declare class JobContextRepository extends Repository<JobContext, JobContextDto, JobContextSk> {
+    constructor(dynamoDbClient: DynamoDBClient, tableName: string, tenant: string);
     create(input: CreateJobContextRequest): Promise<JobContext>;
     /**
      * Get list of records from database.
@@ -18,7 +16,8 @@ export declare class JobContextRepository extends Repository<JobContext, JobCont
      * @memberof Repository
      */
     getContexts(paginationInfo: PaginationInfo): Promise<PaginationQueryResponse<JobContext>>;
-    update(sk: string, input: Partial<JobContext>): Promise<JobContext | null>;
+    update(sk: JobContextSk, input: UpdateJobContextRequest): Promise<JobContext>;
     convertToDynamoDbItem(item: JobContext): JobContextDto;
     convertToItem(dbItem: JobContextDto): JobContext;
+    protected buildSk(sk: JobContextSk): string;
 }