66var fs = require ( 'fs' ) ;
77var path = require ( 'path' ) ;
88var _ = require ( 'lodash' ) ;
9+ var mysql = require ( 'mysql' ) ;
10+ var Promise = require ( "bluebird" ) ;
911var yargs = require ( 'yargs' )
1012 . usage ( 'Usage: $0 <command> [options]' )
1113 . command ( 'init' , '初始化数据库' )
12- . example ( '$0 init --dbname codepush --dbhost localhost --dbuser root --dbpassword 123456' , '初始化code-push-server数据库' )
14+ . example ( '$0 init --dbname codepush --dbhost localhost --dbuser root --dbpassword 123456 --dbport 3306 --force ' , '初始化code-push-server数据库' )
1315 . default ( { dbname : 'codepush' , dbhost : 'localhost' , dbuser : 'root' , dbpassword : null } )
1416 . help ( 'h' )
1517 . alias ( 'h' , 'help' ) ;
16- var argv = yargs
17- . argv ;
18+ var argv = yargs . argv ;
1819var command = argv . _ [ 0 ] ;
19- var mysql = require ( 'mysql' ) ;
20- var Promise = require ( "bluebird" ) ;
2120var dbname = argv . dbname ? argv . dbname : 'codepush' ;
2221var dbhost = argv . dbhost ? argv . dbhost : 'localhost' ;
2322var dbuser = argv . dbuser ? argv . dbuser : 'root' ;
23+ var dbport = argv . dbport ? argv . dbport : 3306 ;
2424var dbpassword = _ . isString ( argv . dbpassword ) ? argv . dbpassword : null ;
2525if ( command === 'init' ) {
2626 var connection2 ;
2727 var connection = mysql . createConnection ( {
2828 host : dbhost ,
2929 user : dbuser ,
30- password : dbpassword
30+ password : dbpassword ,
31+ port : dbport
3132 } ) ;
33+ var createDatabaseSql = argv . force ? `CREATE DATABASE if not exists ${ dbname } ` :
34+ `CREATE DATABASE ${ dbname } ` ;
3235 Promise . promisifyAll ( connection ) ;
3336 connection . connect ( ) ;
34- connection . queryAsync ( `CREATE DATABASE ${ dbname } ` )
37+ connection . queryAsync ( createDatabaseSql )
3538 . then ( function ( ) {
3639 connection2 = mysql . createConnection ( {
3740 host : dbhost ,
3841 user : dbuser ,
3942 password : dbpassword ,
4043 database : dbname ,
41- multipleStatements : true
44+ multipleStatements : true ,
45+ port : dbport
4246 } ) ;
4347 connection2 . connect ( ) ;
4448 Promise . promisifyAll ( connection2 ) ;
@@ -68,4 +72,4 @@ if (command === 'init') {
6872 } ) ;
6973} else {
7074 yargs . showHelp ( ) ;
71- }
75+ }
0 commit comments