Support partition keys in where