mongo2rethink/dbs.go

// dbs
package main

import (
	"errors"
	"fmt"
	"log"
	"os"
	"strings"

	// rt "gopkg.in/dancannon/gorethink.v2"
	//	"gopkg.in/mgo.v2"
)

var (
	dbs = Dbs{
		Mongo:   &Mongo{},
		Rethink: &Rethink{},
	}
)

// Dbs structure
type Dbs struct {
	MongoURI   string
	MongoDB    string
	RethinkURI string
	RethinkDB  string
	Mongo      *Mongo
	Rethink    *Rethink
}

func (db *Dbs) poolRethink() {
	var err error

	if opts.Debug {
		fmt.Printf("DBS: %+v\n", db)
	}

	if db.RethinkURI != "" {
		uri := strings.Split(dbs.RethinkURI, ",")
		if opts.Debug {
			fmt.Printf("RT_URI: %s\n", uri)
		}

		database := strings.Split(dbs.RethinkDB, ":")
		if len(database) > 1 {
			db.Rethink, err = NewRethinkDB(uri, database[0], database[1])
		} else {
			err = errors.New("DB format error - DB:Table")
		}

		if err != nil {
			fmt.Println("RethinkDB connect Error: ", err.Error())
			os.Exit(-4)
		}
	}

	if opts.Debug {
		fmt.Printf("DBS: %+v\n", db)
	}

}

func (db *Dbs) connectMongo() {
	var err error
	db.Mongo, err = NewMongoDB(db.MongoURI, db.MongoDB)
	if err != nil {
		log.Println("Mongodb connect Error: ", err.Error())
		os.Exit(-3)
	}
}