Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2018-2020 [email protected]
*
* Licensed under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package com.github.yag.ipc.client
import com.codahale.metrics.MetricRegistry
import com.github.yag.ipc.Daemon
import com.github.yag.ipc.Packet
import com.github.yag.ipc.PlainBody
import com.github.yag.ipc.Prompt
import com.github.yag.ipc.ResponseHeader
import com.github.yag.retry.DefaultErrorHandler
import com.github.yag.retry.Retry
import io.netty.buffer.ByteBuf
import io.netty.buffer.Unpooled
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import java.nio.ByteBuffer
import java.util.concurrent.CompletableFuture
import java.util.concurrent.Future
import java.util.concurrent.atomic.AtomicBoolean
/**
* The IPC Client class send messages to IPC server can execute callback when response was received.
*
* @param T The call type class, typically string, numbers and enums.
* @property config client config
* @property metric metric registry
* @property id client id, will be used in log and thread name.
*/
class IPCClient(
private val config: IPCClientConfig,
private val promptHandler: (Prompt) -> ByteArray,
private val metric: MetricRegistry,
private val id: String
) : AutoCloseable {
private val retry = Retry(config.connectRetry, config.connectBackOff, DefaultErrorHandler(), config.backOffRandomRange)
private var client: RawIPCClient
val sessionId: String
inner class Monitor(private val shouldStop: AtomicBoolean) : Runnable {
override fun run() {
while (!shouldStop.get()) {
try {
client.channel.closeFuture().await()
LOG.warn("Connection broken.")
client.close()
val uncompleted = client.getUncompletedRequests()
Thread.sleep(config.connectBackOff.baseIntervalMs)
client = retry.call {
RawIPCClient(config, promptHandler, sessionId, metric, id)
}.also {
for (call in uncompleted) {
it.send(call.request.type, call.request.packet, call.callback.func)
}
}
} catch (e: InterruptedException) {
}
}
}
}
private var monitor: Daemon
init {
client = retry.call {
RawIPCClient(config, promptHandler, null, metric, id)
}
sessionId = client.connection.sessionId
monitor = Daemon("connection-monitor") {
Monitor(it)
}.also { it.start() }
}
/**
* Send request packet to server.
* @param type call type
* @param body request body
* @param callback code block to handle response packet
*/
fun send(type: RequestType, body: Body, callback: (Packet) -> Any?) {
client.send(type, body, callback)
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @param callback code block to handle response packet
*/
fun send(type: RequestType, data: ByteBuf, callback: (Packet) -> Any?) {
send(type, PlainBody(data), callback)
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @param callback code block to handle response packet
*/
fun send(type: RequestType, data: ByteBuffer, callback: (Packet) -> Any?) {
send(type, Unpooled.wrappedBuffer(data), callback)
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @param callback code block to handle response packet
*/
fun send(type: RequestType, data: ByteArray, callback: (Packet) -> Any?) {
send(type, Unpooled.wrappedBuffer(data), callback)
}
/**
* Send request packet to server.
* @param type call type
* @param body request body
* @return Future object of response packet
*/
fun send(type: RequestType, body: Body): Future> {
val future = CompletableFuture>()
send(type, body) {
future.complete(it.also {
it.body.data().retain()
})
}
return future
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return Future object of response packet
*/
fun send(type: RequestType, data: ByteBuf): Future> {
return send(type, PlainBody(data))
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return Future object of response packet
*/
fun send(type: RequestType, data: ByteBuffer): Future> {
return send(type, Unpooled.wrappedBuffer(data))
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return Future object of response packet
*/
fun send(type: RequestType, data: ByteArray): Future> {
return send(type, Unpooled.wrappedBuffer(data))
}
/**
* Send request packet to server.
* @param type call type
* @param data request body
* @return response packet
*/
fun sendSync(type: RequestType, body: Body): Packet {
return send(type, body).get()
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return response packet
*/
fun sendSync(type: RequestType, data: ByteBuf): Packet {
return send(type, PlainBody(data)).get()
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return response packet
*/
fun sendSync(type: RequestType, data: ByteBuffer): Packet {
return send(type, Unpooled.wrappedBuffer(data)).get()
}
/**
* Send request packet to server.
* @param type call type
* @param data request data
* @return response packet
*/
fun sendSync(type: RequestType, data: ByteArray): Packet {
return send(type, Unpooled.wrappedBuffer(data)).get()
}
/**
* Close IPC client and release all related resources..
*/
override fun close() {
monitor.close()
client.close()
}
/**
* Check connection is well.
* @return true is connected.
*/
fun isConnected(): Boolean {
return client.isConnected()
}
companion object {
private val LOG: Logger = LoggerFactory.getLogger(IPCClient::class.java)
}
}
