fix: split

This commit is contained in:
2025-02-17 22:35:48 +07:00
parent a04285b63f
commit c1a21a5fa4
3 changed files with 17 additions and 8 deletions

View File

@@ -16,6 +16,5 @@ func main() {
defer file.Close()
structFields := utils.GetStructFields(file, structName)
generation.Generate(structName, structFields)
}

View File

@@ -0,0 +1,14 @@
package utils
import "strings"
func SplitBySingleSpace(input string) []string {
parts := strings.Split(strings.TrimSpace(input), " ")
var result []string
for _, p := range parts {
if p != "" {
result = append(result, p)
}
}
return result
}

View File

@@ -11,7 +11,6 @@ func SplitStructField(field string) *entities.Field {
if strings.Contains(field, "type") {
return nil
}
startBacktip := strings.Index(field, "`")
endBacktip := -1
var metadata []entities.Medatada
@@ -39,19 +38,16 @@ func SplitStructField(field string) *entities.Field {
} else {
startBacktip = len(field)
}
field = strings.TrimSpace(field[:startBacktip])
data := strings.Split(field, " ")
data := SplitBySingleSpace(field)
if len(data) < 2 {
return nil
}
name := strings.TrimSpace(data[0])
dataType := strings.TrimSpace(data[1])
name := data[0]
dataType := data[1]
return &entities.Field{
Medatada: metadata,
Type: dataType,