feat: Skip source directory for gen addition

utils/utils.go

@@ -12,6 +12,8 @@ import (
 )
 
 var gormSourceDir string
+var skipSourceDirs []string
+var matchSkipSourceDir = false
 
 func init() {
 	_, file, _, _ := runtime.Caller(0)
@@ -19,15 +21,33 @@ func init() {
 	gormSourceDir = regexp.MustCompile(`utils.utils\.go`).ReplaceAllString(file, "")
 }
 
+func AddSkipSourceDir(dirs ...string) {
+	skipSourceDirs = append(skipSourceDirs, dirs...)
+	if len(skipSourceDirs) > 0 {
+		matchSkipSourceDir = true
+	}
+}
+
 // FileWithLineNum return the file name and line number of the current file
 func FileWithLineNum() string {
 	// the second caller usually from gorm internal, so set i start from 2
 	for i := 2; i < 15; i++ {
 		_, file, line, ok := runtime.Caller(i)
 		if ok && (!strings.HasPrefix(file, gormSourceDir) || strings.HasSuffix(file, "_test.go")) {
+			matchSkip := false
+			if matchSkipSourceDir {
+				for _, dir := range skipSourceDirs {
+					if strings.HasPrefix(file, dir) {
+						matchSkip = true
+						break
+					}
+				}
+			}
+			if !matchSkip {
 				return file + ":" + strconv.FormatInt(int64(line), 10)
 			}
 		}
+	}
 
 	return ""
 }