io.streamnative.pulsar.handlers.kop.storage.AbortedTxn Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pulsar-protocol-handler-kafka Show documentation
Kafka on Pulsar implemented using Pulsar Protocol Handler
The newest version!
/**
 * Copyright (c) 2019 - 2024 StreamNative, Inc.. All Rights Reserved.
 */
/**
 * 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 io.streamnative.pulsar.handlers.kop.storage;

import java.io.DataInputStream;
import java.io.IOException;

/**
 * AbortedTxn is used cache the aborted index.
 */
public record AbortedTxn(long producerId, long firstOffset, long lastOffset, long lastStableOffset) {

    public static final int SIZE = 8 + 8 + 8 + 8;
    private static final int VersionOffset = 0;
    private static final int VersionSize = 2;
    private static final int ProducerIdOffset = VersionOffset + VersionSize;
    private static final int ProducerIdSize = 8;
    private static final int FirstOffsetOffset = ProducerIdOffset + ProducerIdSize;
    private static final int FirstOffsetSize = 8;
    private static final int LastOffsetOffset = FirstOffsetOffset + FirstOffsetSize;
    private static final int LastOffsetSize = 8;
    private static final int LastStableOffsetOffset = LastOffsetOffset + LastOffsetSize;
    private static final int LastStableOffsetSize = 8;
    private static final int TotalSize = LastStableOffsetOffset + LastStableOffsetSize;

    private static final Short CurrentVersion = 0;

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof AbortedTxn that) {
            return this.producerId == that.producerId && this.firstOffset == that.firstOffset
                && this.lastOffset == that.lastOffset && this.lastStableOffset == that.lastStableOffset;
        }
        return false;
    }

    void writeTo(ByteArrayStream stream) throws IOException {
        stream.writeLong(producerId);
        stream.writeLong(firstOffset);
        stream.writeLong(lastOffset);
        stream.writeLong(lastStableOffset);
    }


    static AbortedTxn readFrom(DataInputStream stream) throws IOException {
        final var producerId = stream.readLong();
        final var firstOffset = stream.readLong();
        final var lastOffset = stream.readLong();
        final var lastStableOffset = stream.readLong();
        return new AbortedTxn(producerId, firstOffset, lastOffset, lastStableOffset);
    }
}