Function ConvertToSQLByMgoFields

pkg/sql2code/parser/mongodb.go:238–276

ConvertToSQLByMgoFields convert to mysql table ddl

(tableName string, fields []*MgoField)

// ConvertToSQLByMgoFields convert to mysql table ddl
func ConvertToSQLByMgoFields(tableName string, fields []*MgoField) (string, map[string]string) {
	isHaveID := false
	fieldStr := ""
	srcMongoTypeMap := make(map[string]string) // name:type
	objectStrs := []string{}
	protoObjectStrs := []string{}

	for _, field := range fields {
		switch field.Type {
		case goTypeInterface, goTypeSliceInterface:
			srcMongoTypeMap[field.Name] = xstrings.ToCamelCase(field.Name)
		default:
			srcMongoTypeMap[field.Name] = field.Type
		}
		if field.Name == oidName {
			isHaveID = true
			srcMongoTypeMap["id"] = field.Type
			continue
		}

		fieldStr += fmt.Sprintf("    `%s` %s,\n", field.Name, convertMongoToMysqlType(field.Type))
		if field.ObjectStr != "" {
			objectStrs = append(objectStrs, field.ObjectStr)
			protoObjectStrs = append(protoObjectStrs, field.ProtoObjectStr)
		}
	}

	fieldStr = strings.TrimSuffix(fieldStr, ",\n")
	if isHaveID {
		fieldStr = "    `id` varchar(24),\n" + fieldStr + ",\n    PRIMARY KEY (id)"
	}

	if len(objectStrs) > 0 {
		srcMongoTypeMap[SubStructKey] = strings.Join(objectStrs, "\n") + "\n"
		srcMongoTypeMap[ProtoSubStructKey] = strings.Join(protoObjectStrs, "\n") + "\n"
	}

	return fmt.Sprintf("CREATE TABLE `%s` (\n%s\n);", tableName, fieldStr), srcMongoTypeMap
}

// nolint
func convertMongoToMysqlType(goType string) string {