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() defer file.Close()
structFields := utils.GetStructFields(file, structName) structFields := utils.GetStructFields(file, structName)
generation.Generate(structName, structFields) 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") { if strings.Contains(field, "type") {
return nil return nil
} }
startBacktip := strings.Index(field, "`") startBacktip := strings.Index(field, "`")
endBacktip := -1 endBacktip := -1
var metadata []entities.Medatada var metadata []entities.Medatada
@@ -39,19 +38,16 @@ func SplitStructField(field string) *entities.Field {
} else { } else {
startBacktip = len(field) startBacktip = len(field)
} }
field = strings.TrimSpace(field[:startBacktip]) field = strings.TrimSpace(field[:startBacktip])
data := strings.Split(field, " ") data := SplitBySingleSpace(field)
if len(data) < 2 { if len(data) < 2 {
return nil return nil
} }
name := strings.TrimSpace(data[0]) name := data[0]
dataType := data[1]
dataType := strings.TrimSpace(data[1])
return &entities.Field{ return &entities.Field{
Medatada: metadata, Medatada: metadata,
Type: dataType, Type: dataType,