StakeSquid/dshackle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

solution: allow to setup with ssl

Browse Source
This commit is contained in:
Igor Artamonov
2019-06-12 23:52:28 -04:00
parent 748b94fb6c
commit 591eb73f45
1 changed files with 27 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/main/kotlin/io/emeraldpay/dshackle/GrpcServer.kt
View File

@@ -2,17 +2,21 @@ package io.emeraldpay.dshackle
import io.grpc.Server import io.grpc.Server
import io.grpc.ServerBuilder import io.grpc.ServerBuilder
import org.apache.commons.lang3.StringUtils
import org.slf4j.LoggerFactory import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Autowired import org.springframework.beans.factory.annotation.Autowired
import org.springframework.core.env.Environment
import org.springframework.core.io.ResourceLoader import org.springframework.core.io.ResourceLoader
import org.springframework.stereotype.Service import org.springframework.stereotype.Service
import java.io.File
import javax.annotation.PostConstruct import javax.annotation.PostConstruct
import javax.annotation.PreDestroy import javax.annotation.PreDestroy
@Service @Service
open class GrpcServer( open class GrpcServer(
@Autowired val rpcs: List<io.grpc.BindableService>, @Autowired val rpcs: List<io.grpc.BindableService>,
@Autowired val resourceLoader: ResourceLoader @Autowired val resourceLoader: ResourceLoader,
@Autowired val env: Environment
) { ) {
private val log = LoggerFactory.getLogger(GrpcServer::class.java) private val log = LoggerFactory.getLogger(GrpcServer::class.java)
@@ -27,11 +31,28 @@ open class GrpcServer(
serverBuilder.addService(it) serverBuilder.addService(it)
} }
// serverBuilder var secure = false
// .useTransportSecurity( if (env.getProperty("ssl", "false") == "true") {
// resourceLoader.getResource("127.0.0.1.crt").inputStream, secure = true
// resourceLoader.getResource("127.0.0.1.p8.key").inputStream if (StringUtils.isEmpty(env.getProperty("ssl.cert"))) {
// ) log.warn("ssl.cert property is not set (path to server TLS certificate)")
secure = false
}
if (StringUtils.isEmpty(env.getProperty("ssl.key"))) {
log.warn("ssl.key property is not set (path to server TLS certificate key)")
secure = false
}
}
if (secure) {
log.info("Using TLS")
serverBuilder
.useTransportSecurity(
File(env.getProperty("ssl.cert")!!),
File(env.getProperty("ssl.key")!!)
)
} else {
log.warn("Using insecure transport")
}
val server = serverBuilder.build() val server = serverBuilder.build()
this.server = server this.server = server