jmms/generators/test/db.js

Jmms cli tools, Jmms is a java meta-micro-service framework

const _       = require('lodash');
const log     = require('../log');
const path    = require('path');
const chalk   = require('chalk');
const request = require('sync-request');
const opts    = require('./opts');

const clean = () => {
    log.info("\nCleaning db...");
    const url = opts.testUrl + '/db';
    const res = request('DELETE', url);
    if(res.statusCode == 200) {
        const affected = JSON.parse(res.body);
        log.info(chalk.green("Clean db done, " + affected + " rows deleted\n"));
    }else {
        log.error("Unable clean db, status " + res.statusCode + ", body : " + res.body);
        process.exit(1);
    }
}

const drop = () => {
    log.info("\nDrop and recreate db...");
    const url = opts.testUrl + '/db/drop';
    const res = request('DELETE', url);
    if(res.statusCode == 200) {
        log.info(chalk.green("Drop and recreate db done\n"));
    }else {
        log.error("Unable drop db, status " + res.statusCode + ", body : " + res.body);
        process.exit(1);
    } 
}

module.exports = {
    prepare: () => {
        // if(opts.options.drop == true) {
        //     drop();
        // }else 
        if(opts.options.clean == true) {
            clean();
        }
    }
}