Implementing services

§Implementing services

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

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

import com.lightbend.lagom.scaladsl.api.ServiceCall
import scala.concurrent.Future

class HelloServiceImpl extends HelloService {
  override def sayHello = ServiceCall { name =>
    Future.successful(s"Hello $name!")
  }
}

As you can see, the sayHello method is implemented using the apply factory method on ServiceCall. This takes a function of Request => Future[Response] and returns a ServiceCall whose invoke method simply delegates to that function. An important thing to realise here is that the invocation of sayHello itself does not execute the call, it only returns the service 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.

Let’s have a look at our ServiceCall interface again:

trait ServiceCall[Request, Response] {
  def invoke(request: Request): Future[Response]
}

It will take the request, and return the response as a Future. If you’ve never seen Future before, it is a value that may not be available until later. When an API returns a future, 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 map and flatMap methods. Future with its map and flatMap methods are fundamental building blocks for doing reactive programming in Scala, they allow your code to be asynchronous, not waiting for things to happen, but attaching callbacks that react to computations being completed.

Of course, a simple hello world computation is not asynchronous, all it needs to do is build a String, and that returns immediately. In this case, we need to wrap the result of that in a Future. This can be done by calling Future.successful(), which returns a future that has an immediately available value.

§Working with streams

When the request and response bodies are strict, working with them is straightforward. If they are streamed, however, 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:

import scala.concurrent.duration._
import akka.stream.scaladsl.Source

override def tick(intervalMs: Int) = ServiceCall { tickMessage =>
  Future.successful(Source.tick(
    intervalMs.milliseconds, intervalMs.milliseconds, tickMessage
  ).mapMaterializedValue(_ => NotUsed))
}

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.

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

override def sayHello = ServiceCall { names =>
  Future.successful(names.map(name => s"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 these 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 Future[Response], it returns a Future[(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.

The ServerServiceCall companion object provides a factory method for creating ServerServiceCall’s that work with both request and response headers. It may seem counter intuitive to be able to create a ServerServiceCall that doesn’t work with headers, but the reason for doing this is to assist in service call composition, where a composing service call might want to compose both types of service call.

Here’s an example of working with the headers:

import com.lightbend.lagom.scaladsl.server.ServerServiceCall
import com.lightbend.lagom.scaladsl.api.transport.ResponseHeader

override def sayHello = ServerServiceCall { (requestHeader, name) =>
  val user = requestHeader.principal
    .map(_.getName).getOrElse("No one")
  val response = s"$user wants to say hello to $name"

  val responseHeader = ResponseHeader.Ok
    .withHeader("Server", "Hello service")

  Future.successful((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:

def logged[Request, Response](serviceCall: ServerServiceCall[Request, Response]) =
  ServerServiceCall.compose { requestHeader =>
    println(s"Received ${requestHeader.method} ${requestHeader.uri}")
    serviceCall
  }

This uses the compose method from ServerServiceCall, 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:

override def sayHello = logged(ServerServiceCall { name =>
  Future.successful(s"Hello $name!")
})

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

trait UserStorage {
  def lookupUser(username: String): Future[Option[User]]
}

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

import com.lightbend.lagom.scaladsl.api.transport.Forbidden

def authenticated[Request, Response](
  serviceCall: User => ServerServiceCall[Request, Response]
) = ServerServiceCall.composeAsync { requestHeader =>

  // First lookup user
  val userLookup = requestHeader.principal
    .map(principal => userStorage.lookupUser(principal.getName))
    .getOrElse(Future.successful(None))

  // Then, if it exists, apply it to the service call
  userLookup.map {
    case Some(user) => serviceCall(user)
    case None => throw 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:

override def sayHello = authenticated { user =>
  ServerServiceCall { name =>
    Future.successful(s"$user is saying hello to $name")
  }
}

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:

def filter[Request, Response](
  serviceCall: User => ServerServiceCall[Request, Response]
) = logged(authenticated(serviceCall))

Using this in the hello service:

override def sayHello = filter { user =>
  ServerServiceCall { name =>
    Future.successful(s"$user is saying hello to $name")
  }
}

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.