com.github.kpavlov.jreactive8583.netty.codec.Iso8583Encoder.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of netty-iso8583 Show documentation

ISO8583 protocol client and server Netty connectors.

There is a newer version: 1.5.0

@file:JvmName("Iso8583Encoder")

package com.github.kpavlov.jreactive8583.netty.codec

import com.solab.iso8583.IsoMessage
import io.netty.buffer.ByteBuf
import io.netty.channel.ChannelHandler.Sharable
import io.netty.channel.ChannelHandlerContext
import io.netty.handler.codec.MessageToByteEncoder
import io.netty.util.CharsetUtil

@Sharable
public class Iso8583Encoder(
    private val lengthHeaderLength: Int,
    private val encodeLengthHeaderAsString: Boolean,
) : MessageToByteEncoder() {
    public override fun encode(
        ctx: ChannelHandlerContext,
        isoMessage: IsoMessage,
        out: ByteBuf,
    ) {
        when {
            lengthHeaderLength == 0 -> {
                val bytes = isoMessage.writeData()
                out.writeBytes(bytes)
            }

            encodeLengthHeaderAsString -> {
                val bytes = isoMessage.writeData()
                val lengthHeader = String.format("%0" + lengthHeaderLength + "d", bytes.size)
                out.writeBytes(lengthHeader.toByteArray(CharsetUtil.US_ASCII))
                out.writeBytes(bytes)
            }

            else -> {
                val byteBuffer = isoMessage.writeToBuffer(lengthHeaderLength)
                out.writeBytes(byteBuffer)
            }
        }
    }
}