akka.grpc.SSLContextUtils.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of akka-grpc-runtime_2.12 Show documentation

Akka gRPC - Support for building streaming gRPC servers and clients on top of Akka Streams.

There is a newer version: 2.3.4

Show newest version

/*
 * Copyright (C) 2018-2019 Lightbend Inc. 
 */

package akka.grpc

import java.io.{ File, FileInputStream, IOException, InputStream }

import io.grpc.netty.shaded.io.grpc.netty.GrpcSslContexts
import io.grpc.netty.shaded.io.netty.handler.ssl.{ JdkSslContext, SslProvider }
import javax.net.ssl.SSLContext

object SSLContextUtils {

  def fromStream(certStream: InputStream): SSLContext = {
    val sslBuilder = try { GrpcSslContexts.forClient.trustManager(certStream) } finally certStream.close()
    GrpcSslContexts.configure(sslBuilder, SslProvider.JDK).build.asInstanceOf[JdkSslContext].context
  }

  def sslContextFromResource(certificateResourcePath: String): SSLContext = {
    // Use Netty's SslContextBuilder internally to help us construct a SSLContext
    val certStream: InputStream = getClass.getResourceAsStream(certificateResourcePath)
    if (certStream == null) throw new IOException(s"Couldn't find '$certificateResourcePath' on the classpath")
    fromStream(certStream)
  }

  def sslContextFromFile(certPath: File): SSLContext = {
    fromStream(new FileInputStream(certPath))
  }

}