Sam Konstas has left Sri Lanka ahead of Australia's second Test in Galle to play Sheffield Shield for NSW.