如何从Java websocket服务器访问客户端主机名,http标头等?

时间:2015-01-28 09:01:15

标签: java http servlets websocket

我已经遵循了许多教程和示例代码示例,但我还没有看到一种方法来访问客户端的HTTP头,主机名等,就像我们在Servlet的请求对象中一样。

我该如何解决?

假设我已经onOpen定义为 -

@OnOpen
    public void onOpen(Session session) {

} 

在上面的方法中,有没有办法可以使用会话字段访问底层的HTTP连接细节?即使我可以访问底层Servlet(如果有的话),我也没关系

2 个答案:

答案 0 :(得分:1)

Tyrus user guide的第4.1.1.5章。从ServerEnpointConfig.Configurator获取信息到端点实例需要一些工作量,但是可以完成。 (见ModifyRequestResponseHeadersTest.java

答案 1 :(得分:0)

基于link1link2

我终于想通了我们可以使用以下两个类获取客户端IP,实际上你可以使用公开的httpservletRequest做更多的事情......

ServletAwareConfigurator .java

package examples;
import java.lang.reflect.Field;

import javax.servlet.http.HttpServletRequest;

import javax.websocket.HandshakeResponse;
import javax.websocket.server.HandshakeRequest;
import javax.websocket.server.ServerEndpointConfig;
import javax.websocket.server.ServerEndpointConfig.Configurator;

public class ServletAwareConfigurator extends ServerEndpointConfig.Configurator {

    @Override
    public void modifyHandshake(ServerEndpointConfig config, HandshakeRequest request, HandshakeResponse response) {
        HttpServletRequest httpservletRequest = getField(request, HttpServletRequest.class);
        String sClientIP = httpservletRequest.getRemoteAddr();
        config.getUserProperties().put("clientip", sClientIP);
        // ...
    }

    //hacking reflector to expose fields...
    private static < I, F > F getField(I instance, Class < F > fieldType) {
        try {
            for (Class < ? > type = instance.getClass(); type != Object.class; type = type.getSuperclass()) {
                for (Field field: type.getDeclaredFields()) {
                    if (fieldType.isAssignableFrom(field.getType())) {
                        field.setAccessible(true);
                        return (F) field.get(instance);
                    }
                }
            }
        } catch (Exception e) {
            // Handle?
        }
        return null;
    }
}

<强> GetHttpSessionSocket.java

package examples;
import java.io.IOException;
import javax.servlet.http.HttpSession;
import javax.websocket.EndpointConfig;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint(value = "/example",
    configurator = ServletAwareConfigurator.class)
public class GetHttpSessionSocket {
    private Session wsSession;
    private String sClientIP;

    @OnOpen
    public void open(Session session, EndpointConfig config) {
        this.wsSession = session;
        this.sClientIP = (String) config.getUserProperties()
            .get("clientip");
    }
    @OnMessage
    public void echo(String msg) throws IOException {
        wsSession.getBasicRemote().sendText(msg);
    }
}