fix: add generated property to model properties

Signed-off-by: Muhammad Aaqil <[email protected]>

Author: aaqilniz

lib/discovery.js

@@ -135,6 +135,13 @@ function mixinDiscovery(PostgreSQL) {
         cb(err, results);
       } else {
         results.map(function(r) {
+          // PostgreSQL returns ALWAYS in case the property is generated,
+          // otherwise it returns NEVER
+          if (r.generated === 'ALWAYS') {
+            r.generated = true;
+          } else {
+            r.generated = false;
+          }
           // PostgreSQL accepts float(1) to float(24) as selecting the `real` type,
           // while float(25) to float(53) select `double precision`
           // https://www.postgresql.org/docs/9.4/static/datatype-numeric.html
@@ -166,6 +173,7 @@ function mixinDiscovery(PostgreSQL) {
     if (owner) {
       sql = this.paginateSQL('SELECT table_schema AS "owner", table_name AS "tableName", column_name AS "columnName",'
           + 'data_type AS "dataType", character_maximum_length AS "dataLength", numeric_precision AS "dataPrecision",'
+          + ' is_generated AS "generated",'
           + ' numeric_scale AS "dataScale", is_nullable AS "nullable"'
           + ' FROM information_schema.columns'
           + ' WHERE table_schema=\'' + owner + '\''
@@ -175,6 +183,7 @@ function mixinDiscovery(PostgreSQL) {
       sql = this.paginateSQL('SELECT current_schema() AS "owner", table_name AS "tableName",'
           + ' column_name AS "columnName",'
           + ' data_type AS "dataType", character_maximum_length AS "dataLength", numeric_precision AS "dataPrecision",'
+          + ' is_generated AS "generated",'
           + ' numeric_scale AS "dataScale", is_nullable AS "nullable"'
           + ' FROM information_schema.columns'
           + (table ? ' WHERE table_name=\'' + table + '\'' : ''),

test/schema.sql

@@ -319,7 +319,8 @@ CREATE TABLE inventory (
     product_id character varying(20),
     location_id character varying(20),
     available integer,
-    total integer
+    total integer,
+    token integer GENERATED ALWAYS AS (total * 1000) STORED
 );