Function getTableColumnsCode

pkg/sql2code/parser/parser.go:806–849

(data tmplData, isEmbed bool)

}

func getTableColumnsCode(data tmplData, isEmbed bool) ([]byte, error) {
	if data.DBDriver == DBDriverMongodb {
		for _, field := range data.Fields {
			if field.Name == "ID" {
				field.ColName = "_id"
				data.Fields = append(data.Fields, field)
				break
			}
		}
	}
	if isEmbed {
		var fields = []tmplField{
			{
				ColName: "id",
			},
			{
				ColName: "created_at",
			},
			{
				ColName: "updated_at",
			},
			{
				ColName: "deleted_at",
			},
		}
		for _, field := range data.Fields {
			if field.Name == __mysqlModel__ {
				continue
			}
			fields = append(fields, field)
		}
		data.Fields = fields
	}
	builder := strings.Builder{}
	err := tableColumnsTmpl.Execute(&builder, data)
	if err != nil {
		return nil, fmt.Errorf("tableColumnsTmpl.Execute error: %v", err)
	}
	code, err := format.Source([]byte(builder.String()))
	if err != nil {
		return nil, fmt.Errorf("tableColumnsTmpl format.Source error: %v", err)
	}
	return code, err
}

func getModelCode(data modelCodes) (string, error) {
	builder := strings.Builder{}