com.palantir.lock.AtlasLockDescriptorRanges Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lock-api-objects Show documentation
Palantir open source project
There is a newer version: 0.1193.0
/*
 * (c) Copyright 2019 Palantir Technologies 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 com.palantir.lock;

import com.google.common.collect.Range;

public final class AtlasLockDescriptorRanges {
    private AtlasLockDescriptorRanges() {
        // no
    }

    public static Range fullTable(String qualifiedTableName) {
        byte[] tableNameBytes = bytesForTableName(qualifiedTableName);
        return Range.closedOpen(
                new LockDescriptor(tableNameBytes),
                new LockDescriptor(createExclusiveEndNameForPrefixScan(tableNameBytes)));
    }

    public static Range rowPrefix(String qualifiedTableName, byte[] prefix) {
        LockDescriptor start = AtlasRowLockDescriptor.of(qualifiedTableName, prefix);
        return Range.closedOpen(start, new LockDescriptor(createExclusiveEndNameForPrefixScan(start.getBytes())));
    }

    public static Range rowRange(String qualifiedTableName, byte[] startInc, byte[] endExc) {
        return Range.closedOpen(
                AtlasRowLockDescriptor.of(qualifiedTableName, startInc),
                AtlasRowLockDescriptor.of(qualifiedTableName, endExc));
    }

    public static Range exactRow(String qualifiedTableName, byte[] row) {
        LockDescriptor descriptor = AtlasRowLockDescriptor.of(qualifiedTableName, row);
        return Range.closed(descriptor, descriptor);
    }

    public static Range exactCell(String qualifiedTableName, byte[] row, byte[] col) {
        LockDescriptor descriptor = AtlasCellLockDescriptor.of(qualifiedTableName, row, col);
        return Range.closed(descriptor, descriptor);
    }

    private static byte[] bytesForTableName(String tableName) {
        return tableName.getBytes();
    }

    private static byte[] createExclusiveEndNameForPrefixScan(byte[] prefix) {
        for (int i = prefix.length - 1; i >= 0; i--) {
            if ((prefix[i] & 0xff) != 0xff) {
                byte[] ret = new byte[i + 1];
                System.arraycopy(prefix, 0, ret, 0, i + 1);
                ret[i]++;
                return ret;
            }
        }
        // this is unreachable when a part of prefix was created from a String
        throw new IllegalArgumentException("The prefix must be constructed using a valid table name.");
    }
}