com.github.yag.ipc.ResponsePacket.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ipc Show documentation

A simple async IPC library written in Kotlin.

There is a newer version: 0.18.3

/*
 * 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

import io.netty.buffer.ByteBuf
import java.net.ConnectException
import java.util.concurrent.TimeoutException

class ResponsePacketHeader(thrift: ResponseHeader = ResponseHeader()) : PacketHeader(thrift, {
    it.contentLength
}, {
    it.callId == -1L
})

fun Packet.status(): StatusCode = header.thrift.statusCode

fun Packet.isSuccessful(): Boolean = status().isSuccessful()

fun Packet.body(): ByteBuf {
    val status = status()
    return if (status.isSuccessful()) {
        body.data()
    } else {
        val bodyArray = body.data().use {
            it.readArray()
        }
        when (status) {
            StatusCode.NOT_FOUND -> throw UnsupportedOperationException()
            StatusCode.TIMEOUT -> throw TimeoutException()
            StatusCode.CONNECTION_ERROR -> throw ConnectException()
            StatusCode.INTERNAL_ERROR -> throw RemoteException(bodyArray)
            else -> throw IllegalStateException("Impossible status: $status")
        }
    }
}