2 files changed, 6 insertions, 17 deletions
diff --git a/services/votes/votes.class.ts b/services/votes/votes.class.ts
deleted file mode 100644
index 3220ee7..0000000
--- a/services/votes/votes.class.ts
+++ /dev/null
@@ -1,14 +0,0 @@
-import PollModel from '../../models/polls/poll.model';
-import { PollSchema } from '../../models/polls/poll.schema';
-
-export default class Votes {
-  async create(data: any, params: any): Promise<PollSchema | null> {
-    const poll = await PollModel.findById(params.route.id);
-    if (poll) {
-      const updatedPoll = await poll.vote(params.user._id, data.which);
-      return updatedPoll.toObject();
-    }
-    return null;
-  }
-}
-
diff --git a/services/votes/votes.service.ts b/services/votes/votes.service.ts
index 3947d9b..81f767c 100644
--- a/services/votes/votes.service.ts
+++ b/services/votes/votes.service.ts
@@ -1,10 +1,13 @@
 import { Application } from '@feathersjs/express';
-import Votes from './votes.class';
+import service from 'feathers-mongoose';
+import Model from '../../models/votes/vote.model';
 
 import hooks from './votes.hooks';
 
+const VoteService = service({ Model });
+
 export default (app: Application): void => {
-  app.use('/polls/:id/votes/', new Votes());
-  app.service('/polls/:id/votes/').hooks(hooks);
+  app.use('/votes/', VoteService);
+  app.service('votes').hooks(hooks);
 };