All files / src/driver-integrations pgp.ts

100% Statements 22/22
100% Branches 9/9
100% Functions 12/12
100% Lines 22/22

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93      2x         68x 8x 7x 1x     7x             68x         13x   9x       68x         6x   5x       68x         7x   6x       68x         7x     6x               68x         5x   4x       68x                      
import { IModel, ICollection } from '../core';
import { ICreateForDriverOptions, ICoreIntegratedDriver } from './index';
 
export const createForPGP = ({
  core,
  db,
  logError
}: ICreateForDriverOptions): ICoreIntegratedDriver => {
  const defaultErrorHandler = (err: Error) => {
    if (!(err.name === 'QueryResultError')) {
      if (logError) {
        logError(err);
      }
    }
    throw err;
  };
 
  /* ------------------------------------------------------------------------*/
  /* Query functions --------------------------------------------------------*/
  /* ------------------------------------------------------------------------*/
 
  const one = <T extends IModel>(
    query: string,
    values?: object,
    errorHandler = defaultErrorHandler
  ): Promise<T> => {
    return db
      .many(query, values)
      .then((rows: any) => core.createOneFromDatabase(rows))
      .catch(errorHandler);
  };
 
  const oneOrNone = <T extends IModel>(
    query: string,
    values?: object,
    errorHandler = defaultErrorHandler
  ): Promise<T | void> => {
    return db
      .any(query, values)
      .then((rows: any) => core.createOneOrNoneFromDatabase(rows))
      .catch(errorHandler);
  };
 
  const many = <T extends ICollection<IModel>>(
    query: string,
    values?: object,
    errorHandler = defaultErrorHandler
  ): Promise<T> => {
    return db
      .any(query, values)
      .then((rows: any) => core.createManyFromDatabase(rows))
      .catch(errorHandler);
  };
 
  const any = <T extends ICollection<IModel>>(
    query: string,
    values?: object,
    errorHandler = defaultErrorHandler
  ): Promise<T | void> => {
    return db
      .result(query, values)
      .then((result: any) =>
        core.createAnyFromDatabase(
          result.rows,
          result.fields[0].name.split('#')[0]
        )
      )
      .catch(errorHandler);
  };
 
  const none = (
    query: string,
    values?: object,
    errorHandler = defaultErrorHandler
  ): Promise<void> => {
    return db
      .none(query, values)
      .then(() => null)
      .catch(errorHandler);
  };
 
  return Object.assign({}, core, {
    // Query Functions
    one,
    oneOrNone,
    many,
    any,
    none,
    // provide direct access to db
    db
  });
};