Arduino web socket autoreconnect

I am using 2 esp8266, both connected to same network.
Esp1 send some data to esp2 using websocket. So basically esp2 connect to esp1 websocket and read data. My problem is that if esp1 is restarted, after is up again, esp2 does not connect anymore to it and needs restart to connect.
This is code I am using on esp2

#include <ESP8266WiFi.h>
#include <WebSocketsClient.h>
#include <ESP8266WebServer.h>
#include <SoftwareSerial.h>

#define USE_SERIAL Serial
#define LEDS 0 //Status led

WebSocketsClient webSocket;
SoftwareSerial COM(5, 4); 

long previousMillis = 0;
long interval = 50;

String data;

void webSocketEvent(WStype_t type, uint8_t * payload, size_t length) {
    switch(type) {
        case WStype_DISCONNECTED:
            yield();
            break;
        case WStype_CONNECTED: {
            webSocket.sendTXT("Connected");
        }
   yield();
            break;
        case WStype_TEXT:
      if(payload[0] == '#') {
        uint32_t dataz = (uint32_t) strtol((const char *) &payload[1], NULL, 10);
        if (dataz < 99911) {
          data = "0" + String(dataz);
        }
        else {
          data = String(dataz);
        }
      }
      yield();
if (data.length() == 6 ) {
COM.print('<'); // start marker
COM.print(data);
COM.println('>'); // end marker
}
       break;
    }

}

void setup() {
  pinMode(LEDS, OUTPUT);
  digitalWrite(LEDS, LOW);
    for(uint8_t t = 4; t > 0; t--) {
        delay(1000);
    }

 WiFi.mode(WIFI_STA);
 WiFi.begin("xxx", "xxx");
 while (WiFi.status() != WL_CONNECTED)
  {
    delay(500);
  }
    // server address, port and URL
    webSocket.begin("192.168.0.30", 81, "/");

    // event handler
    webSocket.onEvent(webSocketEvent);

    // use HTTP Basic Authorization this is optional remove if not needed
    //webSocket.setAuthorization("user", "Password");

    // try ever 5000 again if connection has failed
    webSocket.setReconnectInterval(5000);
  digitalWrite(LEDS, HIGH);
}

void loop() {
  unsigned long currentMillis = millis();
  if(currentMillis - previousMillis > interval) {
        previousMillis = currentMillis;

    webSocket.loop();
  }
}

I would need some help to make it reconnect when the esp1 is available again after was disconnected Thank you very much

Why WebSockets in a local network? use common sockets (WiFiServer and WiFiClient) without HTTP protocol

Do you have an example of this?

Thanks

I start a WiFiServer for example on port 2323 on one module and connect to it with WiFiClient from the other module and then they can interchange data by write, read, print and other functions known from Serial

But do you have an example of working code? When server is turned off , then on again does the client reconnect ?
Also which exactly libs you are using cause there are few versions...
Thanks

for esp8266 there is only one WiFi library, only one WiFiClient and one WiFiServer