huncwot/db.js

// Copyright 2019 Zaiste & contributors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

const config = require('config');

const Logger = require('./logger');

const pg = require('pg');
const sqorn = require('@sqorn/pg');

const connection = config.get('db');

const pool = new pg.Pool(connection);

pool.connect(error => {
  if (error) {
    Logger.printError(error, 'Data Layer');

    process.exit(1);
  }
});

module.exports = Object.assign(
  sqorn({ pg, pool }),
  {
    execute: async (statement) => {
      const { rows } = await pool.query(statement);
      return rows;
    }
  }
);