Support multiple primary key on get/update function