Java Slice.length方法代码示例

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@VisibleForTesting
static int parseRgb(Slice color)
{
    if (color.length() != 4 || color.getByte(0) != '#') {
        return -1;
    }

    int red = Character.digit((char) color.getByte(1), 16);
    int green = Character.digit((char) color.getByte(2), 16);
    int blue = Character.digit((char) color.getByte(3), 16);

    if (red == -1 || green == -1 || blue == -1) {
        return -1;
    }

    // replicate the nibbles to turn a color of the form #rgb => #rrggbb (css semantics)
    red = (red << 4) | red;
    green = (green << 4) | green;
    blue = (blue << 4) | blue;

    return (int) rgb(red, green, blue);
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
private static int serializedSize(List<? extends Type> types, Page expectedPage)
{
    DynamicSliceOutput sliceOutput = new DynamicSliceOutput(1024);
    writePages(blockEncodingManager, sliceOutput, expectedPage);
    Slice slice = sliceOutput.slice();

    Iterator<Page> pageIterator = readPages(blockEncodingManager, slice.getInput());
    if (pageIterator.hasNext()) {
        assertPageEquals(types, pageIterator.next(), expectedPage);
    }
    else {
        assertEquals(expectedPage.getPositionCount(), 0);
    }
    assertFalse(pageIterator.hasNext());

    return slice.length();
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
public FixedWidthBlock(int fixedSize, int positionCount, Slice slice, Slice valueIsNull)
{
    super(fixedSize);

    if (positionCount < 0) {
        throw new IllegalArgumentException("positionCount is negative");
    }
    this.positionCount = positionCount;

    this.slice = Objects.requireNonNull(slice, "slice is null");

    if (valueIsNull.length() < positionCount) {
        throw new IllegalArgumentException("valueIsNull length is less than positionCount");
    }
    this.valueIsNull = valueIsNull;
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
private static int hashBytes(int initialValue, Slice bytes)
{
    int result = initialValue;
    for (int i = 0; i < bytes.length(); i++) {
        result = result * 31 + bytes.getByte(i);
    }
    return result;
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
private int decompressSnappy(Slice in)
        throws IOException
{
    byte[] inArray = (byte[]) in.getBase();
    int inOffset = (int) (in.getAddress() - ARRAY_BYTE_BASE_OFFSET);
    int inLength = in.length();

    int uncompressedLength = Snappy.getUncompressedLength(inArray, inOffset);
    checkArgument(uncompressedLength <= maxBufferSize, "Snappy requires buffer (%s) larger than max size (%s)", uncompressedLength, maxBufferSize);
    allocateOrGrowBuffer(uncompressedLength, false);

    return Snappy.uncompress(inArray, inOffset, inLength, buffer, 0);
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@ScalarFunction
@SqlType(StandardTypes.VARCHAR)
public static Slice render(@SqlType(StandardTypes.VARCHAR) Slice value, @SqlType(ColorType.NAME) long color)
{
    StringBuilder builder = new StringBuilder(value.length());

    // color
    builder.append(ansiColorEscape(color))
            .append(value.toStringUtf8())
            .append(ANSI_RESET);

    return utf8Slice(builder.toString());
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@Description("returns index of first occurrence of a substring (or 0 if not found)")
@ScalarFunction("strpos")
@SqlType(StandardTypes.BIGINT)
public static long stringPosition(@SqlType(StandardTypes.VARCHAR) Slice string, @SqlType(StandardTypes.VARCHAR) Slice substring)
{
    if (substring.length() == 0) {
        return 1;
    }

    int index = string.indexOf(substring);
    if (index < 0) {
        return 0;
    }
    return countCodePoints(string, 0, index) + 1;
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@ScalarFunction
@SqlType("array<varchar>")
public static Block split(@SqlType(StandardTypes.VARCHAR) Slice string, @SqlType(StandardTypes.VARCHAR) Slice delimiter, @SqlType(StandardTypes.BIGINT) long limit)
{
    checkCondition(limit > 0, INVALID_FUNCTION_ARGUMENT, "Limit must be positive");
    checkCondition(limit <= Integer.MAX_VALUE, INVALID_FUNCTION_ARGUMENT, "Limit is too large");
    checkCondition(delimiter.length() > 0, INVALID_FUNCTION_ARGUMENT, "The delimiter may not be the empty string");
    BlockBuilder parts = VARCHAR.createBlockBuilder(new BlockBuilderStatus(), 1, string.length());
    // If limit is one, the last and only element is the complete string
    if (limit == 1) {
        VARCHAR.writeSlice(parts, string);
        return parts.build();
    }

    int index = 0;
    while (index < string.length()) {
        int splitIndex = string.indexOf(delimiter, index);
        // Found split?
        if (splitIndex < 0) {
            break;
        }
        // Add the part from current index to found split
        VARCHAR.writeSlice(parts, string, index, splitIndex - index);
        // Continue searching after delimiter
        index = splitIndex + delimiter.length();
        // Reached limit-1 parts so we can stop
        if (parts.getPositionCount() == limit - 1) {
            break;
        }
    }
    // Rest of string
    VARCHAR.writeSlice(parts, string, index, string.length() - index);

    return parts.build();
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@ScalarOperator(OperatorType.CAST)
@SqlType(RegexpType.NAME)
public static Regex castToRegexp(@SqlType(StandardTypes.VARCHAR) Slice pattern)
{
    Regex regex;
    try {
        // When normal UTF8 encoding instead of non-strict UTF8) is used, joni can infinite loop when invalid UTF8 slice is supplied to it.
        regex = new Regex(pattern.getBytes(), 0, pattern.length(), Option.DEFAULT, NonStrictUTF8Encoding.INSTANCE, Syntax.Java);
    }
    catch (Exception e) {
        throw new PrestoException(INVALID_FUNCTION_ARGUMENT, e);
    }
    return regex;
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@Description("replaces substrings matching a regular expression by given string")
@ScalarFunction
@SqlType(StandardTypes.VARCHAR)
public static Slice regexpReplace(@SqlType(StandardTypes.VARCHAR) Slice source, @SqlType(RegexpType.NAME) Regex pattern, @SqlType(StandardTypes.VARCHAR) Slice replacement)
{
    Matcher matcher = pattern.matcher(source.getBytes());
    SliceOutput sliceOutput = new DynamicSliceOutput(source.length() + replacement.length() * 5);

    int lastEnd = 0;
    int nextStart = 0; // nextStart is the same as lastEnd, unless the last match was zero-width. In such case, nextStart is lastEnd + 1.
    while (true) {
        int offset = matcher.search(nextStart, source.length(), Option.DEFAULT);
        if (offset == -1) {
            break;
        }
        if (matcher.getEnd() == matcher.getBegin()) {
            nextStart = matcher.getEnd() + 1;
        }
        else {
            nextStart = matcher.getEnd();
        }
        Slice sliceBetweenReplacements = source.slice(lastEnd, matcher.getBegin() - lastEnd);
        lastEnd = matcher.getEnd();
        sliceOutput.appendBytes(sliceBetweenReplacements);
        appendReplacement(sliceOutput, source, pattern, matcher.getEagerRegion(), replacement);
    }
    sliceOutput.appendBytes(source.slice(lastEnd, source.length() - lastEnd));

    return sliceOutput.slice();
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@Description("length of the given binary")
@ScalarFunction
@SqlType(StandardTypes.BIGINT)
public static long length(@SqlType(StandardTypes.VARBINARY) Slice slice)
{
    return slice.length();
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@Description("decode hex encoded binary data")
@ScalarFunction("from_hex")
@SqlType(StandardTypes.VARBINARY)
public static Slice fromHexVarchar(@SqlType(StandardTypes.VARCHAR) Slice slice)
{
    if (slice.length() % 2 != 0) {
        throw new PrestoException(INVALID_FUNCTION_ARGUMENT, "invalid input length " + slice.length());
    }

    byte[] result = new byte[slice.length() / 2];
    for (int i = 0; i < slice.length(); i += 2) {
        result[i / 2] = (byte) ((hexDigitCharToInt(slice.getByte(i)) << 4) | hexDigitCharToInt(slice.getByte(i + 1)));
    }
    return Slices.wrappedBuffer(result);
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
/**
 * Sets the element of this big array at specified index.
 *
 * @param index a position in this big array.
 */
public void set(long index, Slice value)
{
    Slice currentValue = array.get(index);
    if (currentValue != null) {
        sizeOfSlices -= currentValue.length();
    }
    if (value != null) {
        sizeOfSlices += value.length();
    }
    array.set(index, value);
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@ScalarOperator(CAST)
@SqlType(StandardTypes.BOOLEAN)
public static boolean castToBoolean(@SqlType(StandardTypes.VARCHAR) Slice value)
{
    if (value.length() == 1) {
        byte character = toUpperCase(value.getByte(0));
        if (character == 'T' || character == '1') {
            return true;
        }
        if (character == 'F' || character == '0') {
            return false;
        }
    }
    if ((value.length() == 4) &&
            (toUpperCase(value.getByte(0)) == 'T') &&
            (toUpperCase(value.getByte(1)) == 'R') &&
            (toUpperCase(value.getByte(2)) == 'U') &&
            (toUpperCase(value.getByte(3)) == 'E')) {
        return true;
    }
    if ((value.length() == 5) &&
            (toUpperCase(value.getByte(0)) == 'F') &&
            (toUpperCase(value.getByte(1)) == 'A') &&
            (toUpperCase(value.getByte(2)) == 'L') &&
            (toUpperCase(value.getByte(3)) == 'S') &&
            (toUpperCase(value.getByte(4)) == 'E')) {
        return false;
    }
    throw new PrestoException(INVALID_CAST_ARGUMENT, format("Cannot cast '%s' to BOOLEAN", value.toStringUtf8()));
}
 

import io.airlift.slice.Slice; //导入方法依赖的package包/类
@Test
public void testSizeInBytes()
        throws Exception
{
    Slice[] expectedValues = createExpectedValues(10);
    DictionaryBlock dictionaryBlock = createDictionaryBlock(expectedValues, 100);

    int sizeInBytes = 0;
    for (Slice expectedValue : expectedValues) {
        sizeInBytes += expectedValue.length();
    }
    assertEquals(dictionaryBlock.getSizeInBytes(), sizeInBytes + (100 * SIZE_OF_INT));
}