fix:set package name

a631807682 · Aug 17, 2021 · ec87a8b · ec87a8b
1 parent 95c3e48
commit ec87a8b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/internal/check/checkinterface.go b/internal/check/checkinterface.go
@@ -133,6 +133,10 @@ func (f *InterfaceMethod) checkResult(result []parser.Param) (err error) {
 			if !f.ResultData.IsNull() {
 				return fmt.Errorf("query method cannot return more than 1 data value in [%s.%s]", f.InterfaceName, f.MethodName)
 			}
+			if param.Package == "" && !(param.AllowType() || param.IsMap() || param.IsTime()) {
+				param.Package = f.Package
+			}
+
 			param.SetName("result")
 			f.ResultData = param
 			//return fmt.Errorf("illegal parameter：%s.%s on struct %s.%s generated method %s", param.Package, param.Type, f.OriginStruct.Package, f.OriginStruct.Type, f.MethodName)

diff --git a/internal/parser/export.go b/internal/parser/export.go
@@ -29,8 +29,6 @@ func GetInterfacePath(v interface{}) (paths []*InterfacePath, err error) {
 		arg := value.Type().In(i)
 		path.FullName = arg.String()
 
-		path.Package = arg.String()
-
 		// keep the last model
 		for _, n := range strings.Split(arg.String(), ".") {
 			path.Name = n

diff --git a/internal/parser/parser.go b/internal/parser/parser.go
@@ -40,7 +40,7 @@ func (i *InterfaceSet) ParseFile(paths []*InterfacePath) error {
 				return fmt.Errorf("file not found：%s", file)
 			}
 
-			err = i.getInterfaceFromFile(absFilePath, path.Name, path.Package)
+			err = i.getInterfaceFromFile(absFilePath, path.Name, path.FullName)
 			if err != nil {
 				return fmt.Errorf("can't get interface from %s:%s", path.FullName, err)
 			}