Compare commits

...

2 Commits

Author SHA1 Message Date
62db070f61 添加对应 php注册登录的token 验证逻辑 2024-04-15 17:01:42 +08:00
de7cd23deb 日志初步结构 2024-04-15 16:08:17 +08:00
17 changed files with 365 additions and 87 deletions

View File

@ -1,6 +1,8 @@
package model package model
import ( import (
"strings"
"gorm.io/gorm" "gorm.io/gorm"
) )
@ -9,3 +11,140 @@ type KillaraCatalogLanguageModel struct {
db *gorm.DB db *gorm.DB
TableName string // 表名 TableName string // 表名
} }
func (m *KillaraCatalogLanguageModel) InsertLanguage(language *KillaraCatalogLanguage) (uint64, error) {
if err := m.db.Model(&KillaraCatalogLanguage{}).Create(language).Error; err != nil {
return 0, err
}
return *language.LanguageId, nil
}
func (m *KillaraCatalogLanguageModel) UpdateLanguage(language *KillaraCatalogLanguage) error {
return m.db.Model(&KillaraCatalogLanguage{}).Where("language_id = ?", *language.LanguageId).Updates(language).Error
}
func (m *KillaraCatalogLanguageModel) DeleteLanguage(language *KillaraCatalogLanguage) error {
return m.db.Where("language_id = ?", *language.LanguageId).Delete(&KillaraCatalogLanguage{}).Error
}
func (m *KillaraCatalogLanguageModel) GetLanguage(languageID uint64) (*KillaraCatalogLanguage, error) {
var language KillaraCatalogLanguage
err := m.db.Where("language_id = ?", languageID).First(&language).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &language, err
}
func (m *KillaraCatalogLanguageModel) GetLanguagesForBackEnd(data map[string]interface{}) ([]*KillaraCatalogLanguage, error) {
var languages []*KillaraCatalogLanguage
query := m.db.Model(&KillaraCatalogLanguage{})
var conditions []interface{}
if filterName, ok := data["filter_name"].(string); ok && filterName != "" {
conditions = append(conditions, m.db.Where("name LIKE ?", filterName+"%"))
}
if filterCode, ok := data["filter_code"].(string); ok && filterCode != "" {
conditions = append(conditions, m.db.Where("code LIKE ?", filterCode+"%"))
}
if filterStatus, ok := data["filter_status"]; ok {
conditions = append(conditions, m.db.Where("status = ?", filterStatus))
}
if len(conditions) > 0 {
query = query.Where(strings.Join(make([]string, len(conditions)), " AND "), conditions...)
}
if sort, ok := data["sort"].(string); ok && sort != "" {
order := "DESC"
if sortOrder, ok := data["order"].(string); ok {
order = sortOrder
}
query = query.Order(sort + " " + order)
}
if start, ok := data["start"].(int); ok && start > 0 {
query = query.Offset(start)
}
if limit, ok := data["limit"].(int); ok && limit > 0 {
query = query.Limit(limit)
} else {
query = query.Limit(10)
}
err := query.Find(&languages).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return languages, err
}
func (m *KillaraCatalogLanguageModel) GetTotalLanguagesForBackEnd(data map[string]interface{}) (int64, error) {
var count int64
query := m.db.Model(&KillaraCatalogLanguage{})
var conditions []interface{}
if filterName, ok := data["filter_name"].(string); ok && filterName != "" {
conditions = append(conditions, m.db.Where("name LIKE ?", filterName+"%"))
}
if filterCode, ok := data["filter_code"].(string); ok && filterCode != "" {
conditions = append(conditions, m.db.Where("code LIKE ?", filterCode+"%"))
}
if filterStatus, ok := data["filter_status"]; ok {
conditions = append(conditions, m.db.Where("status = ?", filterStatus))
}
if len(conditions) > 0 {
query = query.Where(strings.Join(make([]string, len(conditions)), " AND "), conditions...)
}
err := query.Count(&count).Error
if err == gorm.ErrRecordNotFound {
return 0, nil
}
return count, err
}
func (m *KillaraCatalogLanguageModel) GetLanguageForBackEnd(languageID uint64) (*KillaraCatalogLanguage, error) {
var language KillaraCatalogLanguage
err := m.db.Where("language_id = ?", languageID).First(&language).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &language, err
}
func (m *KillaraCatalogLanguageModel) GetLanguageByCodeForBackEnd(code string) (*KillaraCatalogLanguage, error) {
var language KillaraCatalogLanguage
err := m.db.Where("code = ?", code).First(&language).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &language, err
}
// 前台使用的方法
func (m *KillaraCatalogLanguageModel) GetLanguagesForFrontEnd() ([]*KillaraCatalogLanguage, error) {
var languages []*KillaraCatalogLanguage
err := m.db.Where("status = 1").Order("sort_order").Find(&languages).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return languages, err
}
func (m *KillaraCatalogLanguageModel) GetLanguageForFrontEnd(languageID uint64) (*KillaraCatalogLanguage, error) {
var language KillaraCatalogLanguage
err := m.db.Where("status = 1 AND language_id = ?", languageID).First(&language).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &language, err
}

View File

@ -32,6 +32,9 @@ func (m *KillaraCustomerModel) DeleteCustomer(customer *KillaraCustomer) error {
func (m *KillaraCustomerModel) GetCustomer(customerID uint64) (*KillaraCustomer, error) { func (m *KillaraCustomerModel) GetCustomer(customerID uint64) (*KillaraCustomer, error) {
var customer KillaraCustomer var customer KillaraCustomer
err := m.db.Where("customer_id = ? AND status = 1", customerID).First(&customer).Error err := m.db.Where("customer_id = ? AND status = 1", customerID).First(&customer).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &customer, err return &customer, err
} }

View File

@ -34,6 +34,7 @@ func (m *KillaraCustomerTokenModel) InsertToken(data *KillaraCustomerToken) erro
return m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error { return m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error {
// 查找是否存在相同客户ID、平台和Token的记录 // 查找是否存在相同客户ID、平台和Token的记录
var existingToken KillaraCustomerToken var existingToken KillaraCustomerToken
err := tx.Where("customer_id = ? AND platform = ? AND token = ?", data.CustomerId, data.Platform, *data.Token).First(&existingToken).Error err := tx.Where("customer_id = ? AND platform = ? AND token = ?", data.CustomerId, data.Platform, *data.Token).First(&existingToken).Error
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) { if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
@ -57,7 +58,7 @@ func (m *KillaraCustomerTokenModel) InsertToken(data *KillaraCustomerToken) erro
return err return err
} }
return tx.Commit().Error // 提交事务 return err // 提交事务
}) })
} }
@ -113,12 +114,12 @@ func (m *KillaraCustomerTokenModel) GetToken(token string) (*KillaraCustomerToke
// return nil, nil // return nil, nil
// } // }
func (m *KillaraCustomerTokenModel) CheckToken(token string) (*KillaraCustomerToken, error) { func (m *KillaraCustomerTokenModel) CheckToken(tokenstr string) (*KillaraCustomerToken, error) {
var resultToken *KillaraCustomerToken var resultToken *KillaraCustomerToken
err := m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error { err := m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error {
// 查找 Token 记录 // 查找 Token 记录
var token KillaraCustomerToken var token KillaraCustomerToken
err := tx.Where("token = ?", token).First(&token).Error err := tx.Where("token = ?", tokenstr).First(&token).Error
if err != nil { if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) { if errors.Is(err, gorm.ErrRecordNotFound) {
return nil return nil
@ -138,7 +139,7 @@ func (m *KillaraCustomerTokenModel) CheckToken(token string) (*KillaraCustomerTo
} }
resultToken = &token resultToken = &token
return tx.Commit().Error // 提交事务 return err // 提交事务
}) })
if err != nil { if err != nil {
@ -150,7 +151,7 @@ func (m *KillaraCustomerTokenModel) CheckToken(token string) (*KillaraCustomerTo
func (m *KillaraCustomerTokenModel) ClearDuplicateToken(customerID uint64, currentToken string, platform int) error { func (m *KillaraCustomerTokenModel) ClearDuplicateToken(customerID uint64, currentToken string, platform int) error {
return m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error { return m.db.Model(&KillaraCustomerToken{}).Transaction(func(tx *gorm.DB) error {
var tokens []KillaraCustomerToken var tokens []*KillaraCustomerToken
err := tx.Where("customer_id = ? AND platform = ?", customerID, platform).Find(&tokens).Error err := tx.Where("customer_id = ? AND platform = ?", customerID, platform).Find(&tokens).Error
if err != nil { if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) { if errors.Is(err, gorm.ErrRecordNotFound) {
@ -164,7 +165,7 @@ func (m *KillaraCustomerTokenModel) ClearDuplicateToken(customerID uint64, curre
for _, token := range tokens { for _, token := range tokens {
if *token.Token != currentToken { if *token.Token != currentToken {
// err := tx.Delete(token).Error todo: 不太明白php为什么不删除token, 难道用来链路跟踪? 但是客户端id也被更新了, 没有存在的意义了 // err := tx.Delete(token).Error todo: 不太明白php为什么不删除token, 难道用来链路跟踪? 但是客户端id也被更新了, 没有存在的意义了
err := tx.Where("token = ?", token).Update("customer_id", 0).Error err := tx.Where("token = ?", *token.Token).Update("customer_id", 0).Error
if err != nil { if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) { if errors.Is(err, gorm.ErrRecordNotFound) {
return fmt.Errorf("tokens 查询出来, 自身都不存在, 疑似出了什么错误2") return fmt.Errorf("tokens 查询出来, 自身都不存在, 疑似出了什么错误2")

View File

@ -1,6 +1,7 @@
package actions package actions
import ( import (
"encoding/json"
"strings" "strings"
"time" "time"
@ -17,16 +18,99 @@ var CompanyKey = "vestmore-bjwl"
// @Action base/getToken // @Action base/getToken
// Base_GetToken // Base_GetToken
// action: string; // randstr: string;
// app_market: string; // sign: string;
// lang: string; // action: string;
// token: string; // lang: string;
// app_market: int64;
// timestamp: int64;
// token: string;
func BaseGetToken(ctx *ActionContext[BaseGetTokenParam]) (resp *basic.Response) { func BaseGetToken(ctx *ActionContext[BaseGetTokenParam]) (resp *basic.Response) {
// model.Models.KillaraCustomerModel.Find() const (
log.Println() TokenExpiry = 864000 // Token过期时间单位
)
return resp.Success() type TokenData struct {
LanguageID uint64 `json:"language_id"`
LanguageCode string `json:"language_code"`
}
modelToken := model.Models.KillaraCustomerTokenModel
modelLanguage := model.Models.KillaraCatalogLanguageModel
lang := ctx.Lang
language, err := modelLanguage.GetLanguageByCodeForBackEnd(lang)
if err != nil || language == nil {
language, _ = modelLanguage.GetLanguageByCodeForBackEnd("en")
}
tokenData := TokenData{
LanguageID: *language.LanguageId,
LanguageCode: *language.Code,
}
isLogin := false
var token string
if ctx.Param.Token != "" {
result, err := modelToken.CheckToken(ctx.Param.Token)
if err == nil && result != nil {
token = ctx.Param.Token
// Token存在并且有效
modelToken.UpdateTokenExpiry(token, time.Now().Unix()+TokenExpiry)
tokenDataJSON, _ := json.Marshal(tokenData)
modelToken.UpdateTokenData(token, map[string]interface{}{
"data": string(tokenDataJSON),
})
if result.CustomerId != nil && *result.CustomerId != 0 {
modelCustomer := model.Models.KillaraCustomerModel
customer, err := modelCustomer.GetCustomer(*result.CustomerId)
if err == nil && customer != nil {
isLogin = true
} else {
modelToken.UpdateTokenCustomerID(token, 0)
}
}
}
}
if token == "" {
// Token不存在或失效重新创建一个
token = auth.GenerateMD5()
tokenDataJSON, err := json.Marshal(tokenData)
if err != nil {
return resp.Error(basic.ErrJSONUnMarshal)
}
tokenItem := &model.KillaraCustomerToken{
Token: &token,
CustomerId: basic.Uint64Ptr(0),
Data: basic.StringPtr(string(tokenDataJSON)),
Expiry: basic.Int64Ptr(time.Now().Unix() + TokenExpiry),
Platform: basic.Int64Ptr(1),
}
err = modelToken.InsertToken(tokenItem)
log.Println(*tokenItem.Token)
if err != nil {
return resp.ErrorErr(1, err)
}
}
return resp.Success(map[string]interface{}{
"token": token,
"is_login": isLogin,
"lang_data": map[string]interface{}{
"language_id": tokenData.LanguageID,
"language_code": tokenData.LanguageCode,
}})
} }
// @Action account/loginWithTelephonePassword // @Action account/loginWithTelephonePassword
@ -258,7 +342,6 @@ func AccountLoginWithEmailPassword(ctx *ActionContext[AccountLoginWithEmailPassw
return resp.ErrorErr(1, err) return resp.ErrorErr(1, err)
} }
ctx.Localize(translator.AccountNotRegistered)
if customer == nil { if customer == nil {
return resp.ErrorTrCode(ctx, translator.AccountNotRegistered) return resp.ErrorTrCode(ctx, translator.AccountNotRegistered)
} }

View File

@ -233,9 +233,12 @@ func AccountRegisterSmsCodeHandler(ctx *gin.Context) {
} }
type BaseGetTokenParam struct { type BaseGetTokenParam struct {
Randstr string `json:"randstr" form:"randstr" binding:"-"`
Sign string `json:"sign" form:"sign" binding:"-"`
Action string `json:"action" form:"action" binding:"-"` Action string `json:"action" form:"action" binding:"-"`
AppMarket string `json:"app_market" form:"app_market" binding:"-"`
Lang string `json:"lang" form:"lang" binding:"-"` Lang string `json:"lang" form:"lang" binding:"-"`
AppMarket int64 `json:"app_market" form:"app_market" binding:"-"`
Timestamp int64 `json:"timestamp" form:"timestamp" binding:"-"`
Token string `json:"token" form:"token" binding:"-"` Token string `json:"token" form:"token" binding:"-"`
} }

View File

@ -1,11 +1,11 @@
package main package main
import ( import (
"log"
"testing" "testing"
_ "github.com/go-sql-driver/mysql" _ "github.com/go-sql-driver/mysql"
"github.com/iapologizewhenimwrong/Vestmore_GO/translator" "github.com/iapologizewhenimwrong/Vestmore_GO/translator"
"github.com/iapologizewhenimwrong/Vestmore_GO/utils/log"
"github.com/nicksnyder/go-i18n/v2/i18n" "github.com/nicksnyder/go-i18n/v2/i18n"
"github.com/pelletier/go-toml/v2" "github.com/pelletier/go-toml/v2"
"golang.org/x/text/language" "golang.org/x/text/language"

View File

@ -21,7 +21,27 @@ func init() {
globpath := filepath.Join(currentDir, "/*.toml") globpath := filepath.Join(currentDir, "/*.toml")
Bundle = i18n.NewBundle(language.Chinese) Bundle = i18n.NewBundle(language.Chinese)
Bundle.RegisterUnmarshalFunc("toml", toml.Unmarshal) Bundle.RegisterUnmarshalFunc("toml", func(data []byte, v interface{}) error {
var m map[string]any
err := toml.Unmarshal(data, &m)
if err != nil {
return err
}
for _, v := range m {
vm := v.(map[string]any)
desc := vm["description"].(string)
vm["one"] = desc
vm["other"] = desc
}
fixdata, err := toml.Marshal(m)
if err != nil {
return err
}
return toml.Unmarshal(fixdata, v)
})
matches, err := filepath.Glob(globpath) matches, err := filepath.Glob(globpath)
if err != nil { if err != nil {
panic(err) panic(err)
@ -30,14 +50,6 @@ func init() {
Bundle.MustLoadMessageFile(m) Bundle.MustLoadMessageFile(m)
} }
// localizer := i18n.NewLocalizer(Bundle, "zh_cn")
// log.Println(localizer.Localize(&i18n.LocalizeConfig{
// MessageID: "format_account_is_insufficient",
// TemplateData: map[string]any{
// "CatalogCurrency": "USDT",
// },
// }))
} }
func Localize(MessageID TrCode, Lang string, MessageTemplateParam any) (string, error) { func Localize(MessageID TrCode, Lang string, MessageTemplateParam any) (string, error) {

View File

@ -13,6 +13,7 @@ import (
"text/template" "text/template"
"unicode" "unicode"
"github.com/nicksnyder/go-i18n/v2/i18n"
"github.com/pelletier/go-toml/v2" "github.com/pelletier/go-toml/v2"
) )
@ -118,7 +119,10 @@ func createTrCode(filePath string) {
} }
func TestTrOnline(t *testing.T) { func TestTrOnline(t *testing.T) {
var localizer = i18n.NewLocalizer(Bundle, "zh_cn")
log.Println(localizer.Localize(&i18n.LocalizeConfig{
MessageID: string("account_not_registered"),
}))
} }
func toCamelCase(s string) string { func toCamelCase(s string) string {

View File

@ -280,5 +280,6 @@ description = "默认交易时间段"
description = "非默认交易时间段" description = "非默认交易时间段"
[all_day_transaction] [all_day_transaction]
description = "全天交易" description = "全天交易"

View File

View File

@ -1,21 +1,40 @@
package auth package auth
import ( import (
"crypto/md5"
"fmt" "fmt"
"math/rand" "math/rand"
"strconv"
"time"
) )
var myRand = rand.New(rand.NewSource(time.Now().UnixNano()))
func GenerateVerificationCode() string { func GenerateVerificationCode() string {
var code string var code string
for i := 0; i < 3; i++ { for i := 0; i < 3; i++ {
// 生成两个 10-99 之间的随机数 // 生成两个 10-99 之间的随机数
a := rand.Intn(90-10+1) + 10 a := myRand.Intn(90-10+1) + 10
b := rand.Intn(90-10+1) + 10 b := myRand.Intn(90-10+1) + 10
c := rand.Intn(90-10+1) + 10 c := myRand.Intn(90-10+1) + 10
// 将随机数拼接到验证码字符串 // 将随机数拼接到验证码字符串
code += fmt.Sprintf("%d%d%d", a, b, c) code += fmt.Sprintf("%d%d%d", a, b, c)
} }
return code return code
} }
func GenerateMD5() string {
// 将时间戳转换为字节slice
timestampBytes := []byte(strconv.FormatInt(time.Now().UnixNano(), 10))
// 计算MD5哈希
hashBytes := md5.Sum(timestampBytes)
// 将哈希值转换为十六进制字符串
hashHex := fmt.Sprintf("%x", hashBytes)
return hashHex
}

View File

@ -8,7 +8,8 @@ type ErrorCode struct {
var ( var (
ErrRespNotNil = &ErrorCode{Code: 10000, Message: "resp must not nil"} ErrRespNotNil = &ErrorCode{Code: 10000, Message: "resp must not nil"}
ErrEncGcm = &ErrorCode{Code: 10001, Message: "gmc加密错误"} ErrEncGcm = &ErrorCode{Code: 10001, Message: "gmc加密错误"}
ErrJSONUnMarshal = &ErrorCode{Code: 10002, Message: "json 解析错误"}
ErrParamParse = &ErrorCode{Code: 10100, Message: "参数解析错误"} ErrParamParse = &ErrorCode{Code: 10100, Message: "参数解析错误"}

View File

@ -1,6 +1,9 @@
package basic package basic
import "reflect" import (
"reflect"
"strings"
)
func GetLangString(param any) string { func GetLangString(param any) string {
// 获取参数的反射值 // 获取参数的反射值
@ -18,10 +21,11 @@ func GetLangString(param any) string {
// 如果字段存在并且是字符串类型 // 如果字段存在并且是字符串类型
if langField.IsValid() && langField.Kind() == reflect.String { if langField.IsValid() && langField.Kind() == reflect.String {
return langField.String() return strings.TrimSpace(langField.String())
} }
} }
// 如果无法获取有效的字符串值则返回zh_cn // 如果无法获取有效的字符串值则返回zh_cn
return "zh_cn" return "zh_cn"
} }

View File

@ -5,6 +5,8 @@ import (
"github.com/iapologizewhenimwrong/Vestmore_GO/utils/log" "github.com/iapologizewhenimwrong/Vestmore_GO/utils/log"
) )
var respLog = log.New(0)
// 全局返回的结构体 // 全局返回的结构体
type Response struct { type Response struct {
Data interface{} `json:"data"` Data interface{} `json:"data"`
@ -25,7 +27,7 @@ func (resp *Response) Error(errcode *ErrorCode, Data ...interface{}) *Response {
resp.ErrorText = errcode.Message resp.ErrorText = errcode.Message
resp.IsSuccess = false resp.IsSuccess = false
resp.setData(Data) resp.setData(Data)
log.Error(resp.ErrorText) respLog.Error(resp.ErrorText)
return resp return resp
} }
@ -37,7 +39,8 @@ func (resp *Response) ErrorErr(Code int, err error, Data ...interface{}) *Respon
resp.ErrorText = err.Error() resp.ErrorText = err.Error()
resp.IsSuccess = false resp.IsSuccess = false
resp.setData(Data) resp.setData(Data)
log.Error(resp.ErrorText)
respLog.Error(resp.ErrorText)
return resp return resp
} }
@ -49,7 +52,7 @@ func (resp *Response) ErrorMsg(Code int, Message string, Data ...interface{}) *R
resp.ErrorText = Message resp.ErrorText = Message
resp.IsSuccess = false resp.IsSuccess = false
resp.setData(Data) resp.setData(Data)
log.Error(resp.ErrorText) respLog.Error(resp.ErrorText)
return resp return resp
} }

View File

@ -19,9 +19,9 @@ type levelSkip struct {
// JSONFormatter formats logs into parsable json // JSONFormatter formats logs into parsable json
type JSONFormatter struct { type JSONFormatter struct {
skip []*levelSkip diffSkip int
skip []*levelSkip
once sync.Once once sync.Once
} }
// Format renders a single log entry // Format renders a single log entry
@ -49,12 +49,11 @@ func (h *JSONFormatter) Format(e *logrus.Entry) ([]byte, error) {
}) })
var fileinfo string var fileinfo string
if _, file, line, ok := runtime.Caller(skipOnce.Skip); ok { if _, file, line, ok := runtime.Caller(skipOnce.Skip + h.diffSkip); ok {
if e.Level == logrus.InfoLevel { if e.Level == logrus.InfoLevel {
fileinfo = fmt.Sprintf("%s:%d", file, line) fileinfo = fmt.Sprintf("%s:%d", file, line)
} else { } else {
ps := strings.Split(file, "/") ps := strings.Split(file, "/")
ps = ps[len(ps)-4:]
fileinfo = fmt.Sprintf("%s:%d", strings.Join(ps, "/"), line) fileinfo = fmt.Sprintf("%s:%d", strings.Join(ps, "/"), line)
} }

View File

@ -12,28 +12,34 @@ import (
"github.com/sirupsen/logrus" "github.com/sirupsen/logrus"
) )
var l *logrus.Logger var dlog *logrus.Logger
func init() { func init() {
l = logrus.New() dlog = New(0)
}
func New(skip int) *logrus.Logger {
myl := logrus.New()
// 配置 Logstash 作为输出 // 配置 Logstash 作为输出
myl.AddHook(NewUTCTimeHook())
l.AddHook(NewUTCTimeHook())
jf := &JSONFormatter{ jf := &JSONFormatter{
skip: make([]*levelSkip, len(logrus.AllLevels)), skip: make([]*levelSkip, len(logrus.AllLevels)),
diffSkip: skip,
} }
for i := range jf.skip { for i := range jf.skip {
jf.skip[i] = &levelSkip{} jf.skip[i] = &levelSkip{}
} }
l.Formatter = jf myl.Formatter = jf
myl.SetReportCaller(true)
// l.AddHook(&SkipHook{}) if dlog != nil {
myl.SetOutput(dlog.Out)
l.SetReportCaller(true) }
return myl
} }
type SkipHook struct { type SkipHook struct {
@ -99,154 +105,154 @@ func (hook *UTCTimeHook) Fire(entry *logrus.Entry) error {
// this new returned entry. // this new returned entry.
// If you want multiple fields, use `WithFields`. // If you want multiple fields, use `WithFields`.
func WithField(key string, value interface{}) *logrus.Entry { func WithField(key string, value interface{}) *logrus.Entry {
return l.WithField(key, value) return dlog.WithField(key, value)
} }
// Adds a struct of fields to the log entry. All it does is call `WithField` for // Adds a struct of fields to the log entry. All it does is call `WithField` for
// each `Field`. // each `Field`.
func WithFields(fields logrus.Fields) *logrus.Entry { func WithFields(fields logrus.Fields) *logrus.Entry {
return l.WithFields(fields) return dlog.WithFields(fields)
} }
// Add an error as single field to the log entry. All it does is call // Add an error as single field to the log entry. All it does is call
// `WithError` for the given `error`. // `WithError` for the given `error`.
func WithError(err error) *logrus.Entry { func WithError(err error) *logrus.Entry {
return l.WithError(err) return dlog.WithError(err)
} }
// Add a context to the log entry. // Add a context to the log entry.
func WithContext(ctx context.Context) *logrus.Entry { func WithContext(ctx context.Context) *logrus.Entry {
return l.WithContext(ctx) return dlog.WithContext(ctx)
} }
// Overrides the time of the log entry. // Overrides the time of the log entry.
func WithTime(t time.Time) *logrus.Entry { func WithTime(t time.Time) *logrus.Entry {
return l.WithTime(t) return dlog.WithTime(t)
} }
func Logf(level logrus.Level, format string, args ...interface{}) { func Logf(level logrus.Level, format string, args ...interface{}) {
l.Logf(level, format, args...) dlog.Logf(level, format, args...)
} }
func Tracef(format string, args ...interface{}) { func Tracef(format string, args ...interface{}) {
l.Tracef(format, args...) dlog.Tracef(format, args...)
} }
func Debugf(format string, args ...interface{}) { func Debugf(format string, args ...interface{}) {
l.Debugf(format, args...) dlog.Debugf(format, args...)
} }
func Infof(format string, args ...interface{}) { func Infof(format string, args ...interface{}) {
l.Infof(format, args...) dlog.Infof(format, args...)
} }
func Printf(format string, args ...interface{}) { func Printf(format string, args ...interface{}) {
l.Printf(format, args...) dlog.Printf(format, args...)
} }
func Warnf(format string, args ...interface{}) { func Warnf(format string, args ...interface{}) {
l.Warnf(format, args...) dlog.Warnf(format, args...)
} }
func Warningf(format string, args ...interface{}) { func Warningf(format string, args ...interface{}) {
l.Warningf(format, args...) dlog.Warningf(format, args...)
} }
func Errorf(format string, args ...interface{}) { func Errorf(format string, args ...interface{}) {
l.Errorf(format, args...) dlog.Errorf(format, args...)
} }
func Fatalf(format string, args ...interface{}) { func Fatalf(format string, args ...interface{}) {
l.Fatalf(format, args...) dlog.Fatalf(format, args...)
} }
func Panicf(format string, args ...interface{}) { func Panicf(format string, args ...interface{}) {
l.Panicf(format, args...) dlog.Panicf(format, args...)
} }
func Log(level logrus.Level, args ...interface{}) { func Log(level logrus.Level, args ...interface{}) {
l.Log(level, args...) dlog.Log(level, args...)
} }
func Trace(args ...interface{}) { func Trace(args ...interface{}) {
l.Trace(args...) dlog.Trace(args...)
} }
func Debug(args ...interface{}) { func Debug(args ...interface{}) {
l.Debug(args...) dlog.Debug(args...)
} }
func Info(args ...interface{}) { func Info(args ...interface{}) {
l.Info(args...) dlog.Info(args...)
} }
func Print(args ...interface{}) { func Print(args ...interface{}) {
l.Print(args...) dlog.Print(args...)
} }
func Warn(args ...interface{}) { func Warn(args ...interface{}) {
l.Warn(args...) dlog.Warn(args...)
} }
func Warning(args ...interface{}) { func Warning(args ...interface{}) {
l.Warning(args...) dlog.Warning(args...)
} }
func Error(args ...interface{}) { func Error(args ...interface{}) {
l.Error(args...) dlog.Error(args...)
} }
func Fatal(args ...interface{}) { func Fatal(args ...interface{}) {
l.Fatal(args...) dlog.Fatal(args...)
} }
func Panic(args ...interface{}) { func Panic(args ...interface{}) {
l.Panic(args...) dlog.Panic(args...)
} }
func Logln(level logrus.Level, args ...interface{}) { func Logln(level logrus.Level, args ...interface{}) {
l.Logln(level, args...) dlog.Logln(level, args...)
} }
func Traceln(args ...interface{}) { func Traceln(args ...interface{}) {
l.Traceln(args...) dlog.Traceln(args...)
} }
func Debugln(args ...interface{}) { func Debugln(args ...interface{}) {
l.Debugln(args...) dlog.Debugln(args...)
} }
func Infoln(args ...interface{}) { func Infoln(args ...interface{}) {
l.Infoln(args...) dlog.Infoln(args...)
} }
func Println(args ...interface{}) { func Println(args ...interface{}) {
l.Println(args...) dlog.Println(args...)
} }
func Warnln(args ...interface{}) { func Warnln(args ...interface{}) {
l.Warnln(args...) dlog.Warnln(args...)
} }
func Warningln(args ...interface{}) { func Warningln(args ...interface{}) {
l.Warningln(args...) dlog.Warningln(args...)
} }
func Errorln(args ...interface{}) { func Errorln(args ...interface{}) {
l.Errorln(args...) dlog.Errorln(args...)
} }
func Fatalln(args ...interface{}) { func Fatalln(args ...interface{}) {
l.Fatalln(args...) dlog.Fatalln(args...)
} }
func Panicln(args ...interface{}) { func Panicln(args ...interface{}) {
l.Panicln(args...) dlog.Panicln(args...)
} }
func Exit(code int) { func Exit(code int) {
l.Exit(code) dlog.Exit(code)
} }

View File

@ -14,7 +14,7 @@ func DebuglnTrackTime(do func(), fargs ...interface{}) {
var out []interface{} var out []interface{}
out = append(out, t) out = append(out, t)
out = append(out, fargs...) out = append(out, fargs...)
l.Debugln(out...) dlog.Debugln(out...)
} }
func InfolnTrackTime(do func(), fargs ...interface{}) { func InfolnTrackTime(do func(), fargs ...interface{}) {
@ -24,5 +24,5 @@ func InfolnTrackTime(do func(), fargs ...interface{}) {
var out []interface{} var out []interface{}
out = append(out, t) out = append(out, t)
out = append(out, fargs...) out = append(out, fargs...)
l.Infoln(out...) dlog.Infoln(out...)
} }