re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix logger panic on windows

This commit is contained in:
Jinzhu 2020-06-03 14:39:36 +08:00
parent b32658358c
commit 9934207c42
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
utils/utils.go
View File

@ -5,25 +5,24 @@ import (
"fmt"
"path/filepath"
"reflect"
"regexp"
"runtime"
"strconv"
"strings"
"unicode"
)
var goSrcRegexp, goTestRegexp *regexp.Regexp
var gormSourceDir string
func init() {
_, file, _, _ := runtime.Caller(0)
goSrcRegexp = regexp.MustCompile(filepath.Join(filepath.Dir(filepath.Dir(file)), ".*.go"))
goTestRegexp = regexp.MustCompile(filepath.Join(filepath.Dir(filepath.Dir(file)), ".*test.go"))
gormSourceDir = filepath.Dir(filepath.Dir(file))
}
func FileWithLineNum() string {
for i := 2; i < 15; i++ {
_, file, line, ok := runtime.Caller(i)
if ok && (!goSrcRegexp.MatchString(file) || goTestRegexp.MatchString(file)) {
if ok && (!strings.HasPrefix(file, gormSourceDir) || strings.HasSuffix(file, "_test.go")) {
return fmt.Sprintf("%v:%v", file, line)
}
}