feat: custom primary key generation

Refs: https://github.com/metarhia/Contracts/blob/master/doc/Database.md Refs: #4 Closes: #52
metarhia · Oct 9, 2020 · 996f728 · 996f728
1 parent ca0dd04
commit 996f728
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/application/schemas/CompanyCity.js b/application/schemas/CompanyCity.js
@@ -2,5 +2,5 @@
   company: 'Company',
   city: 'City',
 
-  companyCity: { primary: ['Company', 'City'] },
+  companyCity: { primary: ['company', 'city'] },
 });
diff --git a/lib/schema-pg.js b/lib/schema-pg.js
@@ -70,12 +70,17 @@ const foreignKey = (entityName, def) => {
   );
 };
 
-const primaryKey = entityName => {
-  const fieldName = toLowerCamel(entityName) + 'Id';
-  const constraint = `"pk${entityName}" PRIMARY KEY ("${fieldName}")`;
+const primaryCustom = (entityName, fields) => {
+  const fieldNames = `"${fields.map(name => name + 'Id').join('", "')}"`;
+  const constraint = `"pk${entityName}" PRIMARY KEY (${fieldNames})`;
   return `ALTER TABLE "${entityName}" ADD CONSTRAINT ${constraint};`;
 };
 
+const primaryKey = entityName => {
+  const fieldName = toLowerCamel(entityName);
+  return primaryCustom(entityName, [fieldName]);
+};
+
 class PgSchema extends DatabaseSchema {
   async preprocessEntity(name) {
     const { entity } = this.entities.get(name);
@@ -128,6 +133,10 @@ class PgSchema extends DatabaseSchema {
         if (kind === DB_RELATION) idx.push(foreignKey(name, def));
       }
       if (def.unique || def.index) idx.push(createIndex(name, def));
+      if (def.primary) {
+        idx[0] = primaryCustom(name, def.primary);
+        sql.splice(1, 1);
+      }
     }
     sql[sql.length - 1] = sql[sql.length - 1].slice(0, -1);
     sql.push(');');