ramiel / mongoose-sequence

Sequence and autoincrement handling for mongoose
GNU General Public License v2.0
284 stars 56 forks source link
auto-increment counter mongodb mongoose

Mongoose sequence plugin

Build Status Coverage Status Donate npm

Hello everybody. Look, I have no more time to maintain this old library. If anybody wants to take over is very welcome and I'm available to review PRs. For the moment I have no time, no will to work on this library.

This plugin lets you create fields which autoincrement their value:

This increment can be:

Multiple counters can be set for a collection.

Migrating

From version 3 to version 4

Version 3 is now deprecated. In order to migrate to the new version the only change you need to do is to pass mongoose to the required module as explained in the requiring section.

From version 4 to version 5

An important fix about scoped counters is not backward compatible. You cannot use version 5 with scoped counters already present on your DB.

Requisites

This plugin needs mongoose version 4.0.0 or above.

Installation

npm install --save mongoose-sequence

Requiring

You must pass your DB connection instance for this plugin to work. This is needed in order to create a collection on your DB where to store increment references.

const mongoose = require('mongoose');
const AutoIncrement = require('mongoose-sequence')(mongoose);

If you use different connections you must include it this way

const mongoose = require('mongoose');
const AutoIncrementFactory = require('mongoose-sequence');

const connection = await mongoose.createConnection('mongodb://...');

const AutoIncrement = AutoIncrementFactory(connection);

Global sequences

Let's say you want to have an id field in your user collection which has a unique auto-incremented value.

The user schema is something like this:

UserSchema = mongoose.Schema({
  name: String,
});

mongoose.model('User', UserSchema);

You don't need to define the id field in your schema because the plugin automatically sets it for you. The only thing you have to do is to call:

UserSchema.plugin(AutoIncrement, { inc_field: 'id' });

after requiring the plugin.

Every time a new user is created, the id field will have an incremented number. The operation is atomic and is based on this specification. A commodity collection named counters is created for you. You can override the name of this collection but we will see this later with the options.

If you want to increment the _id field which is special to mongoose, you have to explicitly specify it as a Number and tell mongoose to not interfere:

UserSchema = mongoose.Schema(
  {
    _id: Number,
    name: String,
  },
  { _id: false },
);
UserSchema.plugin(AutoIncrement);

In this case you don't have to specify inc_field because the default value is _id

Not automatic sequences

Let's say our user model has a rank field which gives the rank of the user in a tournament. So it saves the arrival order of a user to the end of our amazing game. This field is of course a sequence but has to be incremented every time an event occurs. Because we have concurrent access to our database we want to be sure that the increment of this counter happens safely. Let's start by modifying our schema:

UserSchema = mongoose.Schema({
  name: String,
  rank: Number,
});

This time we specified explicitly the field rank. There is no difference between defining and omitting the field specification. The only constraint is that the field has to be of type Number, otherwise the plugin will raise an error. So, let's tell the plugin we want the rank field to be a safe counter:

UserSchema.plugin(AutoIncrement, { inc_field: 'rank', disable_hooks: true });

We specified disable_hooks. This avoids the field being incremented when a new document is saved. So, how to increment this field? Your models have a new method: setNext. You must specify which sequence you want to increment and a callback. Here's an example:

User.findOne({ name: 'George' }, function (err, user) {
  user.setNext('rank', function (err, user) {
    if (err) console.log('Cannot increment the rank because ', err);
  });
});

You noticed that the method setNext takes, as argument, the counter field name. It is possible to give a name to the counter and use it as reference. For the previous example we can define the counter like this:

UserSchema.plugin(AutoIncrement, {
  id: 'rank_counter',
  inc_field: 'rank',
  disable_hooks: true,
});

and then use:

user.setNext('rank_counter', function (err, user) {
  //...
});

So, if you do not specify the id, the field name is used. Even if you're not forced to specify an id, its use is strongly suggested. This is because if you have two different counters, which refers to fields with the same name, they will collide and incrementing one will increment the other too. Counters are not bound to the schema they refer to, so two counters for two different schemas can collide. So use a unique id to be sure to avoid collision. In the case of a collision the plugin will raise an error.

As we will see, the use of an id for the counter is mandatory when you're defining a scoped counter.

NOTE: When you call setNext the document is automatically saved. This behavior has changed since version 3.0.0. If you use a prior version you have to call save by yourself.

Advanced

Scoped counters

Let's say our users are organized by country and city and we want to save the inhabitant_number according to these two pieces of information.
The schema is like this:

UserSchema = mongoose.Schema({
  name: String,
  country: String,
  city: String,
  inhabitant_number: Number,
});

Every time a new Parisian is added, the count of Parisians has to be incremented. The inhabitants of New York must not interfere, and have their separate counting. We should define a scoped counter which increments the counter depending on the value of other fields.

UserSchema.plugin(AutoIncrement, {
  id: 'inhabitant_seq',
  inc_field: 'inhabitant_number',
  reference_fields: ['country', 'city'],
});

Notice that we have to use an id for our sequence, otherwise the plugin will raise an error. Now save a new user:

var user = new User({
  name: 'Patrice',
  country: 'France',
  city: 'Paris',
});
user.save();

This user will have the inhabitant_number counter set to 1. If now we add a new inhabitant from New York, this will have its counter set to 1 also, because the counter is referred to by the value of the fields country and city.

If we want to increment this counter manually we have to specify the id of the sequence in the setNext method:

user.setNext('inhabitant_seq', function (err, user) {
  user.inhabitant_number; // the counter value
});

Of course this example is a bit forced and this is for sure not the perfect use case. The fields country and city have to be present and must not change during the life of the document because no automatic hooks are set on the change of those values. But there are situations when you want a similar behavior.

Reset a counter

It's possible to programmatically reset a counter through the Model's static method counterReset(id, reference, callback). The method takes these parameters:

Some examples:

Model.counterReset('counter_id', function (err) {
  // Now the counter is 0
});

Model.counterReset('inhabitants_id', function (err) {
  // If this is a referenced field, now all the counters are 0
});

Model.counterReset(
  'inhabitants_id',
  { country: 'France', city: 'Paris' },
  function (err) {
    // If this is a referenced field, only the counter for Paris/France is 0
  },
);

Nested fields

It is possible to define a nested field as counter, using . as the path separator:

NestedSchema = new mongoose.Schema({
  name: { type: String },
  registration: {
    date: { type: Date },
    number: { type: Number },
  },
});

NestedSchema.plugin(AutoIncrement, {
  id: 'user_registration_seq',
  inc_field: 'registration.number',
});

Options

This plugin accepts the following options:

Notes

When using insertMany the plugin won't increment the counter because the needed hooks are not called. If you need to create several documents at once, use create instead and pass an array of documents (refer to #7).