Implement shanghai jtk balance query