From f8ebf581f8272615f48ce355c6b2cbf8248c73bd Mon Sep 17 00:00:00 2001
From: eug-vs <eug-vs@keemail.me>
Date: Wed, 2 Dec 2020 15:26:25 +0300
Subject: feat: export client class

---
 lib/event.model.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'lib/event.model.ts')

diff --git a/lib/event.model.ts b/lib/event.model.ts
index 62675d4..bea533d 100644
--- a/lib/event.model.ts
+++ b/lib/event.model.ts
@@ -2,7 +2,8 @@ import { model, Schema, Model } from 'mongoose';
 import cron from 'cron';
 import createEventSchema, { EventDocument } from './event.schema';
 import { LogDocument } from './log.schema';
-import LogModel from './log.model';
+import { LogModel } from './log.model';
+import Connection from './connection';
 
 export interface Event<Context> extends EventDocument<Context> {
   log(message: string): Promise<LogDocument>;
@@ -20,8 +21,9 @@ export interface EventModel<Context> extends Model<Event<Context>> {
 
 const { CronJob } = cron;
 
-const createEventModel = <Context>(name: string, contextSchema: Schema): EventModel<Context> => {
+const createEventModel = <Context>(connection: Connection, contextSchema: Schema): EventModel<Context> => {
   const schema = createEventSchema(contextSchema);
+  const LogModel: LogModel = connection.models['Log'];
 
   // Schema methods
   schema.method('log', function (message: string) {
@@ -93,7 +95,7 @@ const createEventModel = <Context>(name: string, contextSchema: Schema): EventMo
     this.nextRunAt = this.computeNextRunAt();
   });
 
-  return model<Event<Context>, EventModel<Context>>(name, schema);
+  return connection.model<Event<Context>, EventModel<Context>>('Event', schema);
 };
 
 
-- 
cgit v1.2.3