cgimport/odbc/odbclient.go

package odbc

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	odb "git.wecise.com/wecise/odb-go/odb"
	ucfg "git.wecise.com/wecise/util/cfg"
	ulog "git.wecise.com/wecise/util/logger"
	"github.com/scylladb/go-set/strset"
)

var DefaultAppName = "cgimport"
var Config = ucfg.MConfig(&ucfg.CfgOption{
	Name: "m:default",
	Type: ucfg.INI_TEXT,
	Values: []string{`
[log]
level=debug
dir=` + filepath.Join("/", "opt", "matrix", "var", "logs") + `
file=` + DefaultAppName + ".log" + `
console=true
color=true
consolelevel=info
format=yyyy-MM-dd HH:mm:ss.SSSSSS [pid] [level] module/file:line msg
eol=\n
size=10MB
count=100
scroll=1d
expire=14d
`}},
	ucfg.GetIniFileCfgOption(filepath.Join(fmt.Sprint(DefaultAppName, ".conf"))),
	ucfg.CFGOPTION_ENVS,
	ucfg.CFGOPTION_ARGS)

var Logger = ulog.New().WithConfig(Config, "log")
var LogFile = Logger.FileOutPath()

var ODBClient odb.Client
var ODBError error
var Debug bool
var Usage string

var default_keyspace = `oktest`
var default_odbpaths = `127.0.0.1:11001`

var default_config = &odb.Config{
	Hosts:    []string{"127.0.0.1:11001,47.92.151.165:11001"},
	Keyspace: "oktest",
	User:     fmt.Sprint("测试客户端"),
	Pass:     "********",
	PoolSize: 20,
	Debug:    false,
}

func config_merge(a *odb.Config, b *odb.Config) *odb.Config {
	if len(b.Hosts) > 0 {
		a.Hosts = b.Hosts
	}
	if b.Port != 0 {
		a.Port = b.Port
	}
	if b.Keyspace != "" {
		a.Keyspace = b.Keyspace
	}
	if b.User != "" {
		a.User = b.User
	}
	if b.Pass != "" {
		a.Pass = b.Pass
	}
	if b.PoolSize != 0 {
		a.PoolSize = b.PoolSize
	}
	if b.Debug {
		a.Debug = b.Debug
	}
	return a
}

// 参数设置为程序默认配置
// 可以通过命令行，环境变量 或 与应用同名的.conf配置文件 设置
// 远程连接需要在白名单中增加本地IP
func ODBC(odbcfgs ...*odb.Config) odb.Client {
	odbcfg := config_merge(&odb.Config{}, default_config)
	for _, c := range odbcfgs {
		odbcfg = config_merge(odbcfg, c)
	}
	default_keyspace = odbcfg.Keyspace
	default_odbpaths = strings.Join(odbcfg.Hosts, ",")
	Usage = `
命令行参数：
odbpath=` + default_odbpaths + ` #指定odbserver路径，默认通过环境变量ODBPATH或通过ETCD相关配置获取
keyspace=` + default_keyspace + ` #指定keyspace，默认通过环境变量KEYSPACE获取
debug=true #开启调试模式，输出更多信息
	
环境变量需求：
ODBPATH=` + default_odbpaths + ` #指定odbserver路径，默认通过配置信息 odbc.odbpath 获取
KEYSPACE=` + default_keyspace + ` #指定keyspace，默认通过配置信息 odbc.keyspace 获取
ETCDPATH=127.0.0.1:2379 #必须设置
ETCDUSER=
ETCDPASS=
CASSANDRAPATH=127.0.0.1
CASSANDRALOOKUP=false
NATSPATH=nats://user:N5JbKeT1C3uOUh317OVXjg==@127.0.0.1:4222

LANG=zh_CN.utf8
LC_ALL=zh_CN.utf8

可通过` + DefaultAppName + `.conf配置运行参数：
[odbc]
;指定odbserver路径
odbpath=` + default_odbpaths + `
;指定keyspace
keyspace=` + default_keyspace + `
`
	Logger.SetFormat("yyyy-MM-dd HH:mm:ss.SSSSSS [pid] [level] msg", "\n")
	odbpaths := strset.New(strings.Split(ucfg.CommandArgs.GetString("odbpath",
		strings.Join(ucfg.Environs.GetStrings("ODBPATH",
			Config.GetStrings("odbc.odbpath", default_odbpaths)...), ",")), ",")...).List()
	keyspace := ucfg.CommandArgs.GetString("keyspace",
		ucfg.Environs.GetString("KEYSPACE", Config.GetString("odbc.keyspace", default_keyspace)))
	Debug = ucfg.CommandArgs.GetBool("debug", false)
	ODBClient, ODBError = odb.NewClient(config_merge(odbcfg, &odb.Config{
		Keyspace: keyspace,
		Hosts:    odbpaths,
		Debug:    Debug,
	}))
	if ODBError != nil {
		fmt.Print(Usage)
		if strings.Contains(ODBError.Error(), "error: EOF") {
			println("\n!!!should add your ip to odbserver whitelist!!!\n")
			os.Exit(1)
		}
		panic(ODBError)
	}
	Logger.Info("odbpath :", ODBClient.Config().Hosts, ODBClient.Config().Port)
	Logger.Info("keyspace:", ODBClient.Config().Keyspace)
	Logger.Info("debug:   ", Debug)
	return ODBClient
}