@or-sdk/pgsql 0.22.5-pgsql.0 → 0.23.0

package/src/utils/getInsertKeys.ts

@@ -0,0 +1,9 @@
+import { Row } from '../types';
+
+function getInsertKeys(rows: Row[]): string[] {
+  const allKeys = {};
+  rows.forEach(x => Object.assign(allKeys, x));
+  return Object.keys(allKeys);
+}
+
+export default getInsertKeys;

package/src/utils/getInsertQueries.ts

@@ -0,0 +1,18 @@
+import { Row } from '../types';
+import { encodeValue, getInsertKeys } from '../utils';
+import _ from 'lodash';
+
+function getInsertQueries(schema: string, table: string, insertRows: Row[], chunkSize: number): string[] {
+  const rows = _.isArray(insertRows) ? insertRows : [insertRows];
+
+  const keys = getInsertKeys(rows);
+  const stringifyRow = (row: Row) => `${keys.map(key => encodeValue(row[key])).join(', ')}`;
+
+  return _.chain(rows)
+    .chunk(chunkSize)
+    .map((chunk) => `insert into ${schema}.${table} (${keys.join(', ')})
+values (${chunk.map(stringifyRow).join('), (')});`)
+    .value();
+}
+
+export default getInsertQueries;

package/src/utils/getListTablesQuery.ts

@@ -0,0 +1,3 @@
+export default (schema: string) => `SELECT table_name FROM information_schema.tables
+      WHERE table_schema = '${schema}'
+      and table_type = 'BASE TABLE';`;

package/src/utils/getSelectAllCountQuery.ts

@@ -0,0 +1 @@
+export default (query: string) => `select count(*) from (${query}) x`;

package/src/utils/getSelectAllQuery.ts

@@ -0,0 +1 @@
+export default (query: string, offset: number, chunkSize: number) => `${query} offset ${offset} limit ${chunkSize}`;

package/src/utils/index.ts

@@ -3,3 +3,19 @@
  */
 
 export { default as extractDatabaseNames } from './extractDatabaseNames';
+export { default as getInsertKeys } from './getInsertKeys';
+export { default as encodeValue } from './encodeValue';
+export { default as splitPrimaryKeys } from './splitPrimaryKeys';
+export { default as getListTablesQuery } from './getListTablesQuery';
+export { default as createSchemaQuery } from './createSchemaQuery';
+export { default as getDropSchemaQuery } from './getDropSchemaQuery';
+export { default as getCreateTableQuery } from './getCreateTableQuery';
+export { default as getDropTableQuery } from './getDropTableQuery';
+export { default as getAddColumnsQuery } from './getAddColumnsQuery';
+export { default as getInsertQueries } from './getInsertQueries';
+export { default as getSelectAllCountQuery } from './getSelectAllCountQuery';
+export { default as getSelectAllQuery } from './getSelectAllQuery';
+export { default as getEditRowQuery } from './getEditRowQuery';
+export { default as getDeleteRowsQuery } from './getDeleteRowsQuery';
+export { default as getGenerateTableSchemaQuery } from './getGenerateTableSchemaQuery';
+export { default as getGetPrimaryKeysQuery } from './getGetPrimaryKeysQuery';

package/src/utils/splitPrimaryKeys.ts

@@ -0,0 +1,3 @@
+import _ from 'lodash';
+
+export default (primaryKeys: string) => primaryKeys.split(',').map(x => _.trim(x));