grails javaee websocket getting 404 error in production mode

Question

Im implementing a websocket server in grails and im getting error on client side connecting to it but only in production mode. This is the error im getting:

WebSocket connection to 'ws://localhost:8080/myApp/websocketDoc' failed: Error during WebSocket handshake: Unexpected response code: 404

Everything works fine in development mode.

I have a class in my src/groovy

package myApp;

import java.util.HashMap;

import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import javax.servlet.annotation.WebListener;
import javax.websocket.DeploymentException;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerContainer;
import javax.websocket.server.ServerEndpoint;

import org.codehaus.groovy.grails.web.json.JSONObject;
import org.codehaus.groovy.grails.web.json.JSONArray;

import org.codehaus.groovy.grails.web.util.WebUtils;

import grails.util.Environment;

@WebListener
@ServerEndpoint("/websocketDoc")
public class WebsocketServletContextListener implements ServletContextListener {

private static final JSONArray jsonOpenedDocuments = new JSONArray();

@Override
public void contextInitialized(ServletContextEvent servletContextEvent) {
    final ServerContainer serverContainer = (ServerContainer) servletContextEvent.getServletContext()
                                                .getAttribute("javax.websocket.server.ServerContainer");

    try {
        if (Environment.current == Environment.DEVELOPMENT) {
            serverContainer.addEndpoint(WebsocketServletContextListener)
        }

    } catch (DeploymentException e) {
        e.printStackTrace();
    }
}

@Override
public void contextDestroyed(ServletContextEvent servletContextEvent) {
}
@OnOpen
public void handleOpen() {
    System.out.println("Client is now connected.");
}
@OnMessage
public String handleMessage(String requestStr, Session userSession) {


    return "msg";
}
@OnClose
public void handeClose(Session userSession) {
    System.out.println("Client is now disconnected. id:" + userSession.id);
}
@OnError
public void handleError(Throwable t) {
    t.printStackTrace();
}

}

In my dependencies, BuilConfig.groovy i have:

dependencies {
    runtime 'mysql:mysql-connector-java:5.1.22'

    compile 'c3p0:c3p0:0.9.1.2'

    bundle('javax.websocket:javax.websocket-api:1.1') {
        export = false
    }

}

To add listener to the web.xml i have the following script in /scripts:

import groovy.xml.StreamingMarkupBuilder

eventWebXmlEnd = {String tmpfile ->
def root = new XmlSlurper().parse(webXmlFile)
root.appendNode {
   'listener' {
       'listener-class' (
           'myApp.WebsocketServletContextListener'
       )
    }
}

webXmlFile.text = new StreamingMarkupBuilder().bind {
    mkp.declareNamespace(
            "": "http://java.sun.com/xml/ns/javaee")
    mkp.yield(root)
}
}

In the client side i get the error in the following line:

var webSocketUrl = "${createLink(uri: '/websocketDoc', absolute: true).replaceFirst(/https/, /wss/).replaceFirst(/http/, /ws/)}";

var webSocket = new WebSocket(webSocketUrl); // getting 404 error when running in production mode

Any clue why this only works in development mode? Thanks