Function TestBuildUniqueKeyRangeEndPreparedQueryViaTemptable

go/sql/builder_test.go:472–524

(t *testing.T)

}

func TestBuildUniqueKeyRangeEndPreparedQueryViaTemptable(t *testing.T) {
	databaseName := "mydb"
	originalTableName := "tbl"
	var chunkSize int64 = 500
	{
		// Different first-column values → 3-part UNION for efficient boundary seeks.
		uniqueKeyColumns := NewColumnList([]string{"name", "position"})
		rangeStartArgs := []interface{}{3, 17}
		rangeEndArgs := []interface{}{103, 117}

		query, explodedArgs, err := BuildUniqueKeyRangeEndPreparedQueryViaTemptable(databaseName, originalTableName, uniqueKeyColumns, rangeStartArgs, rangeEndArgs, chunkSize, false, "test")
		require.NoError(t, err)
		expected := `
			select /* gh-ost mydb.tbl test */ name, position
			from (
				select name, position
				from
					((select name, position from mydb.tbl where name = ? and position > ? order by name asc, position asc limit 500)
					union all
					(select name, position from mydb.tbl where name > ? and name < ? order by name asc, position asc limit 500)
					union all
					(select name, position from mydb.tbl where name = ? and position <= ? order by name asc, position asc limit 500)) t
				order by name asc, position asc
				limit 500
			) select_osc_chunk
			order by name desc, position desc
			limit 1`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{3, 17, 3, 103, 103, 117}, explodedArgs)
	}
	{
		// Same first-column value → single range query (no UNION needed).
		uniqueKeyColumns := NewColumnList([]string{"name", "position"})
		rangeStartArgs := []interface{}{3, 17}
		rangeEndArgs := []interface{}{3, 117}

		query, explodedArgs, err := BuildUniqueKeyRangeEndPreparedQueryViaTemptable(databaseName, originalTableName, uniqueKeyColumns, rangeStartArgs, rangeEndArgs, chunkSize, false, "test")
		require.NoError(t, err)
		expected := `
			select /* gh-ost mydb.tbl test */ name, position
			from (
				select name, position
				from mydb.tbl
				where (name = ? and position > ? and position <= ?)
				order by name asc, position asc
				limit 500
			) select_osc_chunk
			order by name desc, position desc
			limit 1`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{3, 17, 117}, explodedArgs)
	}
}

func TestBuildUniqueKeyRangeEndPreparedQueryTwoColumnEnum(t *testing.T) {
	databaseName := "mydb"