Function TestBuildUniqueKeyRangeEndPreparedQueryViaOffset

go/sql/builder_test.go:420–470

(t *testing.T)

}

func TestBuildUniqueKeyRangeEndPreparedQueryViaOffset(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 := BuildUniqueKeyRangeEndPreparedQueryViaOffset(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 > ? 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 1
			offset 499`
		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 := BuildUniqueKeyRangeEndPreparedQueryViaOffset(databaseName, originalTableName, uniqueKeyColumns, rangeStartArgs, rangeEndArgs, chunkSize, false, "test")
		require.NoError(t, err)
		expected := `
			select /* gh-ost mydb.tbl test */
				name, position
			from
				mydb.tbl
			where
				(name = ? and position > ? and position <= ?)
			order by
				name asc, position asc
			limit 1
			offset 499`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{3, 17, 117}, explodedArgs)
	}
}

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