Implementing services

§Implementing services

Services are implemented by providing an implementation of the service descriptor interface, implementing each call specified by that descriptor.

For example, here’s an implementation of the HelloService descriptor:

import com.lightbend.lagom.javadsl.api.*;
import akka.NotUsed;
import static java.util.concurrent.CompletableFuture.completedFuture;

public class HelloServiceImpl implements HelloService {

    public ServiceCall<String, String> sayHello() {
        return name -> completedFuture("Hello " + name);
    }
}

As you can see, the sayHello() method is implemented using a lambda. An important thing to realise here is that the invocation of sayHello() itself does not execute the call, it only returns the call to be executed. The advantage here is that when it comes to composing the call with other cross cutting concerns, such as authentication, this can easily be done using ordinary function based composition.

If you’ve used a Java based web framework before, you may be familiar with using annotations for composition of cross cutting concerns. Annotations have their limits - they don’t compose, that is, if you have two different annotations that you want to apply to many different methods, it’s not straight forward to simply create a new annotation that combines them. In contrast, functions are just methods, if you want to compose two methods together, you create a new method that invokes both of them. Additionally, you’re in complete control over how they get composed, you know exactly what order they are composed in, as opposed to annotations where it’s up to the framework to magically read them via reflection and somehow gain meaning from that.

Now let’s have a look at our ServiceCall interface again:

interface ServiceCall<Request, Response> {
  CompletionStage<Response> invoke(Request request);
}

It will take the request, and return the response as a CompletionStage. If you’ve never seen CompletionStage before, it is a promise. When an API returns a promise, that value might not yet be computed, but the API promises that at some point in future, it will be. Since the value isn’t computed yet, you can’t interact with it immediately. What you can do though is attach callbacks that transform the promise to a promise of a new value, using the thenApply and thenCompose methods. CompletionStage with its thenApply and thenCompose methods are fundamental building blocks for building reactive applications in Java, they allow your code to be asynchronous, not waiting for things to happen, but attaching callback that react to computations being completed.

Of course, a simple hello world computation is not asynchronous, all it needs is to concatenate two Strings, and that returns immediately. In this case, we need to wrap the result of that in a CompletionStage. This can be done by calling CompletableFuture.completedFuture(), which returns a subclass of CompletionStage wrapping an immediately available value. As a convenience, Lagom provides a sync method on Service that does this for you. It’s being used in our sayHello implementation.

Having provided an implementation of the service, we can now register that with the Lagom framework. Lagom is built on top of Play Framework, and so uses Play’s Guice based dependency injection support to register components. To register a service, you’ll need to implement a Guice module. This can be done by creating a class called Module in the root package:

import com.google.inject.AbstractModule;
import com.lightbend.lagom.javadsl.server.ServiceGuiceSupport;

public class Module extends AbstractModule implements ServiceGuiceSupport {

    protected void configure() {
        bindServices(serviceBinding(HelloService.class, HelloServiceImpl.class));
    }
}

As you can see, the module extends Guice’s AbstractModule, as well as Lagom’s ServiceGuiceSupport. In this module, you can provide any Guice bindings you like. In this case, we’re just providing a binding for the Guice service. Lagom allows you to bind multiple services, however, bindServices() may only be invoked once, as this will bind a router for Play to use to route Lagom service calls, which if bound multiple times, will cause a Guice configuration error.

By convention, Play will automatically load a module called Module in the root package if it can find one, however if you’d prefer to call your module another name, or not put it in the root package, then you can manually add your module to Play’s list of enabled modules by adding the following to application.conf:

play.modules.enabled += com.example.MyModule

§Working with streams

When the request and response bodies are strict, working with them as quite straight forward. If however they are streamed, you’ll need to use Akka streams to work with them. Let’s take a look at how some of the streamed service calls in the service descriptors examples might be implemented.

The tick service call is going to return a Source that sends messages at the specified interval. Akka streams has a helpful constructor for such a stream:

public ServerServiceCall<String, Source<String, ?>> tick(int intervalMs) {
  return tickMessage -> {
    FiniteDuration interval = FiniteDuration.create(intervalMs, TimeUnit.MILLISECONDS);
    return completedFuture(Source.tick(interval, interval, tickMessage));
  };
}

The first two arguments are the delay before messages should be sent, and the interval at which they should be sent. The third argument is the message that should be sent on each tick. Calling this service call with an interval of 1000 and a request message of tick will result in a stream being returned that sent a tick message every second.

The sayHello service call can be implemented by mapping the incoming Source of the names to say hello to:

public ServerServiceCall<Source<String, ?>, Source<String, ?>> sayHello() {
  return names -> completedFuture(names.map(name -> "Hello " + name));
}

When you map a Source, you get back a new Source that applies the map transformation to each message that the incoming Source produces.

These examples of working with streams are obviously quite trivial. The sections on Publish-Subscribe and Persistent Read-Side show real examples of using streams in Lagom.

§Handling headers

Sometimes you may need to handle the request header, or add information to the response header. ServiceCall provides handleRequestHeader and handleResponseHeader methods to allow you to do this, however it is not recommended that you implement this directly, rather, you should use ServerServiceCall.

ServerServiceCall is an interface that extends ServiceCall, and provides an additional method, invokeWithHeaders. This is different from the regular invoke method because in addition to the Request parameter, it also accepts a RequestHeader parameter. And rather than returning a CompletionStage<Response>, it returns a CompletionStage<Pair<ResponseHeader, Response>>. Hence it allows you to handle the request header, and send a custom response header. ServerServiceCall implements the handleRequestHeader and handleResponseHeader methods, so that when Lagom calls the invoke method, it is delegated to the invokeWithHeaders method.

ServerServiceCall is a functional interface, leaving the original invoke method abstract, so when an interface requires you to pass or return a ServerServiceCall, if you implement it with a lambda, you aren’t forced to handle the headers. An additional functional interface is provided, HeaderServiceCall, this extends ServerServiceCall and makes invokeWithHeaders the abstract method. This can be used to handle headers with a lambda implemented service call, in two ways.

If you’re implementing the service call directly, you can simply change the return type to be HeaderServiceCall, like so:

public HeaderServiceCall<String, String> sayHello() {
  return (requestHeader, name) -> {

    String user = requestHeader.principal()
        .map(Principal::getName).orElse("No one");
    String response = user + " wants to say hello to " + name;

    ResponseHeader responseHeader = ResponseHeader.OK
        .withHeader("Server", "Hello service");

    return completedFuture(Pair.create(responseHeader, response));
  };
}

If you’re required to pass or return a ServerServiceCall, you can use the HeaderServiceCall.of method, like so:

public ServerServiceCall<String, String> sayHello() {
  return HeaderServiceCall.of((requestHeader, name) -> {

    String user = requestHeader.principal()
        .map(Principal::getName).orElse("No one");
    String response = user + " wants to say hello to " + name;

    ResponseHeader responseHeader = ResponseHeader.OK
        .withHeader("Server", "Hello service");

    return completedFuture(Pair.create(responseHeader, response));
  });
}

§Service call composition

You may have situations where you want to compose service calls with cross cutting concerns such as security or logging. In Lagom, this is done by composing service calls explicitly. The following shows a simple logging service call:

public <Request, Response> ServerServiceCall<Request, Response> logged(
    ServerServiceCall<Request, Response> serviceCall) {
  return HeaderServiceCall.compose(requestHeader -> {
    System.out.println("Received " + requestHeader.method() + " " + requestHeader.uri());
    return serviceCall;
  });
}

This uses the compose method from HeaderServiceCall, which takes a callback that takes the request header, and returns a service call.

If we were to implement the HelloService to be logged, we would use it like this:

public ServerServiceCall<String, String> sayHello() {
  return logged(
      name -> completedFuture("Hello " + name)
  );
}

Another common cross cutting concern is authentication. Imagine you have a user storage interface:

interface UserStorage {
  CompletionStage<Optional<User>> lookupUser(String username);
}

You could use it like this to implement an authenticated service call:

public <Request, Response> ServerServiceCall<Request, Response> authenticated(
    Function<User, ServerServiceCall<Request, Response>> serviceCall) {
  return HeaderServiceCall.composeAsync(requestHeader -> {

    // First lookup user
    CompletionStage<Optional<User>> userLookup = requestHeader.principal()
        .map(principal -> userStorage.lookupUser(principal.getName()))
        .orElse(completedFuture(Optional.empty()));

    // Then, if it exists, apply it to the service call
    return userLookup.thenApply(maybeUser -> {
      if (maybeUser.isPresent()) {
        return serviceCall.apply(maybeUser.get());
      } else {
        throw new Forbidden("User must be authenticated to access this service call");
      }
    });
  });
}

This time, since the lookup of the user is asynchronous, we’re using composeAsync, which allows us to asynchronously return the service call to handle the service. Also, instead of simply accepting a service call, we accept a function of a user to a service call. This means the service call can access the user:

public ServerServiceCall<String, String> sayHello() {
  return authenticated( user ->
      name -> completedFuture("Hello " + user)
  );
}

Notice here, in contrast to other frameworks where a user object may be passed using thread locals or in an untyped map by filters, the user object is explicitly passed. If your code requires access to the user object, it’s impossible to have a configuration error where you forgot to put the filter in place, the code simply will not compile.

Often you will want to compose multiple service calls together. This is where the power of function based composition really shines, in contrast to annotations. Since service calls are just regular methods, you can simply define a new method that combines them, like so:

public <Request, Response> ServerServiceCall<Request, Response> filter(
    Function<User, ServerServiceCall<Request, Response>> serviceCall) {
  return logged(authenticated(serviceCall));
}

Using this in the hello service:

public ServerServiceCall<String, String> sayHello() {
  return filter( user ->
      name -> completedFuture("Hello " + user)
  );
}

Found an error in this documentation? The source code for this page can be found here. Please feel free to edit and contribute a pull request.