fix:兼容多module，main不在逻辑module中的情况

cmd/build.go

@@ -83,6 +83,7 @@ func runBuild(args []string, wd string) {
 		ModRootPath:              gocBuild.ModRootPath,
 		OneMainPackage:           true, // it is a go build
 		GlobalCoverVarImportPath: gocBuild.GlobalCoverVarImportPath,
+		CoverModName:             "coverPackageMod",
 	}
 	err = cover.Execute(ci)
 	if err != nil {

cmd/cover.go

@@ -53,6 +53,7 @@ func runCover(target string) {
 		Center:         center,
 		Singleton:      singleton,
 		OneMainPackage: false,
+		CoverModName:   "coverPackageMod",
 	}
 	_ = cover.Execute(ci)
 }

pkg/build/gomodules.go

@@ -78,7 +78,8 @@ func (b *Build) updateGoModFile() (updateFlag bool, newModFile []byte, err error
 		// absolute path no need to rewrite
 		if newVersion == "" && !filepath.IsAbs(newPath) {
 			var absPath string
-			fullPath := filepath.Join(b.ModRoot, newPath)
+			//替换原路径为目标路径
+			fullPath := filepath.Join(b.TmpDir, newPath)
 			absPath, _ = filepath.Abs(fullPath)
 			// DropReplace & AddReplace will not return error
 			// so no need to check the error

pkg/cover/cover.go

@@ -141,6 +141,7 @@ type CoverInfo struct {
 	AgentPort                string
 	Center                   string
 	Singleton                bool
+	CoverModName             string
 }
 
 // Execute inject cover variables for all the .go files in the target folder
@@ -154,6 +155,7 @@ func Execute(coverInfo *CoverInfo) error {
 	center := coverInfo.Center
 	singleton := coverInfo.Singleton
 	globalCoverVarImportPath := coverInfo.GlobalCoverVarImportPath
+	coverPackageMod := coverInfo.CoverModName
 
 	if coverInfo.IsMod {
 		globalCoverVarImportPath = filepath.Join(coverInfo.ModRootPath, globalCoverVarImportPath)
@@ -170,7 +172,7 @@ func Execute(coverInfo *CoverInfo) error {
 		listArgs = append(listArgs, args)
 	}
 	listArgs = append(listArgs, "./...")
-	pkgs, err := ListPackages(target, strings.Join(listArgs, " "), newGopath)
+	pkgs, err := ListPackagesInAllModule(target, strings.Join(listArgs, " "), newGopath)
 	if err != nil {
 		log.Errorf("Fail to list all packages, the error: %v", err)
 		return err
@@ -183,7 +185,7 @@ func Execute(coverInfo *CoverInfo) error {
 		if pkg.Name == "main" {
 			log.Printf("handle package: %v", pkg.ImportPath)
 			// inject the main package
-			mainCover, mainDecl := AddCounters(pkg, mode, globalCoverVarImportPath)
+			mainCover, mainDecl := AddCounters(pkg, mode, coverPackageMod)
 			allDecl += mainDecl
 			// new a testcover for this service
 			tc := TestCover{
@@ -206,7 +208,7 @@ func Execute(coverInfo *CoverInfo) error {
 
 				//only focus package neither standard Go library nor dependency library
 				if depPkg, ok := pkgs[dep]; ok {
-					packageCover, depDecl := AddCounters(depPkg, mode, globalCoverVarImportPath)
+					packageCover, depDecl := AddCounters(depPkg, mode, coverPackageMod)
 					allDecl += depDecl
 					tc.DepsCover = append(tc.DepsCover, packageCover)
 					seen[dep] = packageCover
@@ -267,6 +269,63 @@ func ListPackages(dir string, args string, newgopath string) (map[string]*Packag
 	return pkgs, nil
 }
 
+// ListPackagesInAllModule 递归地列出指定目录及其子模块中的所有Go包
+func ListPackagesInAllModule(dir string, args string, newgopath string) (map[string]*Package, error) {
+	pkgs := make(map[string]*Package)
+
+	// processModule 处理一个Go模块目录，运行go list并收集包信息
+	processModule := func(modDir string) error {
+		cmd := exec.Command("/bin/bash", "-c", "go list "+args)
+		cmd.Dir = modDir
+		if newgopath != "" {
+			cmd.Env = append(os.Environ(), "GOPATH="+newgopath)
+		}
+
+		var errbuf bytes.Buffer
+		cmd.Stderr = &errbuf
+
+		out, err := cmd.Output()
+		if err != nil {
+			return fmt.Errorf("%s: %v", errbuf.String(), err)
+		}
+
+		dec := json.NewDecoder(bytes.NewReader(out))
+		for {
+			var pkg Package
+			if err := dec.Decode(&pkg); err == io.EOF {
+				break
+			} else if err != nil {
+				return err
+			}
+			pkgs[pkg.ImportPath] = &pkg
+		}
+
+		return nil
+	}
+	// 遍历目录查找子模块
+	err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if info.IsDir() && info.Name() == "vendor" {
+			return filepath.SkipDir // 跳过 vendor 目录
+		}
+		if info.Name() == "go.mod" {
+			modDir := filepath.Dir(path)
+			if err := processModule(modDir); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return pkgs, nil
+}
+
 // AddCounters is different from official go tool cover
 // 1. only inject covervar++ into source file
 // 2. no declarartions for these covervars

pkg/cover/instrument.go

@@ -17,10 +17,15 @@
 package cover
 
 import (
+	"bufio"
 	"fmt"
+	log "github.com/sirupsen/logrus"
+	"io/ioutil"
 	"os"
+	"os/exec"
 	"path"
 	"path/filepath"
+	"strings"
 	"text/template"
 )
 
@@ -61,7 +66,7 @@
 	"syscall"
 	"testing"
 
-	_cover {{.GlobalCoverVarImportPath | printf "%q"}}
+	_cover "coverPackageMod"
 
 )
 
@@ -511,6 +516,188 @@
 		return err
 	}
 	_, err = coverFile.WriteString(content)
+	if err != nil {
+		log.Errorf("err:", err)
+		return err
+	}
+
+	goVersion, err := getModuleGoVersion(filepath.Join(ci.Target, "go.mod"))
+	if goVersion == "" || err != nil {
+		goVersion = "1.13" //其他情况默认使用1.13版本
+	}
+
+	//将cover.go所在package模块化
+	modFilePath := filepath.Join(ci.Target, ci.GlobalCoverVarImportPath, "go.mod")
+	modContent := fmt.Sprintf("module %s\n\ngo %s\n", extractSuffix(ci.GlobalCoverVarImportPath), goVersion)
+	modFile, err := os.Create(modFilePath)
+	if err != nil {
+		log.Errorf("create modFile err:", err)
+		return err
+	}
+	defer modFile.Close()
+
+	_, err = modFile.WriteString(modContent)
+	if err != nil {
+		log.Errorf("modFile write err:", err)
+		return err
+	}
+
+	// 更新所有模块的 go.mod 文件以包含对新模块的引用
+	err = UpdateAllModuleDependencies(ci.Target, ci.GlobalCoverVarImportPath, ci.CoverModName)
+	if err != nil {
+		return err
+	}
+	startDir := ci.Target // 从当前目录开始搜索，你可以修改为任何起始路径
+	err = filepath.Walk(startDir, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		// Skip if it's not a directory
+		if !info.IsDir() {
+			return nil
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		log.Errorf("Error walking the path %q: %v\n", startDir, err)
+	}
 
 	return err
 }
+
+func extractSuffix(input string) string {
+	// Find the last index of "/"
+	lastIndex := strings.LastIndex(input, "/")
+	if lastIndex == -1 {
+		return input // 如果没有找到"/"，则返回整个字符串
+	}
+	// 返回最后一个"/"之后的子串
+	return input[lastIndex+1:]
+}
+
+// UpdateAllModuleDependencies 递归修改所有模块替换依赖
+func UpdateAllModuleDependencies(moduleRootPath, modFileDir, coverPackageMod string) error {
+	// 获取 cover.go 文件所在的模块名称
+	coverModFilePath := filepath.Join(moduleRootPath, modFileDir, "go.mod")
+	coverModuleName, err := getModuleName(coverModFilePath)
+	if err != nil {
+		return err
+	}
+	coverLocalPath := filepath.Join(moduleRootPath, modFileDir)
+	// 遍历根目录下的所有子目录
+	return filepath.WalkDir(moduleRootPath, func(path string, d os.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+
+		// 忽略非目录项
+		if !d.IsDir() {
+			return nil
+		}
+
+		// 检查目录下是否有 go.mod 文件
+		thisModFilePath := filepath.Join(path, "go.mod")
+		if _, err := os.Stat(thisModFilePath); err == nil {
+			// 获取当前模块名称
+			currentModuleName, err := getModuleName(thisModFilePath)
+			if err != nil {
+				log.Errorf("err:", err)
+				return err
+			}
+
+			// 如果当前模块是 cover.go 所在的模块，则跳过
+			if currentModuleName == coverModuleName {
+				return nil
+			}
+
+			// 添加依赖
+			return addLocalDependencyToModFile(thisModFilePath, coverPackageMod, coverLocalPath, coverModuleName)
+		}
+		return nil
+	})
+}
+
+// addLocalDependencyToModFile 替换依赖
+func addLocalDependencyToModFile(modFilePath, importPath, localPath, rootModuleName string) error {
+	// 获取当前模块名称
+	currentModuleName, err := getModuleName(modFilePath)
+	if err != nil {
+		return err
+	}
+
+	// 如果当前模块是根模块，则跳过
+	if currentModuleName == rootModuleName {
+		return nil
+	}
+
+	// 添加 replace 指令
+	replaceCmd := exec.Command("go", "mod", "edit", fmt.Sprintf("-replace=%s=%s", importPath, localPath), modFilePath)
+	replaceOutput, err := replaceCmd.CombinedOutput()
+	if err != nil {
+		log.Errorf("go mod edit -replace output:\n%s\n", string(replaceOutput))
+		return fmt.Errorf("failed to add replace for %s: %w", importPath, err)
+	}
+
+	// 添加 require 指令
+	requireCmd := exec.Command("go", "mod", "edit", fmt.Sprintf("-require=%[email protected]", importPath), modFilePath)
+	requireOutput, err := requireCmd.CombinedOutput()
+	if err != nil {
+		log.Infof("go mod edit -require output:\n%s\n", string(requireOutput))
+		return fmt.Errorf("failed to add require for %s: %w", importPath, err)
+	}
+
+	return nil
+}
+
+// getModuleName 获取module名
+func getModuleName(modFilePath string) (string, error) {
+	// 读取 go.mod 文件内容
+	content, err := ioutil.ReadFile(modFilePath)
+	if err != nil {
+		return "", err
+	}
+
+	// 查找 module 语句
+	lines := strings.Split(string(content), "\n")
+	for _, line := range lines {
+		if strings.HasPrefix(line, "module ") {
+			// 获取模块名称
+			return strings.TrimSpace(strings.TrimPrefix(line, "module ")), nil
+		}
+	}
+
+	return "", fmt.Errorf("module directive not found in %s", modFilePath)
+}
+
+// getModuleGoVersion 获取mudule的版本号
+func getModuleGoVersion(modFilePath string) (string, error) {
+
+	// 打开文件
+	file, err := os.Open(modFilePath)
+	if err != nil {
+		log.Fatalf("Error opening go.mod file: %v", err)
+	}
+	defer file.Close()
+
+	// 创建文件的 bufio.Reader
+	scanner := bufio.NewScanner(file)
+
+	// 读取文件行并查找 Go 版本
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, "go ") {
+			// 找到 Go 版本行
+			version := strings.TrimSpace(strings.TrimPrefix(line, "go"))
+			return version, nil
+		}
+	}
+
+	if err := scanner.Err(); err != nil {
+		log.Fatalf("Error reading go.mod file: %v", err)
+		return "", err
+	}
+	return "", nil
+}