aggiunto come opzione il nome del Database di destinazione

This commit is contained in:
Miki 2016-05-16 16:03:23 +02:00
parent 4a160cf55b
commit e968c82c21
3 changed files with 6 additions and 2 deletions

5
dbs.go
View file

@ -16,11 +16,13 @@ var (
dbs = Dbs{
MongoUri: "mongodb://127.0.0.1:27018",
RedisUri: "127.0.0.1:6379",
Database: "lastlogin",
}
)
type Dbs struct {
MongoUri string
Database string
RedisUri string
rdb *redis.Pool //*redis.Client
mdb *mgo.Session
@ -80,5 +82,6 @@ func (db *Dbs) connectMongo() {
log.Println("Mongodb connect Error: ", err.Error())
os.Exit(-3)
}
db.ll = db.mdb.DB("lastlogin").C(fmt.Sprintf("lastlogin_%s", opts.Month))
db.ll = db.mdb.DB(db.Database).C(fmt.Sprintf("lastlogin_%s", opts.Month))
}

View file

@ -11,7 +11,7 @@ import (
)
const (
_VERSION = "v2.7.1"
_VERSION = "v2.7.2"
)
var (

View file

@ -50,6 +50,7 @@ func init() {
flag.StringVar(&opts.Influxdb, "i", "", "influxdb server")
flag.StringVar(&opts.Hostname, "H", "", "hostname")
flag.StringVar(&dbs.MongoUri, "m", dbs.MongoUri, "Mongodb")
flag.StringVar(&dbs.Database, "d", dbs.Database, "Mongodb Database")
flag.StringVar(&dbs.RedisUri, "r", dbs.RedisUri, "Redis")
flag.StringVar(&opts.LogFile, "l", opts.LogFile, "Logs filename")
flag.DurationVar(&opts.RedisTTL, "t", opts.RedisTTL, "Redis keys TTL")