feature: impl_interfaces 参数支持正则传参, 默认忽略所有接口

Author: ssoor

generator.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"path"
 	"path/filepath"
+	"regexp"
 	"sort"
 	"strconv"
 	"strings"
@@ -23,8 +24,9 @@ type generator struct {
 	indent                    string
 	mockNames                 map[string]string // may be empty
 	mockInterfaces            map[string]bool   // may be empty
-	filename                  string            // may be empty
-	srcPackage, srcInterfaces string            // may be empty
+	mockInterfaceRegexps      []*regexp.Regexp
+	filename                  string // may be empty
+	srcPackage, srcInterfaces string // may be empty
 	copyrightHeader           string
 
 	packageMap map[string]string // map from import path to package name
@@ -164,8 +166,18 @@ func (g *generator) generate(pkg *model.Package, dstPkg *model.Package, outputPk
 	}
 
 	for _, intf := range pkg.Interfaces {
-		if len(g.mockInterfaces) != 0 && !g.mockInterfaces[intf.Name] {
-			continue
+		if !g.mockInterfaces[intf.Name] {
+			match := false
+			for _, v := range g.mockInterfaceRegexps {
+				if v.Match([]byte(intf.Name)) {
+					match = true
+					break
+				}
+			}
+
+			if !match {
+				continue
+			}
 		}
 
 		sn, exist := namesMap[g.mockName(intf.Name)]

mockgen.go

@@ -30,6 +30,7 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
+	"regexp"
 	"strings"
 	"unicode"
 
@@ -128,8 +129,9 @@ func main() {
 	}
 
 	g := &generator{
-		mockNames:      make(map[string]string),
-		mockInterfaces: make(map[string]bool),
+		mockNames:            make(map[string]string),
+		mockInterfaces:       make(map[string]bool),
+		mockInterfaceRegexps: make([]*regexp.Regexp, 0),
 	}
 	if *destination != "" {
 		g.dstFileName = *destination
@@ -147,7 +149,16 @@ func main() {
 	if *implInterfaces != "" {
 		for _, v := range strings.Split(*implInterfaces, ",") {
 			v := strings.TrimSpace(v)
-			g.mockInterfaces[v] = true
+
+			if v[0] == '^' {
+				exp, err := regexp.Compile(v)
+				if err != nil {
+					log.Fatalf("Failed reading impl_interfaces arguments: %v", err)
+				}
+				g.mockInterfaceRegexps = append(g.mockInterfaceRegexps, exp)
+			} else {
+				g.mockInterfaces[v] = true
+			}
 		}
 	}
 	if *copyrightFile != "" {