in.erail.service.HelloService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of api-framework-amazon-lambda Show documentation

Lib to run API Framework code on Amazon Lambda

There is a newer version: 2.5.3

Show newest version

package in.erail.service;

import com.google.common.net.MediaType;
import in.erail.model.Event;
import io.reactivex.Maybe;
import io.reactivex.MaybeSource;
import io.vertx.core.json.JsonArray;
import io.vertx.core.json.JsonObject;
import java.util.Optional;

public class HelloService extends RESTServiceImpl {

  private JsonArray mHelloData = new JsonArray();

  public HelloService() {
    mHelloData.add("S1");
    mHelloData.add("S2");
    mHelloData.add("S3");
    mHelloData.add("S4");
    mHelloData.add("S5");
  }

  @Override
  public MaybeSource process(Maybe pEvent) {
    return pEvent
            .doOnSuccess(e -> {

              int contentLength = e.getRequest().bodyAsString().length();
              String header = Optional.ofNullable(e.getRequest().getHeaders().get("aws-event-source")).orElse("");

              byte[] result;

              if (contentLength == 0) {
                result = getHelloData().toString().getBytes();
              } else {
                JsonObject newResp = new JsonObject();
                newResp.put("length", contentLength);
                newResp.put("aws-event-source", header);
                result = newResp.toString().getBytes();
              }

              e
                      .getResponse()
                      .setBody(result)
                      .setMediaType(MediaType.JSON_UTF_8);
            });
  }

  public JsonArray getHelloData() {
    return mHelloData;
  }

  public void setHelloData(JsonArray pHelloData) {
    this.mHelloData = pHelloData;
  }

}