Java socket connection - how to set a timeout when connecting server restarts

Question

I have a Java socket connection established with a server A. Server A randomly restarts during the day and my Java socket connection infinitely waits because it thinks that