Function getTableStmt

backend/plugin/db/starrocks/dump.go:416–461

getTableStmt gets the create statement of a table.

(db *sql.DB, dbName, tblName, tblType string)

// getTableStmt gets the create statement of a table.
func getTableStmt(db *sql.DB, dbName, tblName, tblType string) (string, error) {
	switch tblType {
	case baseTableType:
		query := fmt.Sprintf("SHOW CREATE TABLE `%s`.`%s`;", dbName, tblName)
		var stmt, unused string
		if err := db.QueryRow(query).Scan(&unused, &stmt); err != nil {
			if err == sql.ErrNoRows {
				return "", common.FormatDBErrorEmptyRowWithQuery(query)
			}
			return "", err
		}
		// Remove trailing semicolon if present, as the format string will add one
		stmt = strings.TrimSuffix(stmt, ";")
		return fmt.Sprintf(tableStmtFmt, tblName, stmt), nil
	case viewTableType:
		// This differs from mysqldump as it includes.
		query := fmt.Sprintf("SHOW CREATE VIEW `%s`.`%s`;", dbName, tblName)
		var createStmt, unused string
		if err := db.QueryRow(query).Scan(&unused, &createStmt, &unused, &unused); err != nil {
			if err == sql.ErrNoRows {
				return "", common.FormatDBErrorEmptyRowWithQuery(query)
			}
			return "", err
		}
		// Remove trailing semicolon if present, as the format string will add one
		createStmt = strings.TrimSuffix(createStmt, ";")
		return fmt.Sprintf(viewStmtFmt, tblName, createStmt), nil
	case materializedViewType:
		// For Doris materialized views, we use SHOW CREATE MATERIALIZED VIEW.
		// This command returns 2 columns: materialized view name and create statement.
		query := fmt.Sprintf("SHOW CREATE MATERIALIZED VIEW `%s`.`%s`;", dbName, tblName)
		var createStmt, unused string
		if err := db.QueryRow(query).Scan(&unused, &createStmt); err != nil {
			if err == sql.ErrNoRows {
				return "", common.FormatDBErrorEmptyRowWithQuery(query)
			}
			return "", err
		}
		// Remove trailing semicolon if present, as the format string will add one
		createStmt = strings.TrimSuffix(createStmt, ";")
		return fmt.Sprintf(materializedViewStmtFmt, tblName, createStmt), nil
	default:
		slog.Debug("skip unsupported table type", slog.String("dbName", dbName), slog.String("tableName", tblName), slog.String("tableType", tblType))
		return "", nil
	}
}

// getViewColumns gets the create statement of a table.
func getViewColumns(db *sql.DB, dbName, tblName string) ([]string, error) {