Function TestBuildRangeInsertPreparedQuery

go/sql/builder_test.go:369–418

(t *testing.T)

}

func TestBuildRangeInsertPreparedQuery(t *testing.T) {
	databaseName := "mydb"
	originalTableName := "tbl"
	ghostTableName := "ghost"
	sharedColumns := []string{"id", "name", "position"}
	{
		uniqueKey := "name_position_uidx"
		uniqueKeyColumns := NewColumnList([]string{"name", "position"})
		rangeStartArgs := []interface{}{3, 17}
		rangeEndArgs := []interface{}{103, 117}

		query, explodedArgs, err := BuildRangeInsertPreparedQuery(databaseName, originalTableName, ghostTableName, sharedColumns, sharedColumns, uniqueKey, uniqueKeyColumns, rangeStartArgs, rangeEndArgs, true, true, true)
		require.NoError(t, err)
		expected := `
			insert /* gh-ost mydb.tbl */ ignore
			into mydb.ghost (id, name, position)
			(select id, name, position from mydb.tbl force index (name_position_uidx) where name = ? and position >= ? for share nowait)
			union all
			(select id, name, position from mydb.tbl force index (name_position_uidx) where name > ? and name < ? for share nowait)
			union all
			(select id, name, position from mydb.tbl force index (name_position_uidx) where name = ? and position <= ? for share nowait)`
		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).
		uniqueKey := "name_position_uidx"
		uniqueKeyColumns := NewColumnList([]string{"name", "position"})
		rangeStartArgs := []interface{}{3, 17}
		rangeEndArgs := []interface{}{3, 117}

		query, explodedArgs, err := BuildRangeInsertPreparedQuery(databaseName, originalTableName, ghostTableName, sharedColumns, sharedColumns, uniqueKey, uniqueKeyColumns, rangeStartArgs, rangeEndArgs, true, true, true)
		require.NoError(t, err)
		expected := `
			insert /* gh-ost mydb.tbl */ ignore
			into
				mydb.ghost
				(id, name, position)
			(
				select id, name, position
				from
					mydb.tbl
				force index (name_position_uidx)
					where (name = ? and position >= ? and position <= ?)
					for share nowait
			)`
		require.Equal(t, normalizeQuery(expected), normalizeQuery(query))
		require.Equal(t, []interface{}{3, 17, 117}, explodedArgs)
	}
}

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