com.ociweb.ExampleRestStage Maven / Gradle / Ivy

Go to download
package com.ociweb;

import com.ociweb.json.encode.JSONRenderer;
import com.ociweb.pronghorn.network.config.HTTPContentType;
import com.ociweb.pronghorn.network.config.HTTPContentTypeDefaults;
import com.ociweb.pronghorn.network.config.HTTPHeader;
import com.ociweb.pronghorn.network.config.HTTPRevision;
import com.ociweb.pronghorn.network.config.HTTPSpecification;
import com.ociweb.pronghorn.network.config.HTTPVerb;
import com.ociweb.pronghorn.network.config.HTTPVerbDefaults;
import com.ociweb.pronghorn.network.module.AbstractAppendablePayloadResponseStage;
import com.ociweb.pronghorn.network.schema.HTTPRequestSchema;
import com.ociweb.pronghorn.network.schema.ServerResponseSchema;
import com.ociweb.pronghorn.pipe.ChannelReader;
import com.ociweb.pronghorn.pipe.Pipe;
import com.ociweb.pronghorn.pipe.StructuredReader;
import com.ociweb.pronghorn.stage.scheduling.GraphManager;
import com.ociweb.pronghorn.util.AppendableByteWriter;

public class ExampleRestStage<      T extends Enum & HTTPContentType,
									R extends Enum & HTTPRevision,
									V extends Enum & HTTPVerb,
									H extends Enum & HTTPHeader> extends AbstractAppendablePayloadResponseStage {

    private static final byte[] HEY = "Hey, ".getBytes();
	private static final byte[] BANG = "!".getBytes();
	
	private static final JSONRenderer jsonRenderer = new JSONRenderer()
            .beginObject()
            .string("message", (reader,target) ->  {target.write(HEY); reader.readText(WebFields.name, target).write(BANG);} )
            .bool("happy", reader -> !reader.readBoolean(WebFields.happy))
            .integer("age", reader -> reader.readInt(WebFields.age) * 2)
            .endObject();
	
	public static ExampleRestStage newInstance(GraphManager graphManager, 
			Pipe inputPipes,
			Pipe outputPipe, 
			HTTPSpecification httpSpec) {
		return new ExampleRestStage(graphManager, inputPipes, outputPipe, httpSpec);
	}
	
	public ExampleRestStage(GraphManager graphManager, 
			Pipe inputPipes,
			Pipe outputPipes, 
			HTTPSpecification httpSpec) {
		
		super(graphManager, join(inputPipes), join(outputPipes), httpSpec, 1<<8);

	}

	@Override
	public HTTPContentType contentType() {
		return HTTPContentTypeDefaults.JSON;
	}

	@Override
	protected boolean payload(AppendableByteWriter payload, GraphManager gm, 
			                 ChannelReader params, HTTPVerbDefaults verb) {	
		jsonRenderer.render(payload, params.structured());
		return true;
	}

}