From 55ba61f35cf5c91b02138ca4d515c85430fe0326 Mon Sep 17 00:00:00 2001 From: Zhangmei Li Date: Sat, 22 Dec 2018 00:29:23 +0800 Subject: [PATCH] Support for sortable base64 encoding implement #11 Change-Id: I39b8f26d025e095d9d654f655657df6d6dc2543f --- .../baidu/hugegraph/util/LongEncoding.java | 76 ++- .../com/baidu/hugegraph/util/NumericUtil.java | 66 ++- .../com/baidu/hugegraph/util/VersionUtil.java | 11 + .../baidu/hugegraph/unit/UnitTestSuite.java | 4 +- .../hugegraph/unit/event/EventHubTest.java | 12 +- .../hugegraph/unit/util/LongEncodingTest.java | 473 ++++++++++++++++++ .../hugegraph/unit/util/VersionUtilTest.java | 16 + 7 files changed, 636 insertions(+), 22 deletions(-) create mode 100644 src/test/java/com/baidu/hugegraph/unit/util/LongEncodingTest.java diff --git a/src/main/java/com/baidu/hugegraph/util/LongEncoding.java b/src/main/java/com/baidu/hugegraph/util/LongEncoding.java index e0d5c07d8..1fccbddb3 100644 --- a/src/main/java/com/baidu/hugegraph/util/LongEncoding.java +++ b/src/main/java/com/baidu/hugegraph/util/LongEncoding.java @@ -24,20 +24,77 @@ public final class LongEncoding { private static final String BASE_SYMBOLS = - "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-~"; + "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ_abcdefghijklmnopqrstuvwxyz~"; - public static long decode(String s) { - return decode(s, BASE_SYMBOLS); + private static final String LENGTH_SYMBOLS = "0123456789ABCDEF"; + private static final char NEG = LENGTH_SYMBOLS.charAt(0); + + private static final long FULL_LONG = 0x8000000000000000L; + + public static String encodeNumber(Object number) { + Number num = NumericUtil.convertToNumber(number); + long value = NumericUtil.numberToSortableLong(num); + return encodeSortable(value); + } + + public static Number decodeNumber(String str, Class clazz) { + long value = decodeSortable(str); + Number number = NumericUtil.sortableLongToNumber(value, clazz); + return number; + } + + public static String encodeSortable(long num) { + boolean negative = false; + if (num < 0L) { + negative = true; + num += FULL_LONG; + } + + String encoded = encode(num, BASE_SYMBOLS); + int length = encoded.length(); + E.checkArgument(length <= LENGTH_SYMBOLS.length(), + "Length symbols can't represent encoded number '%s'", + encoded); + StringBuilder sb = new StringBuilder(length + 2); + if (negative) { + sb.append(NEG); + } + char len = LENGTH_SYMBOLS.charAt(length); + sb.append(len); + sb.append(encoded); + + return sb.toString(); + } + + public static long decodeSortable(String str) { + E.checkArgument(str.length() >= 2, + "Length of sortable encoded string must be >=2"); + boolean negative = str.charAt(0) == NEG; + int prefix = 1; + if (negative) { + prefix = 2; + } + String encoded = str.substring(prefix); + long value = decode(encoded); + if (negative) { + value -= FULL_LONG; + } + return value; } public static String encode(long num) { return encode(num, BASE_SYMBOLS); } - public static long decode(String s, String symbols) { + public static long decode(String str) { + return decode(str, BASE_SYMBOLS); + } + + public static long decode(String str, String symbols) { final int B = symbols.length(); + E.checkArgument(B > 0, "The symbols parameter can't be empty"); long num = 0; - for (char ch : s.toCharArray()) { + for (char ch : str.toCharArray()) { num *= B; int pos = symbols.indexOf(ch); if (pos < 0) @@ -49,13 +106,16 @@ public static long decode(String s, String symbols) { } public static String encode(long num, String symbols) { - E.checkArgument(num >= 0, "Expected non-negative number: %s", num); final int B = symbols.length(); + E.checkArgument(num >= 0, "Expected non-negative number: %s", num); + E.checkArgument(B > 0, "The symbols parameter can't be empty"); + StringBuilder sb = new StringBuilder(); - while (num != 0) { + do { sb.append(symbols.charAt((int) (num % B))); num /= B; - } + } while (num != 0); + return sb.reverse().toString(); } } diff --git a/src/main/java/com/baidu/hugegraph/util/NumericUtil.java b/src/main/java/com/baidu/hugegraph/util/NumericUtil.java index 16127c8d2..ee608d3e1 100644 --- a/src/main/java/com/baidu/hugegraph/util/NumericUtil.java +++ b/src/main/java/com/baidu/hugegraph/util/NumericUtil.java @@ -103,6 +103,48 @@ public static int sortableFloatBits(int bits) { return bits ^ (bits >> 31) & 0x7fffffff; } + public static long numberToSortableLong(Number number) { + if (number instanceof Double) { + return doubleToSortableLong(number.doubleValue()); + } else if (number instanceof Float) { + return floatToSortableInt(number.floatValue()); + } else if (number instanceof Long || number instanceof Integer || + number instanceof Short || number instanceof Byte) { + return number.longValue(); + } else if (number instanceof BigDecimal) { + BigDecimal bd = (BigDecimal) number; + boolean intNumber = bd.stripTrailingZeros().scale() <= 0; + return intNumber ? bd.longValueExact() : + doubleToSortableLong(bd.doubleValue()); + } + // TODO: support other number types + throw new IllegalArgumentException(String.format( + "Unsupported number type: %s(%s)", + number.getClass().getSimpleName(), number)); + } + + public static Number sortableLongToNumber(long value, Class clazz) { + assert NumericUtil.isNumber(clazz); + + if (clazz == Double.class) { + return sortableLongToDouble(value); + } else if (clazz == Float.class) { + return sortableIntToFloat((int) value); + } else if (clazz == Long.class) { + return value; + } else if (clazz == Integer.class) { + return (int) value; + } else if (clazz == Short.class) { + return (short) value; + } else if (clazz == Byte.class) { + return (byte) value; + } + + // TODO: support other number types + throw new IllegalArgumentException(String.format( + "Unsupported number type: %s", clazz.getSimpleName())); + } + public static byte[] numberToSortableBytes(Number number) { if (number instanceof Long) { return longToBytes(number.longValue()); @@ -117,7 +159,9 @@ public static byte[] numberToSortableBytes(Number number) { } // TODO: support other number types - return null; + throw new IllegalArgumentException(String.format( + "Unsupported number type: %s(%s)", + number.getClass().getSimpleName(), number)); } public static Number sortableBytesToNumber(byte[] bytes, Class clazz) { @@ -138,7 +182,8 @@ public static Number sortableBytesToNumber(byte[] bytes, Class clazz) { } // TODO: support other number types - return null; + throw new IllegalArgumentException(String.format( + "Unsupported number type: %s", clazz.getSimpleName())); } public static byte[] longToBytes(long value) { @@ -174,16 +219,23 @@ public static boolean isNumber(Class clazz) { return Number.class.isAssignableFrom(clazz); } - public static Object convertToNumber(Object value) { - if (!isNumber(value) && value != null) { + public static Number convertToNumber(Object value) { + if (value == null) { + return null; + } + + Number number; + if (isNumber(value)) { + number = (Number) value; + } else { if (value instanceof Date) { - value = ((Date) value).getTime(); + number = ((Date) value).getTime(); } else { // TODO: add some more types to convert - value = new BigDecimal(value.toString()); + number = new BigDecimal(value.toString()); } } - return value; + return number; } /** diff --git a/src/main/java/com/baidu/hugegraph/util/VersionUtil.java b/src/main/java/com/baidu/hugegraph/util/VersionUtil.java index a6fd43ffa..b1cbe3643 100644 --- a/src/main/java/com/baidu/hugegraph/util/VersionUtil.java +++ b/src/main/java/com/baidu/hugegraph/util/VersionUtil.java @@ -39,6 +39,17 @@ public static boolean match(Version version, String begin, String end) { version.compareTo(new Version(end)) < 0; } + /** + * Compare if a version is greater than the other one (inclusive) + * @param version The version to be compared + * @param begin The lower bound of the range + * @return true if it's greater than the other, otherwise false + */ + public static boolean gte(String version, String other) { + E.checkArgumentNotNull(version, "The version to match is null"); + return new Version(version).compareTo(new Version(other)) >= 0; + } + /** * Check whether a component version is matched expected range, * throw an exception if it's not matched. diff --git a/src/test/java/com/baidu/hugegraph/unit/UnitTestSuite.java b/src/test/java/com/baidu/hugegraph/unit/UnitTestSuite.java index c178649b0..9f637d7ef 100644 --- a/src/test/java/com/baidu/hugegraph/unit/UnitTestSuite.java +++ b/src/test/java/com/baidu/hugegraph/unit/UnitTestSuite.java @@ -31,6 +31,7 @@ import com.baidu.hugegraph.unit.util.BytesTest; import com.baidu.hugegraph.unit.util.CollectionUtilTest; import com.baidu.hugegraph.unit.util.HashUtilTest; +import com.baidu.hugegraph.unit.util.LongEncodingTest; import com.baidu.hugegraph.unit.util.VersionUtilTest; @RunWith(Suite.class) @@ -46,7 +47,8 @@ BytesTest.class, CollectionUtilTest.class, HashUtilTest.class, - VersionUtilTest.class + VersionUtilTest.class, + LongEncodingTest.class }) public class UnitTestSuite { } diff --git a/src/test/java/com/baidu/hugegraph/unit/event/EventHubTest.java b/src/test/java/com/baidu/hugegraph/unit/event/EventHubTest.java index b9f9188e8..ee572833a 100644 --- a/src/test/java/com/baidu/hugegraph/unit/event/EventHubTest.java +++ b/src/test/java/com/baidu/hugegraph/unit/event/EventHubTest.java @@ -461,7 +461,7 @@ public Object event(Event event) { event.checkArgs(Integer.class); int i = (int) event.args()[0]; - if (i % 100000 == 0) { + if (i % 10000 == 0) { System.out.println("On event '" + notify + "': " + i); } return null; @@ -469,8 +469,8 @@ public Object event(Event event) { }; Thread listenerUpdateThread = new Thread(() -> { - // This will cost about 10s - for (int i = 0; i < 100; i++) { + // This will cost about 1s + for (int i = 0; i < 10; i++) { this.eventHub.listen(notify, listener1); if (!this.eventHub.listeners(notify).contains(listener2)) { this.eventHub.listen(notify, listener2); @@ -478,7 +478,7 @@ public Object event(Event event) { this.wait100ms(); - if (i % 10 == 0) { + if (i % 2 == 0) { this.eventHub.unlisten(notify); } else { this.eventHub.unlisten(notify, listener1); @@ -488,8 +488,8 @@ public Object event(Event event) { listenerUpdateThread.start(); runWithThreads(THREADS_NUM, () -> { - // This will cost about 10s ~ 20s - for (int i = 0; i < 10000 * 100; i++) { + // This will cost about 1s ~ 2s + for (int i = 0; i < 10000 * 10; i++) { this.eventHub.notify(notify, i); Thread.yield(); } diff --git a/src/test/java/com/baidu/hugegraph/unit/util/LongEncodingTest.java b/src/test/java/com/baidu/hugegraph/unit/util/LongEncodingTest.java new file mode 100644 index 000000000..58d041933 --- /dev/null +++ b/src/test/java/com/baidu/hugegraph/unit/util/LongEncodingTest.java @@ -0,0 +1,473 @@ +/* + * Copyright 2017 HugeGraph Authors + * + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with this + * work for additional information regarding copyright ownership. The ASF + * licenses this file to You 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.baidu.hugegraph.unit.util; + +import java.math.BigDecimal; +import java.util.Date; +import java.util.Random; + +import org.junit.Test; + +import com.baidu.hugegraph.testutil.Assert; +import com.baidu.hugegraph.unit.BaseUnitTest; +import com.baidu.hugegraph.util.Bytes; +import com.baidu.hugegraph.util.LongEncoding; +import com.baidu.hugegraph.util.NumericUtil; + +public class LongEncodingTest extends BaseUnitTest { + + @Test + public void testEncode() { + String val0 = LongEncoding.encode(0); + Assert.assertEquals("0", val0); + + String val1 = LongEncoding.encode(1); + Assert.assertEquals("1", val1); + + String val9 = LongEncoding.encode(9); + Assert.assertEquals("9", val9); + + String val10 = LongEncoding.encode(10); + Assert.assertEquals("A", val10); + + String val35 = LongEncoding.encode(35); + Assert.assertEquals("Z", val35); + + String val36 = LongEncoding.encode(36); + Assert.assertEquals("_", val36); + + String val37 = LongEncoding.encode(37); + Assert.assertEquals("a", val37); + + String val62 = LongEncoding.encode(62); + Assert.assertEquals("z", val62); + + String val63 = LongEncoding.encode(63); + Assert.assertEquals("~", val63); + } + + @Test + public void testEncodeWithMultiString() { + String val64 = LongEncoding.encode(64); + Assert.assertEquals("10", val64); + + String val65 = LongEncoding.encode(65); + Assert.assertEquals("11", val65); + + String val99 = LongEncoding.encode(99); + Assert.assertEquals("1Z", val99); + + String val100 = LongEncoding.encode(100); + Assert.assertEquals("1_", val100); + + String val126 = LongEncoding.encode(126); + Assert.assertEquals("1z", val126); + + String val127 = LongEncoding.encode(127); + Assert.assertEquals("1~", val127); + + String val128 = LongEncoding.encode(128); + Assert.assertEquals("20", val128); + + String val200 = LongEncoding.encode(200); + Assert.assertEquals("38", val200); + + String val1000 = LongEncoding.encode(1000); + Assert.assertEquals("Fd", val1000); + + String val1234 = LongEncoding.encode(1234); + Assert.assertEquals("JI", val1234); + + String val10000 = LongEncoding.encode(10000); + Assert.assertEquals("2SG", val10000); + + String val12345 = LongEncoding.encode(12345); + Assert.assertEquals("30u", val12345); + + String val22345 = LongEncoding.encode(22345); + Assert.assertEquals("5T9", val22345); + + String val92345 = LongEncoding.encode(92345); + Assert.assertEquals("MYu", val92345); + + String val12345678 = LongEncoding.encode(12345678); + Assert.assertEquals("k65E", val12345678); + + String val112345678 = LongEncoding.encode(112345678); + Assert.assertEquals("6h_9E", val112345678); + + String valIntMax = LongEncoding.encode(Integer.MAX_VALUE); + Assert.assertEquals("1~~~~~", valIntMax); + + String valLongMax = LongEncoding.encode(Long.MAX_VALUE); + Assert.assertEquals("7~~~~~~~~~~", valLongMax); + } + + @Test + public void testEncodeNegative() { + Assert.assertThrows(IllegalArgumentException.class, () -> { + LongEncoding.encode(-1); + }); + + Assert.assertThrows(IllegalArgumentException.class, () -> { + LongEncoding.encode(Long.MIN_VALUE); + }); + } + + @Test + public void testDecode() { + long valEmpty = LongEncoding.decode(""); + Assert.assertEquals(0, valEmpty); + + long val0 = LongEncoding.decode("0"); + Assert.assertEquals(0, val0); + + long val1 = LongEncoding.decode("1"); + Assert.assertEquals(1, val1); + + long val9 = LongEncoding.decode("9"); + Assert.assertEquals(9, val9); + + long val10 = LongEncoding.decode("A"); + Assert.assertEquals(10, val10); + + long val35 = LongEncoding.decode("Z"); + Assert.assertEquals(35, val35); + + long val36 = LongEncoding.decode("_"); + Assert.assertEquals(36, val36); + + long val37 = LongEncoding.decode("a"); + Assert.assertEquals(37, val37); + + long val62 = LongEncoding.decode("z"); + Assert.assertEquals(62, val62); + + long val63 = LongEncoding.decode("~"); + Assert.assertEquals(63, val63); + } + + @Test + public void testDecodeWithMultiString() { + long val64 = LongEncoding.decode("10"); + Assert.assertEquals(64, val64); + + long val65 = LongEncoding.decode("11"); + Assert.assertEquals(65, val65); + + long val99 = LongEncoding.decode("1Z"); + Assert.assertEquals(99, val99); + + long val100 = LongEncoding.decode("1_"); + Assert.assertEquals(100, val100); + + long val126 = LongEncoding.decode("1z"); + Assert.assertEquals(126, val126); + + long val127 = LongEncoding.decode("1~"); + Assert.assertEquals(127, val127); + + long val128 = LongEncoding.decode("20"); + Assert.assertEquals(128, val128); + + long val200 = LongEncoding.decode("38"); + Assert.assertEquals(200, val200); + + long val1000 = LongEncoding.decode("Fd"); + Assert.assertEquals(1000, val1000); + + long val1234 = LongEncoding.decode("JI"); + Assert.assertEquals(1234, val1234); + + long val10000 = LongEncoding.decode("2SG"); + Assert.assertEquals(10000, val10000); + + long val12345 = LongEncoding.decode("30u"); + Assert.assertEquals(12345, val12345); + + long val22345 = LongEncoding.decode("5T9"); + Assert.assertEquals(22345, val22345); + + long val92345 = LongEncoding.decode("MYu"); + Assert.assertEquals(92345, val92345); + + long val12345678 = LongEncoding.decode("k65E"); + Assert.assertEquals(12345678, val12345678); + + long val112345678 = LongEncoding.decode("6h_9E"); + Assert.assertEquals(112345678, val112345678); + + long valIntMax = LongEncoding.decode("1~~~~~"); + Assert.assertEquals(Integer.MAX_VALUE, valIntMax); + + long valLongMax = LongEncoding.decode("7~~~~~~~~~~"); + Assert.assertEquals(Long.MAX_VALUE, valLongMax); + } + + @Test + public void testEncodeSortable() { + String val1234 = LongEncoding.encodeSortable(1234); + Assert.assertEquals("2JI", val1234); + + String val23 = LongEncoding.encodeSortable(23); + Assert.assertEquals("1N", val23); + + String valIntMax = LongEncoding.encodeSortable(Integer.MAX_VALUE); + Assert.assertEquals("61~~~~~", valIntMax); + + String valLongMax = LongEncoding.encodeSortable(Long.MAX_VALUE); + Assert.assertEquals("B7~~~~~~~~~~", valLongMax); + + String val0 = LongEncoding.encodeSortable(0); + Assert.assertEquals("10", val0); + + String valNeg1 = LongEncoding.encodeSortable(-1); + Assert.assertEquals("0B7~~~~~~~~~~", valNeg1); + + String valIntMin = LongEncoding.encodeSortable(Integer.MIN_VALUE); + Assert.assertEquals("0B7~~~~z00000", valIntMin); + + String valLongMin = LongEncoding.encodeSortable(Long.MIN_VALUE); + Assert.assertEquals("010", valLongMin); + } + + @Test + public void testDecodeSortable() { + long val1234 = LongEncoding.decodeSortable("2JI"); + Assert.assertEquals(1234, val1234); + + long val23 = LongEncoding.decodeSortable("1N"); + Assert.assertEquals(23, val23); + + long valIntMax = LongEncoding.decodeSortable("61~~~~~"); + Assert.assertEquals(Integer.MAX_VALUE, valIntMax); + + long valLongMax = LongEncoding.decodeSortable("B7~~~~~~~~~~"); + Assert.assertEquals(Long.MAX_VALUE, valLongMax); + + long val0 = LongEncoding.decodeSortable("10"); + Assert.assertEquals(0, val0); + + long valn1 = LongEncoding.decodeSortable("0B7~~~~~~~~~~"); + Assert.assertEquals(-1, valn1); + + long valIntMin = LongEncoding.decodeSortable("0B7~~~~z00000"); + Assert.assertEquals(Integer.MIN_VALUE, valIntMin); + + long valLongMin = LongEncoding.decodeSortable("010"); + Assert.assertEquals(Long.MIN_VALUE, valLongMin); + } + + @Test + public void testEncodeNumber() { + String l1234 = LongEncoding.encodeNumber(1234); + Assert.assertEquals("2JI", l1234); + + String d1234 = LongEncoding.encodeNumber(1.234); + Assert.assertEquals("B3~okcR8i3aO", d1234); + + String d21 = LongEncoding.encodeNumber(2.1); + Assert.assertEquals("B400oCoCoCoD", d21); + + String dpi = LongEncoding.encodeNumber(3.1415926); + Assert.assertEquals("B4098ViD4iXA", dpi); + + String fpi = LongEncoding.encodeNumber(3.1415926f); + Assert.assertEquals("610IG~Q", fpi); + + String fn1 = LongEncoding.encodeNumber(-1.0f); + Assert.assertEquals("0B7~~~~~0V~~~", fn1); + + String bMax = LongEncoding.encodeNumber(Byte.MAX_VALUE); + Assert.assertEquals("21~", bMax); + + String bMin = LongEncoding.encodeNumber(Byte.MIN_VALUE); + Assert.assertEquals("0B7~~~~~~~~z0", bMin); + + String sMax = LongEncoding.encodeNumber(Short.MAX_VALUE); + Assert.assertEquals("37~~", sMax); + + String sMin = LongEncoding.encodeNumber(Short.MIN_VALUE); + Assert.assertEquals("0B7~~~~~~~t00", sMin); + + String iMax = LongEncoding.encodeNumber(Integer.MAX_VALUE); + Assert.assertEquals("61~~~~~", iMax); + + String iMin = LongEncoding.encodeNumber(Integer.MIN_VALUE); + Assert.assertEquals("0B7~~~~z00000", iMin); + + String lMax = LongEncoding.encodeNumber(Long.MAX_VALUE); + Assert.assertEquals("B7~~~~~~~~~~", lMax); + + String lMin = LongEncoding.encodeNumber(Long.MIN_VALUE); + Assert.assertEquals("010", lMin); + + String fMax = LongEncoding.encodeNumber(Float.MAX_VALUE); + Assert.assertEquals("61~V~~~", fMax); + + String fMin = LongEncoding.encodeNumber(Float.MIN_VALUE); + Assert.assertEquals("11", fMin); + + String dMax = LongEncoding.encodeNumber(Double.MAX_VALUE); + Assert.assertEquals("B7~k~~~~~~~~", dMax); + + String dMin = LongEncoding.encodeNumber(Double.MIN_VALUE); + Assert.assertEquals("11", dMin); + + String bdLong = LongEncoding.encodeNumber(new BigDecimal("1234")); + Assert.assertEquals("2JI", bdLong); + + String bdFLong = LongEncoding.encodeNumber(new BigDecimal("1234.00")); + Assert.assertEquals("2JI", bdFLong); + + String bdDouble = LongEncoding.encodeNumber(new BigDecimal("1.234")); + Assert.assertEquals("B3~okcR8i3aO", bdDouble); + + @SuppressWarnings("deprecation") + String date = LongEncoding.encodeNumber(new Date(2018, 12, 18)); + Assert.assertEquals("8D~1pQ_G0", date); + } + + @Test + public void testDecodeNumber() { + Number l1234 = LongEncoding.decodeNumber("2JI", Long.class); + Assert.assertEquals(1234L, l1234); + + Number d1234 = LongEncoding.decodeNumber("B3~okcR8i3aO", Double.class); + Assert.assertEquals(1.234, d1234); + + Number d21 = LongEncoding.decodeNumber("B400oCoCoCoD", Double.class); + Assert.assertEquals(2.1, d21); + + Number dpi = LongEncoding.decodeNumber("B4098ViD4iXA", Double.class); + Assert.assertEquals(3.1415926, dpi); + + Number fpi = LongEncoding.decodeNumber("610IG~Q", Float.class); + Assert.assertEquals(3.1415926f, fpi); + + Number fn1 = LongEncoding.decodeNumber("0B7~~~~~0V~~~", Float.class); + Assert.assertEquals(-1.0f, fn1); + + Number bMax = LongEncoding.decodeNumber("21~", Byte.class); + Assert.assertEquals(Byte.MAX_VALUE, bMax); + + Number bMin = LongEncoding.decodeNumber("0B7~~~~~~~~z0", Byte.class); + Assert.assertEquals(Byte.MIN_VALUE, bMin); + + Number sMax = LongEncoding.decodeNumber("37~~", Short.class); + Assert.assertEquals(Short.MAX_VALUE, sMax); + + Number sMin = LongEncoding.decodeNumber("0B7~~~~~~~t00", Short.class); + Assert.assertEquals(Short.MIN_VALUE, sMin); + + Number iMax = LongEncoding.decodeNumber("61~~~~~", Integer.class); + Assert.assertEquals(Integer.MAX_VALUE, iMax); + + Number iMin = LongEncoding.decodeNumber("0B7~~~~z00000", Integer.class); + Assert.assertEquals(Integer.MIN_VALUE, iMin); + + Number lMax = LongEncoding.decodeNumber("B7~~~~~~~~~~", Long.class); + Assert.assertEquals(Long.MAX_VALUE, lMax); + + Number lMin = LongEncoding.decodeNumber("010", Long.class); + Assert.assertEquals(Long.MIN_VALUE, lMin); + + Number fMax = LongEncoding.decodeNumber("61~V~~~", Float.class); + Assert.assertEquals(Float.MAX_VALUE, fMax); + + Number fMin = LongEncoding.decodeNumber("11", Float.class); + Assert.assertEquals(Float.MIN_VALUE, fMin); + + Number dMax = LongEncoding.decodeNumber("B7~k~~~~~~~~", Double.class); + Assert.assertEquals(Double.MAX_VALUE, dMax); + + Number dMin = LongEncoding.decodeNumber("11", Double.class); + Assert.assertEquals(Double.MIN_VALUE, dMin); + } + + @Test + public void testEncodeSortableThenCompare() { + int count = 100000; + Random random1 = new Random(); + Random random2 = new Random(); + for (int i = 0; i < count; i++) { + long num1 = random1.nextLong(); + long num2 = random2.nextLong(); + + String encoded1 = LongEncoding.encodeSortable(num1); + String encoded2 = LongEncoding.encodeSortable(num2); + int cmp = Bytes.compare(encoded1.getBytes(), encoded2.getBytes()); + + if (num1 == num2) { + Assert.assertTrue(cmp == 0); + } else if (num1 > num2) { + Assert.assertTrue(cmp > 0); + } else { + assert num1 < num2; + Assert.assertTrue(cmp < 0); + } + } + } + + @Test + public void testEncodeNumberThenCompare() { + int count = 100000; + Random random1 = new Random(); + Random random2 = new Random(); + + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextInt(), random2.nextInt()); + } + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextLong(), random2.nextLong()); + } + + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextInt(), random2.nextLong()); + } + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextLong(), random2.nextInt()); + } + + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextFloat(), random2.nextFloat()); + } + for (int i = 0; i < count; i++) { + compareEncodedNumber(random1.nextDouble(), random2.nextDouble()); + } + } + + private static void compareEncodedNumber(Number num1, Number num2) { + int cmpExpected = NumericUtil.compareNumber(num1, num2); + + String encoded1 = LongEncoding.encodeNumber(num1); + String encoded2 = LongEncoding.encodeNumber(num2); + int cmp = Bytes.compare(encoded1.getBytes(), encoded2.getBytes()); + + if (cmpExpected == 0) { + Assert.assertTrue(cmp == 0); + } else if (cmpExpected > 0) { + Assert.assertTrue(cmp > 0); + } else { + assert cmpExpected < 0; + Assert.assertTrue(cmp < 0); + } + } +} diff --git a/src/test/java/com/baidu/hugegraph/unit/util/VersionUtilTest.java b/src/test/java/com/baidu/hugegraph/unit/util/VersionUtilTest.java index e7733b4e1..29e91fd83 100644 --- a/src/test/java/com/baidu/hugegraph/unit/util/VersionUtilTest.java +++ b/src/test/java/com/baidu/hugegraph/unit/util/VersionUtilTest.java @@ -74,6 +74,22 @@ public void testVersionWithThreePart() { Assert.assertFalse(VersionUtil.match(version, "3.20.0", "3.3")); } + @Test + public void testVersionGte() { + String version = "0.2"; + + Assert.assertTrue(VersionUtil.gte(version, "0.2")); + Assert.assertTrue(VersionUtil.gte(version, "0.2.0")); + Assert.assertTrue(VersionUtil.gte(version, "0.1")); + Assert.assertTrue(VersionUtil.gte(version, "0.1.9")); + Assert.assertTrue(VersionUtil.gte(version, "0.0.3")); + + Assert.assertFalse(VersionUtil.gte(version, "0.2.0.1")); + Assert.assertFalse(VersionUtil.gte(version, "0.2.1")); + Assert.assertFalse(VersionUtil.gte(version, "0.3")); + Assert.assertFalse(VersionUtil.gte(version, "0.10")); + } + @Test public void testVersionCheck() { Version version = Version.of("0.6.5");