bug fix: #1

README.md

@@ -39,16 +39,20 @@ $ npm i --save nest-schedule
 
 ```typescript
 import { Injectable } from '@nestjs/common';
-import { Schedule } from 'nest-schedule';
+import { Schedule, NestSchedule } from 'nest-schedule';
 
 @Injectable()
-export class ScheduleService {  
+export class ScheduleService extends NestSchedule {  
+  constructor() {
+    super();
+  }
+
   @Schedule({
     cron: '0 0 2 * *',
     startTime: new Date(), 
     endTime: new Date(new Date().getTime() + 24 * 60 * 60 * 1000)
   })
-  syncData() {
+  async syncData() {
     console.log('syncing data ...');
   }
 
@@ -67,7 +71,7 @@ export class ScheduleService {
   intervalJob() {
     console.log('interval job');
 
-    // if you want to cancel the schedule job, you should return true;
+    // if you want to cancel the job, you should return true;
     return true;
   }
 }

lib/NestSchedule.ts

@@ -0,0 +1,58 @@
+import { NEST_SCHEDULE_JOB_KEY } from "./constants";
+import * as schedule from "node-schedule";
+
+export class NestSchedule {
+  private readonly jobs;
+  private readonly timers = {};
+
+  constructor() {
+    this.jobs = Reflect.getMetadata(NEST_SCHEDULE_JOB_KEY, new.target.prototype);
+    this.init();
+  }
+
+  private init() {
+    if (this.jobs) {
+      this.jobs.forEach(job => {
+        if (job.cron) {
+          job = schedule.scheduleJob({
+            startTime: job.startTime,
+            endTime: job.endTime,
+            rule: job.cron
+          }, async () => {
+            let result = this[job.key]();
+            if (result instanceof Promise) {
+              result = await result;
+            }
+            if (result && job) {
+              job.cancel();
+            }
+          });
+        }
+        if (job.interval) {
+          this.timers[job.key] = setInterval(async () => {
+            let result = this[job.key]();
+            if (result instanceof Promise) {
+              result = await result;
+            }
+            if (result && this.timers[job.key]) {
+              clearInterval(this.timers[job.key]);
+              delete this.timers[job.key];
+            }
+          }, job.interval);
+        }
+        if (job.timeout) {
+          this.timers[job.key] = setTimeout(async () => {
+            let result = this[job.key]();
+            if (result instanceof Promise) {
+              result = await result;
+            }
+            if (result && this.timers[job.key]) {
+              clearTimeout(this.timers[job.key]);
+              delete this.timers[job.key];
+            }
+          }, job.timeout);
+        }
+      });
+    }
+  }
+}

lib/Schedule.ts

@@ -0,0 +1,20 @@
+import "reflect-metadata";
+import { NEST_SCHEDULE_JOB_KEY } from "./constants";
+
+export interface Options {
+  cron?: string;
+  interval?: number;
+  timeout?: number;
+  startTime?: Date;
+  endTime?: Date;
+}
+
+export const Schedule = (options: Options) => (target, key, descriptor) => {
+  let jobs = Reflect.getMetadata(NEST_SCHEDULE_JOB_KEY, target);
+  if (!jobs) {
+    jobs = [];
+  }
+
+  jobs.push({ ...options, key });
+  Reflect.defineMetadata(NEST_SCHEDULE_JOB_KEY, jobs, target);
+};

lib/constants.ts

@@ -0,0 +1 @@
+export const NEST_SCHEDULE_JOB_KEY = "nest_schedule_jobs";

lib/index.ts

@@ -1 +1,3 @@
-export * from "./decorators/Schedule";
+export * from "./Schedule";
+export * from "./NestSchedule";
+export * from "./constants";

package.json

@@ -1,6 +1,6 @@
 {
   "name": "nest-schedule",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Nest - modern, fast, powerful node.js web framework (@schedule)",
   "author": "Miaowing <[email protected]>",
   "license": "MIT",