grlogs/funtion.go

package grlogs

import (
	"fmt"
	"github.com/astaxie/beego/logs"
	"os"
	"path"
	"strings"
)

var adatperSetMapper = map[Adapter]func(l *Logger, level Level) error{

	AdapterSocket: func(l *Logger, level Level) error {
		c := ConnLogConfig{
			ReconnectOnMsg: false,
			Reconnect:      true,
			Net:            "",
			Addr:           "127.0.0.1:9888",
			Level:          level,
		}
		return l.SetLogger(logs.AdapterConn, c.String())
	},

	AdapterFile: func(l *Logger, level Level) error {
		if wd, err := os.Getwd(); err == nil {
			c := FileLogConfig{
				Filename: path.Join(wd, fmt.Sprintf("runtime/logs/%s.log", l.Lable)),
				Level:    level,
				Maxlines: 0,
				Daily:    true,
				Maxdays:  7,
				Color:    true,
				Rotate:   true,
			}

			return l.SetLogger(logs.AdapterFile, c.String())
		} else {
			return err
		}
	},

	AdapterConsole: func(l *Logger, level Level) error {
		c := ConsoleLogConfig{Level: level}
		return l.SetLogger(logs.AdapterConsole, c.String())
	},

	AdapterElasticSearch: func(l *Logger, level Level) error {
		c := GenEsConfig(level)
		return l.SetLogger(logs.AdapterEs, c.String())
	},

	AdapterAliLs: func(l *Logger, level Level) error {
		c := GenAliConfig(level)
		return l.SetLogger(logs.AdapterAliLS, c.String())
	},
}

func GenAliConfig(level Level) *AliLSConfig {
	var project string = "gaore-app-logstore"
	var endpoint string

	if os.Getenv(envkey) == "prod" || os.Getenv(envkey) == "" || os.Getenv(envkey) == "gray" {
		endpoint = project + ".cn-shenzhen-intranet.log.aliyuncs.com"
	} else if os.Getenv(envkey) == "dev" {
		endpoint = project + ".cn-shenzhen.log.aliyuncs.com"
	}

	c := &AliLSConfig{
		Project:   project,
		Endpoint:  endpoint,
		KeyID:     "LTAI4GCHwcqtrFD4DHRHxR4k",
		KeySecret: "Ln19xfVYy6OMlJeF9aBvFl4fhRUKBl",
		LogStore:  "gaore-app-logstore",
		Topics:    []string{os.Getenv("GRLOG_APP_NAME")},
		Source:    "",
		Level:     level,
		FlushWhen: 0,
	}
	return c
}

func GenEsConfig(level Level) *EsConfig {
	dsn := "http://es-cn-0pp1mm3hq000dnbh4.public.elasticsearch.aliyuncs.com:9200/"
	if os.Getenv(envkey) == "prod" || os.Getenv(envkey) == "" || os.Getenv(envkey) == "gray" {
		dsn = "http://es-cn-0pp1mm3hq000dnbh4.elasticsearch.aliyuncs.com:9200/"
	}

	c := &EsConfig{
		Username: "elastic",
		Password: "Hellogaore@",
		Dsn:      dsn,
		Level:    level,
		Index:    os.Getenv("GRLOG_APP_NAME"),
	}
	return c
}

var adatperDropMapper = map[Adapter]func(l *Logger) error{

	AdapterAliLs: func(l *Logger) error {
		return l.BeeLogger.DelLogger(logs.AdapterAliLS)
	},

	AdapterFile: func(l *Logger) error {
		return l.BeeLogger.DelLogger(logs.AdapterFile)
	},

	AdapterConsole: func(l *Logger) error {
		return l.BeeLogger.DelLogger(logs.AdapterConsole)
	},

	AdapterElasticSearch: func(l *Logger) error {
		return l.BeeLogger.DelLogger(logs.AdapterEs)
	},
}

func formatLog(f interface{}, v ...interface{}) string {
	var msg string
	switch f.(type) {
	case string:
		msg = f.(string)
		if len(v) == 0 {
			return msg
		}
		if strings.Contains(msg, "%") && !strings.Contains(msg, "%%") {
			//format string
		} else {
			//do not contain format char
			msg += strings.Repeat(" %v", len(v))
		}
	default:
		msg = fmt.Sprint(f)
		if len(v) == 0 {
			return msg
		}
		msg += strings.Repeat(" %v", len(v))
	}
	return fmt.Sprintf(msg, v...)
}