Added support for Graph Insert

Author: NotAmaan

src/index.js

@@ -62,6 +62,8 @@ class Service {
     this.allowedEager = options.allowedEager || '[]'
     this.namedEagerFilters = options.namedEagerFilters
     this.eagerFilters = options.eagerFilters
+    this.allowedInsert = options.allowedInsert
+    this.insertGraphOptions = options.insertGraphOptions
   }
 
   extend (obj) {
@@ -338,8 +340,15 @@ class Service {
   }
 
   _create (data, params) {
-    return this._createQuery(params)
-      .insert(data, this.id)
+    let q = this._createQuery(params)
+
+    if (this.allowedInsert) {
+      q.allowInsert(this.allowedInsert)
+      q.insertGraph(data, this.insertGraphOptions)
+    } else {
+      q.insert(data, this.id)
+    }
+    return q
       .then(row => {
         let id = null
 

test/client.js

@@ -0,0 +1,28 @@
+import { Model } from 'objection'
+
+export default class Client extends Model {
+  static tableName = 'clients'
+  static jsonSchema = {
+    type: 'object',
+    required: ['name'],
+
+    properties: {
+      id: { type: 'integer' },
+      companyId: { type: 'integer' },
+      name: { type: 'string' }
+    }
+  }
+
+  static get relationMappings () {
+    return {
+      company: {
+        relation: Model.BelongsToOneRelation,
+        modelClass: require('./company'),
+        join: {
+          from: 'clients.companyId',
+          to: 'companies.id'
+        }
+      }
+    }
+  }
+}

test/company.js

@@ -32,6 +32,14 @@ export default class Company extends Model {
         from: 'companies.id',
         to: 'employees.companyId'
       }
+    },
+    clients: {
+      relation: Model.HasManyRelation,
+      modelClass: path.join(__dirname, '/client'),
+      join: {
+        from: 'companies.id',
+        to: 'clients.companyId'
+      }
     }
   }
 }

test/index.test.js

@@ -15,6 +15,7 @@ import PeopleRoomsCustomIdSeparator from './people-rooms-custom-id-separator'
 import Company from './company'
 import { Model } from 'objection'
 import Employee from './employee'
+import Client from './client'
 
 const db = knex({
   client: 'sqlite3',
@@ -68,7 +69,7 @@ const app = feathers()
       model: Company,
       id: 'id',
       events: ['testing'],
-      allowedEager: 'ceos',
+      allowedEager: '[ceos, clients]',
       namedEagerFilters: {
         notSnoop (builder) {
           return builder.whereNot('name', 'Snoop')
@@ -81,7 +82,8 @@ const app = feathers()
             return builder.where('age', '<', '25')
           }
         }
-      ]
+      ],
+      allowedInsert: 'clients'
     })
   )
   .use(
@@ -91,6 +93,13 @@ const app = feathers()
       allowedEager: 'company'
     })
   )
+  .use(
+    '/clients',
+    service({
+      model: Client,
+      allowedEager: 'company'
+    })
+  )
 
 let people = app.service('people')
 let peopleRooms = app.service('people-rooms')
@@ -158,6 +167,16 @@ function clean (done) {
             table.integer('companyId').references('companies.id')
             table.string('name')
           })
+      })
+    })
+    .then(() => {
+      return db.schema.dropTableIfExists('clients').then(() => {
+        return db.schema
+          .createTable('clients', table => {
+            table.increments('id')
+            table.integer('companyId').references('companies.id')
+            table.string('name')
+          })
           .then(() => done())
       })
     })
@@ -457,6 +476,36 @@ describe('Feathers Objection Service', () => {
     })
   })
 
+  describe('Graph Insert Queries', () => {
+    before(async () => {
+      await companies.remove(null)
+      await companies
+        .create([
+          {
+            name: 'Google',
+            clients: [
+              {
+                name: 'Dan Davis'
+              },
+              {
+                name: 'Ken Patrick'
+              }
+            ]
+          },
+          {
+            name: 'Apple'
+          }
+        ])
+    })
+
+    it('allows insertGraph queries', () => {
+      return companies.find({ query: { $eager: 'clients' } }).then(data => {
+        expect(data[0].clients).to.be.an('array')
+        expect(data[0].clients).to.have.lengthOf(2)
+      })
+    })
+  })
+
   describe('$like method', () => {
     beforeEach(async () => {
       await people