提交 2e8c928c 编写于 作者: X xiyangxixian

update filepath

上级 0f29e635
...@@ -25,6 +25,7 @@ import ( ...@@ -25,6 +25,7 @@ import (
"io" "io"
"io/ioutil" "io/ioutil"
"os" "os"
"path/filepath"
"regexp" "regexp"
"runtime" "runtime"
"strings" "strings"
...@@ -376,10 +377,10 @@ func FormatDSN(env *dsn) string { ...@@ -376,10 +377,10 @@ func FormatDSN(env *dsn) string {
// SoarVersion soar version information // SoarVersion soar version information
func SoarVersion() { func SoarVersion() {
fmt.Println("Version:", Version) //fmt.Println("Version:", Version)
fmt.Println("Branch:", Branch) //fmt.Println("Branch:", Branch)
fmt.Println("Compile:", Compile) //fmt.Println("Compile:", Compile)
fmt.Println("GitDirty:", GitDirty) //fmt.Println("GitDirty:", GitDirty)
} }
// 因为vitess sqlparser 使用了 glog 中也会使用 flag,为了不让用户困扰我们单独写一个 usage // 因为vitess sqlparser 使用了 glog 中也会使用 flag,为了不让用户困扰我们单独写一个 usage
...@@ -588,21 +589,14 @@ func readCmdFlags() error { ...@@ -588,21 +589,14 @@ func readCmdFlags() error {
Config.SamplingStatisticTarget = *samplingStatisticTarget Config.SamplingStatisticTarget = *samplingStatisticTarget
Config.ConnTimeOut = *connTimeOut Config.ConnTimeOut = *connTimeOut
Config.QueryTimeOut = *queryTimeOut Config.QueryTimeOut = *queryTimeOut
Config.LogLevel = *logLevel Config.LogLevel = *logLevel
if strings.HasPrefix(*logOutput, "/") {
if filepath.IsAbs(*logOutput) || BaseDir == "" {
Config.LogOutput = *logOutput Config.LogOutput = *logOutput
} else { } else {
if BaseDir == "" { Config.LogOutput = filepath.Join(BaseDir, *logOutput)
Config.LogOutput = *logOutput
} else {
if runtime.GOOS == "windows" {
Config.LogOutput = *logOutput
} else {
Config.LogOutput = BaseDir + "/" + *logOutput
}
}
} }
Config.ReportType = strings.ToLower(*reportType) Config.ReportType = strings.ToLower(*reportType)
Config.ReportCSS = *reportCSS Config.ReportCSS = *reportCSS
Config.ReportJavascript = *reportJavascript Config.ReportJavascript = *reportJavascript
...@@ -612,12 +606,13 @@ func readCmdFlags() error { ...@@ -612,12 +606,13 @@ func readCmdFlags() error {
Config.IgnoreRules = strings.Split(*ignoreRules, ",") Config.IgnoreRules = strings.Split(*ignoreRules, ",")
Config.RewriteRules = strings.Split(*rewriteRules, ",") Config.RewriteRules = strings.Split(*rewriteRules, ",")
*blackList = strings.TrimSpace(*blackList) *blackList = strings.TrimSpace(*blackList)
if strings.HasPrefix(*blackList, "/") || *blackList == "" {
if filepath.IsAbs(*blackList) || *blackList == "" {
Config.BlackList = *blackList Config.BlackList = *blackList
} else { } else {
pwd, _ := os.Getwd() Config.BlackList = filepath.Join(BaseDir, *blackList)
Config.BlackList = pwd + "/" + *blackList
} }
Config.MaxJoinTableCount = *maxJoinTableCount Config.MaxJoinTableCount = *maxJoinTableCount
Config.MaxGroupByColsCount = *maxGroupByColsCount Config.MaxGroupByColsCount = *maxGroupByColsCount
Config.MaxDistinctCount = *maxDistinctCount Config.MaxDistinctCount = *maxDistinctCount
...@@ -691,8 +686,8 @@ func ParseConfig(configFile string) error { ...@@ -691,8 +686,8 @@ func ParseConfig(configFile string) error {
if configFile == "" { if configFile == "" {
configs = []string{ configs = []string{
"/etc/soar.yaml", "/etc/soar.yaml",
BaseDir + "/etc/soar.yaml", filepath.Join(BaseDir, "etc", "soar.yaml"),
BaseDir + "/soar.yaml", filepath.Join(BaseDir, "soar.yaml"),
} }
} else { } else {
configs = []string{ configs = []string{
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册