Model functions?

[schealex]

Hey there,

I need to be able to use model document functions as documented here: https://dynamoosejs.com/guide/Model/#modelmethodsdocumentsetname-function

is there a way of doing it using this?

PS: Thanks for this great library!

[hardyscc]

@schealex you can do the model initialization under the module directly like that

@Module({
  imports: [
    DynamooseModule.forFeature([
      {
        name: 'notification',
        schema: NotificationSchema,
      },
    ]),
  ],
  providers: [NotificationService],
  controllers: [NotificationController],
})
export class NotificationModule implements OnModuleInit {
  constructor(
    @InjectModel('notification')
    private readonly notificationModel: any,
  ) {}

  onModuleInit() {
    this.notificationModel.methods.document.set('logMsg', (msg: string) => {
      console.log(msg);
    });
  }
}

later on, u can call this method as normal

@Injectable()
export class NotificationService {
  constructor(
    @InjectModel('notification')
    private readonly model: Model<Notification, NotificationKey>,
  ) {}

  async findOne(key: NotificationKey) {
    const notification = await this.model.get(key);
    // @ts-ignore
    notification.logMsg('hello');
    return notification;
  }