Method parse

tableparser/tableparser.go:156–213

()

}

func (t *Table) parse() error {
	//                           +--------------------------- field type
	//                           |          +---------------- field size / enum values: decimal(10,2) or enum('a','b')
	//                           |          |     +---------- extra info (unsigned, etc)
	//                           |          |     |
	re := regexp.MustCompile(`^(.*?)(?:\((.*?)\)(.*))?$`)

	query := "SELECT * FROM `information_schema`.`COLUMNS` WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? ORDER BY ORDINAL_POSITION"

	constraints := constraintsAsMap(t.Constraints)

	rows, err := t.conn.Query(query, t.Schema, t.Name)
	if err != nil {
		return err
	}
	defer rows.Close()

	cols, err := rows.Columns()
	if err != nil {
		return errors.Wrap(err, "Cannot get column names")
	}

	for rows.Next() {
		var f Field
		var allowNull string
		fields := makeScanRecipients(&f, &allowNull, cols)
		err := rows.Scan(fields...)
		if err != nil {
			log.Errorf("Cannot get table fields: %s", err)
			continue
		}

		allowedValues := []string{}
		if f.DataType == "enum" || f.DataType == "set" {
			m := re.FindStringSubmatch(f.ColumnType)
			if len(m) < 2 {
				continue
			}
			vals := strings.Split(m[2], ",")
			for _, val := range vals {
				val = strings.TrimPrefix(val, "'")
				val = strings.TrimSuffix(val, "'")
				allowedValues = append(allowedValues, val)
			}
		}

		f.SetEnumVals = allowedValues
		f.IsNullable = allowNull == "YES"
		f.Constraint = constraints[f.ColumnName]

		t.Fields = append(t.Fields, f)
	}

	if rows.Err() != nil {
		return rows.Err()
	}
	return nil
}