fixed Process model

andrefs · Oct 18, 2023 · 4f39647 · 4f39647
1 parent 82a1eab
commit 4f39647
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/src/models/Process.ts b/src/models/Process.ts
@@ -1,9 +1,9 @@
-import { Types } from 'mongoose';
+import { Types, Document } from 'mongoose';
 import { Resource } from './Resource';
 import { Triple, TripleClass } from './Triple';
 import { humanize } from 'humanize-digest';
 import { Domain } from './Domain';
-import { PathClass, Path, PathDocument } from './Path';
+import { Path, PathDocument } from './Path';
 import { ProcessTriple } from './ProcessTriple';
 import {
   prop,
@@ -52,6 +52,9 @@ class ParamsClass {
 }
 
 class ProcessClass {
+  createdAt!: Date;
+  updatedAt!: Date;
+
   @prop({ required: true, index: true, unique: true })
   public pid!: string;
 
@@ -131,7 +134,7 @@ class ProcessClass {
     return paths;
   }
 
-  public async extendPathsWithExistingTriples(paths: PathClass[]) {
+  public async extendPathsWithExistingTriples(paths: PathDocument[]) {
     for (const path of paths) {
       const newPathObjs = [];
       const toDelete = new Set();
@@ -363,4 +366,6 @@ const Process = getModelForClass(ProcessClass, {
   schemaOptions: { timestamps: true },
 });
 
-export { Process, ProcessClass };
+type ProcessDocument = ProcessClass & Document;
+
+export { Process, ProcessClass, ProcessDocument };
diff --git a/src/models/Resource.ts b/src/models/Resource.ts
@@ -1,4 +1,4 @@
-import { Types } from 'mongoose';
+import { Types, Document } from 'mongoose';
 import { urlValidator, WorkerError } from '@derzis/common';
 import { Domain } from '@derzis/models';
 import { Path, PathDocument } from './Path';
@@ -22,6 +22,9 @@ class CrawlId {
 }
 
 class ResourceClass {
+  createdAt!: Date;
+  updatedAt!: Date;
+
   @prop({ required: true, validate: urlValidator })
   public url!: string;
 
@@ -220,5 +223,6 @@ class ResourceClass {
 const Resource = getModelForClass(ResourceClass, {
   schemaOptions: { timestamps: true },
 });
+type ResourceDocument = ResourceClass & Document;
 
-export { Resource, ResourceClass };
+export { Resource, ResourceClass, ResourceDocument };