|
3 | 3 | */ |
4 | 4 |
|
5 | 5 | 'use strict'; |
6 | | -<% if (filters.mongoose) { %> |
7 | | -var <%= classedName %> = require('./<%= name %>.model');<% } %><% if (filters.sequelize) { %> |
8 | | -var <%= classedName %> = require('../../sqldb').<%= classedName %>;<% } %> |
9 | | - |
10 | | -exports.register = function(socket) {<% if (filters.sequelize) { %> |
11 | | - <%= classedName %>.hook('afterCreate', function(doc, fields, fn) { |
12 | | - onSave(socket, doc); |
13 | | - fn(null); |
14 | | - });<% } %> |
15 | | - <% if (filters.mongoose) { %><%= classedName %>.schema.post('save', function(doc) {<% } |
16 | | - if (filters.sequelize) { %><%= classedName %>.hook('afterUpdate', function(doc, fields, fn) {<% } %> |
17 | | - onSave(socket, doc);<% if (filters.sequelize) { %> |
18 | | - fn(null);<% } %> |
19 | | - }); |
20 | | - <% if (filters.mongoose) { %><%= classedName %>.schema.post('remove', function(doc) {<% } |
21 | | - if (filters.sequelize) { %><%= classedName %>.hook('afterDestroy', function(doc, fields, fn) {<% } %> |
22 | | - onRemove(socket, doc);<% if (filters.sequelize) { %> |
23 | | - fn(null);<% } %> |
24 | | - }); |
| 6 | + |
| 7 | +var <%= classedName %>Events = require('./<%= name %>.events'); |
| 8 | + |
| 9 | +// Model events to emit |
| 10 | +var events = ['save', 'remove']; |
| 11 | + |
| 12 | +exports.register = function(socket) { |
| 13 | + // Bind model events to socket events |
| 14 | + for (var i = 0, eventsLength = events.length; i < eventsLength; i++) { |
| 15 | + var event = events[i]; |
| 16 | + var listener = createListener('<%= name %>:' + event, socket); |
| 17 | + |
| 18 | + <%= classedName %>Events.on(event, listener); |
| 19 | + socket.on('disconnect', removeListener(event, listener)); |
| 20 | + } |
25 | 21 | }; |
26 | 22 |
|
27 | | -function onSave(socket, doc, cb) { |
28 | | - socket.emit('<%= name %>:save', doc); |
| 23 | + |
| 24 | +function createListener(event, socket) { |
| 25 | + return function(doc) { |
| 26 | + socket.emit(event, doc); |
| 27 | + }; |
29 | 28 | } |
30 | 29 |
|
31 | | -function onRemove(socket, doc, cb) { |
32 | | - socket.emit('<%= name %>:remove', doc); |
| 30 | +function removeListener(event, listener) { |
| 31 | + return function() { |
| 32 | + <%= classedName %>Events.removeListener(event, listener); |
| 33 | + }; |
33 | 34 | } |
0 commit comments