Skip to content

Commit 23ee57d

Browse files
committed
feat(orap): update ttl type from number to Milliseconds
1 parent 50b866b commit 23ee57d

File tree

3 files changed

+17
-15
lines changed

3 files changed

+17
-15
lines changed

packages/orap/src/flow/task.ts

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import type { Milliseconds } from '@ora-io/utils'
12
import { alphabetHex, memoryStore, randomStr } from '@ora-io/utils'
23
import type { Context, TaskRaplized } from '../task'
34
import { StoreManager } from '../store'
@@ -15,15 +16,15 @@ const defaultHandleFn: HandleFn = () => {
1516
}
1617

1718
const defaultToKeyFn: ToKeyFn = _ => randomStr(8, alphabetHex)
18-
export interface TaskFlowTTL { taskTtl: number; doneTtl: number }
19+
export interface TaskFlowTTL { taskTtl: Milliseconds; doneTtl: Milliseconds }
1920

2021
// TODO: add 'Failed-Task:' ?
2122
export class TaskFlow implements Flow {
2223
sm: StoreManager = new StoreManager(memoryStore())
2324
taskPrefix: Prefix = 'Task:'
2425
donePrefix: Prefix = 'Done-Task:'
25-
taskTtl?: number
26-
doneTtl?: number
26+
taskTtl?: Milliseconds
27+
doneTtl?: Milliseconds
2728
toKeyFn: ToKeyFn = defaultToKeyFn
2829
handleFn: HandleFn = defaultHandleFn
2930
successFn: HandleResultFn = defaultSuccessFn
@@ -60,10 +61,10 @@ export class TaskFlow implements Flow {
6061
return this
6162
}
6263

63-
ttl(TTLs: number): this
64+
ttl(TTLs: Milliseconds): this
6465
ttl(TTLs: TaskFlowTTL): this
65-
ttl(taskTtl: number, doneTtl: number): this
66-
ttl(TTLs: TaskFlowTTL | number, doneTtl?: number): this {
66+
ttl(taskTtl: Milliseconds, doneTtl: Milliseconds): this
67+
ttl(TTLs: TaskFlowTTL | Milliseconds, doneTtl?: Milliseconds): this {
6768
if (typeof TTLs === 'number') {
6869
this.taskTtl = TTLs
6970
this.doneTtl = doneTtl ?? TTLs

packages/orap/src/task/storable.ts

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import type { Awaitable, Constructor } from '@ora-io/utils'
1+
import type { Awaitable, Constructor, Milliseconds } from '@ora-io/utils'
22
import type { StoreManager } from '../store/storemanager'
33
import type { Context } from './context'
44
import { TaskBase } from './base'
@@ -9,8 +9,8 @@ export abstract class TaskStorable extends TaskBase {
99
// overwrite these for store key customize
1010
static readonly taskPrefix: string = 'Task:'
1111
static readonly taskPrefixDone: string = 'Done-Task:'
12-
static readonly taskTtl: number | undefined = undefined
13-
static readonly taskTtlDone: number | undefined = undefined
12+
static readonly taskTtl: Milliseconds | undefined = undefined
13+
static readonly taskTtlDone: Milliseconds | undefined = undefined
1414

1515
getTaskPrefix(_context?: Context): Awaitable<string> {
1616
return (this.constructor as TypeofTaskStorable).taskPrefix
@@ -20,19 +20,19 @@ export abstract class TaskStorable extends TaskBase {
2020
return (this.constructor as TypeofTaskStorable).taskPrefixDone
2121
}
2222

23-
get taskTtl(): number | undefined {
23+
get taskTtl(): Milliseconds | undefined {
2424
return (this.constructor as TypeofTaskStorable).taskTtl
2525
}
2626

27-
get taskTtlDone(): number | undefined {
27+
get taskTtlDone(): Milliseconds | undefined {
2828
return (this.constructor as TypeofTaskStorable).taskTtlDone
2929
}
3030

31-
getTaskTtl(_context?: Context): number | undefined {
31+
getTaskTtl(_context?: Context): Milliseconds | undefined {
3232
return (this.constructor as TypeofTaskStorable).taskTtl
3333
}
3434

35-
getTaskTtlDone(_context?: Context): number | undefined {
35+
getTaskTtlDone(_context?: Context): Milliseconds | undefined {
3636
return (this.constructor as TypeofTaskStorable).taskTtlDone
3737
}
3838

packages/orap/src/task/verse.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import type { Milliseconds } from '@ora-io/utils'
12
import { composeFns, isJsonString, isString, stripPrefix } from '@ora-io/utils'
23
import type { TaskFlow } from '../flow'
34
import { HandleSuccessMiddleware } from '../middlewares/HandleSuccessMiddleware'
@@ -28,11 +29,11 @@ export class TaskRaplized extends TaskStorable {
2829
: await this.flow.donePrefix(this.flow.ctx)
2930
}
3031

31-
get taskTtl(): number | undefined {
32+
get taskTtl(): Milliseconds | undefined {
3233
return this.flow.taskTtl
3334
}
3435

35-
get taskTtlDone(): number | undefined {
36+
get taskTtlDone(): Milliseconds | undefined {
3637
return this.flow.doneTtl
3738
}
3839

0 commit comments

Comments
 (0)