const Sequelize = require('sequelize');
const sequelize = new Sequelize('hello_world', 'benchmarkdbuser', 'benchmarkdbpass', {
  host: 'tfb-database',
  dialect: 'postgres',
  logging: false
});

const { DataTypes } = Sequelize;
const Worlds = sequelize.define('World', {
  id: {
    type: DataTypes.INTEGER,
    primaryKey: true
  },
  randomNumber: {
    type: DataTypes.INTEGER,
    field: 'randomnumber'
  },
}, {
    timestamps: false,
    freezeTableName: true
  });

const Fortunes = sequelize.define('Fortune', {
  id: {
    type: DataTypes.INTEGER,
    primaryKey: true
  },
  message: { type: DataTypes.STRING }
}, {
    timestamps: false,
    freezeTableName: true
  });

module.exports = {
  Worlds,
  Fortunes,
};