Merge pull request #10 from go-gorm/lzq

little fix

internal/check/checkstruct.go

@@ -42,10 +42,15 @@ func (b *BaseStruct) getTableName(st interface{}) {
 	b.TableName = stmt.Table
 }
 
-// check if struct is exportable and if member's type is regular
+// check if struct is exportable and if struct in main package and if member's type is regular
 func (b *BaseStruct) check() (err error) {
+	if b.StructInfo.InMainPkg() {
+		err = fmt.Errorf("can't generated data object for struct in main package,ignored:%s", b.StructName)
+		log.Println(err)
+		return
+	}
 	if !isCapitalize(b.StructName) {
-		err = fmt.Errorf("ignoring non exportable struct name:%s", b.NewStructName)
+		err = fmt.Errorf("can't generated data object for non-exportable struct,ignore:%s", b.NewStructName)
 		log.Println(err)
 		return
 	}

internal/parser/export.go

@@ -34,7 +34,7 @@ func GetInterfacePath(v interface{}) (paths []*InterfacePath, err error) {
 		}
 
 		if strings.Split(arg.String(), ".")[0] == "main" {
-			_, file, _, ok := runtime.Caller(2)
+			_, file, _, ok := runtime.Caller(3)
 			if ok {
 				path.Files = append(path.Files, file)
 			}

internal/parser/parser.go

@@ -120,6 +120,10 @@ func (p *Param) IsNull() bool {
 	return p.Package == "" && p.Type == "" && p.Name == ""
 }
 
+func (p *Param) InMainPkg() bool {
+	return p.Package == "main"
+}
+
 func (p *Param) IsTime() bool {
 	return p.Package == "time" && p.Type == "Time"
 }