sca-java-2.x/contrib/modules/binding-comet-runtime-alternate/src/main/java/org/apache/tuscany/sca/binding/comet/runtime/CometBindingHandler.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

package org.apache.tuscany.sca.binding.comet.runtime;

import java.lang.reflect.InvocationTargetException;

import javax.servlet.ServletContext;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;

import org.apache.tuscany.sca.interfacedef.Operation;
import org.apache.tuscany.sca.runtime.RuntimeEndpoint;
import org.atmosphere.annotation.Broadcast;
import org.atmosphere.cpr.Broadcaster;
import org.atmosphere.cpr.DefaultBroadcaster;
import org.atmosphere.jersey.Broadcastable;
import org.atmosphere.jersey.SuspendResponse;

import com.sun.jersey.spi.container.servlet.PerSession;

@Produces("text/html;charset=ISO-8859-1")
@PerSession
public class CometBindingHandler {

    private Broadcaster broadcaster = new DefaultBroadcaster();
    public static final String ENDPOINT_KEY = "org.apache.tuscany.sca.binding.comet.endpoint";
    public static final String OPERATION_KEY = "org.apache.tuscany.sca.binding.comet.operation";

    @Context
    private ServletContext sc;

    @GET
    public SuspendResponse<String> register() {
        System.out.println("Register method: " + sc);
        return new SuspendResponse.SuspendResponseBuilder<String>().broadcaster(broadcaster).outputComments(true)
            .addListener(new EventsLogger()).build();
    }

    @POST
    @Broadcast
    public Broadcastable publish() throws InvocationTargetException {
        System.out.println("Publish method: " + sc);
        RuntimeEndpoint wire = (RuntimeEndpoint)sc.getAttribute(ENDPOINT_KEY);
        Operation operation = (Operation)sc.getAttribute(OPERATION_KEY);
        Object response = wire.invoke(operation, new Object[] {});
        return new Broadcastable(response.toString(), "", broadcaster);
    }

}