Connection was refused when IP address of NN pod changed

Hi,
   I checked the source code of the HDFS and fount out that HDFS client builds a socket with the IP address of NN.  The virtual IP would change if NN restarts, accordingly, Rpc Connection from HDFS client to NN will be refused. Any suggestions to solve this problem?