Merge changes I099c8315,I21e958d3

* changes:
  ArrayUtils: add concat() for byte arrays
  ArrayUtils: rename concatElements() to concat()
This commit is contained in:
Treehugger Robot
2023-02-10 14:08:28 +00:00
committed by Gerrit Code Review
3 changed files with 110 additions and 43 deletions

View File

@@ -39,8 +39,7 @@ import java.util.Set;
import java.util.function.IntFunction; import java.util.function.IntFunction;
/** /**
* ArrayUtils contains some methods that you can call to find out * Static utility methods for arrays that aren't already included in {@link java.util.Arrays}.
* the most efficient increments by which to grow arrays.
*/ */
public class ArrayUtils { public class ArrayUtils {
private static final int CACHE_SIZE = 73; private static final int CACHE_SIZE = 73;
@@ -341,15 +340,16 @@ public class ArrayUtils {
} }
/** /**
* Combine multiple arrays into a single array. * Returns the concatenation of the given arrays. Only works for object arrays, not for
* primitive arrays. See {@link #concat(byte[]...)} for a variant that works on byte arrays.
* *
* @param kind The class of the array elements * @param kind The class of the array elements
* @param arrays The arrays to combine * @param arrays The arrays to concatenate. Null arrays are treated as empty.
* @param <T> The class of the array elements (inferred from kind). * @param <T> The class of the array elements (inferred from kind).
* @return A single array containing all the elements of the parameter arrays. * @return A single array containing all the elements of the parameter arrays.
*/ */
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public static @NonNull <T> T[] concatElements(Class<T> kind, @Nullable T[]... arrays) { public static @NonNull <T> T[] concat(Class<T> kind, @Nullable T[]... arrays) {
if (arrays == null || arrays.length == 0) { if (arrays == null || arrays.length == 0) {
return createEmptyArray(kind); return createEmptyArray(kind);
} }
@@ -390,6 +390,29 @@ public class ArrayUtils {
return (T[]) Array.newInstance(kind, 0); return (T[]) Array.newInstance(kind, 0);
} }
/**
* Returns the concatenation of the given byte arrays. Null arrays are treated as empty.
*/
public static @NonNull byte[] concat(@Nullable byte[]... arrays) {
if (arrays == null) {
return new byte[0];
}
int totalLength = 0;
for (byte[] a : arrays) {
if (a != null) {
totalLength += a.length;
}
}
final byte[] result = new byte[totalLength];
int pos = 0;
for (byte[] a : arrays) {
if (a != null) {
System.arraycopy(a, 0, result, pos, a.length);
pos += a.length;
}
}
return result;
}
/** /**
* Adds value to given array if not already present, providing set-like * Adds value to given array if not already present, providing set-like

View File

@@ -16,8 +16,6 @@
package com.android.internal.util; package com.android.internal.util;
import static com.android.internal.util.ArrayUtils.concatElements;
import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertArrayEquals;
import junit.framework.TestCase; import junit.framework.TestCase;
@@ -156,61 +154,107 @@ public class ArrayUtilsTest extends TestCase {
ArrayUtils.removeLong(new long[] { 1, 2, 3, 1 }, 1)); ArrayUtils.removeLong(new long[] { 1, 2, 3, 1 }, 1));
} }
public void testConcatEmpty() throws Exception { public void testConcat_zeroObjectArrays() {
assertArrayEquals(new Long[] {}, // empty varargs array
concatElements(Long.class, null, null)); assertArrayEquals(new String[] {}, ArrayUtils.concat(String.class));
assertArrayEquals(new Long[] {}, // null varargs array
concatElements(Long.class, new Long[] {}, null)); assertArrayEquals(new String[] {}, ArrayUtils.concat(String.class, (String[][]) null));
assertArrayEquals(new Long[] {},
concatElements(Long.class, null, new Long[] {}));
assertArrayEquals(new Long[] {},
concatElements(Long.class, new Long[] {}, new Long[] {}));
} }
public void testconcatElements() throws Exception { public void testConcat_oneObjectArray() {
assertArrayEquals(new String[] { "1", "2" },
ArrayUtils.concat(String.class, new String[] { "1", "2" }));
}
public void testConcat_oneEmptyObjectArray() {
assertArrayEquals(new String[] {}, ArrayUtils.concat(String.class, (String[]) null));
assertArrayEquals(new String[] {}, ArrayUtils.concat(String.class, new String[] {}));
}
public void testConcat_twoObjectArrays() {
assertArrayEquals(new Long[] { 1L }, assertArrayEquals(new Long[] { 1L },
concatElements(Long.class, new Long[] { 1L }, new Long[] {})); ArrayUtils.concat(Long.class, new Long[] { 1L }, new Long[] {}));
assertArrayEquals(new Long[] { 1L }, assertArrayEquals(new Long[] { 1L },
concatElements(Long.class, new Long[] {}, new Long[] { 1L })); ArrayUtils.concat(Long.class, new Long[] {}, new Long[] { 1L }));
assertArrayEquals(new Long[] { 1L, 2L }, assertArrayEquals(new Long[] { 1L, 2L },
concatElements(Long.class, new Long[] { 1L }, new Long[] { 2L })); ArrayUtils.concat(Long.class, new Long[] { 1L }, new Long[] { 2L }));
assertArrayEquals(new Long[] { 1L, 2L, 3L, 4L }, assertArrayEquals(new Long[] { 1L, 2L, 3L, 4L },
concatElements(Long.class, new Long[] { 1L, 2L }, new Long[] { 3L, 4L })); ArrayUtils.concat(Long.class, new Long[] { 1L, 2L }, new Long[] { 3L, 4L }));
} }
public void testConcatElements_threeWay() { public void testConcat_twoEmptyObjectArrays() {
assertArrayEquals(new Long[] {}, ArrayUtils.concat(Long.class, null, null));
assertArrayEquals(new Long[] {}, ArrayUtils.concat(Long.class, new Long[] {}, null));
assertArrayEquals(new Long[] {}, ArrayUtils.concat(Long.class, null, new Long[] {}));
assertArrayEquals(new Long[] {},
ArrayUtils.concat(Long.class, new Long[] {}, new Long[] {}));
}
public void testConcat_threeObjectArrays() {
String[] array1 = { "1", "2" }; String[] array1 = { "1", "2" };
String[] array2 = { "3", "4" }; String[] array2 = { "3", "4" };
String[] array3 = { "5", "6" }; String[] array3 = { "5", "6" };
String[] expectation = { "1", "2", "3", "4", "5", "6" }; String[] expectation = { "1", "2", "3", "4", "5", "6" };
String[] concatResult = ArrayUtils.concatElements(String.class, array1, array2, array3); assertArrayEquals(expectation, ArrayUtils.concat(String.class, array1, array2, array3));
assertArrayEquals(expectation, concatResult);
} }
public void testConcat_threeObjectArraysWithNull() {
public void testConcatElements_threeWayWithNull() {
String[] array1 = { "1", "2" }; String[] array1 = { "1", "2" };
String[] array2 = null; String[] array2 = null;
String[] array3 = { "5", "6" }; String[] array3 = { "5", "6" };
String[] expectation = { "1", "2", "5", "6" }; String[] expectation = { "1", "2", "5", "6" };
String[] concatResult = ArrayUtils.concatElements(String.class, array1, array2, array3); assertArrayEquals(expectation, ArrayUtils.concat(String.class, array1, array2, array3));
assertArrayEquals(expectation, concatResult);
} }
public void testConcatElements_zeroElements() { public void testConcat_zeroByteArrays() {
String[] expectation = new String[0]; // empty varargs array
assertArrayEquals(new byte[] {}, ArrayUtils.concat());
String[] concatResult = ArrayUtils.concatElements(String.class); // null varargs array
assertArrayEquals(expectation, concatResult); assertArrayEquals(new byte[] {}, ArrayUtils.concat((byte[][]) null));
} }
public void testConcatElements_oneNullElement() { public void testConcat_oneByteArray() {
String[] expectation = new String[0]; assertArrayEquals(new byte[] { 1, 2 }, ArrayUtils.concat(new byte[] { 1, 2 }));
String[] concatResult = ArrayUtils.concatElements(String.class, null);
assertArrayEquals(expectation, concatResult);
} }
public void testConcat_oneEmptyByteArray() {
assertArrayEquals(new byte[] {}, ArrayUtils.concat((byte[]) null));
assertArrayEquals(new byte[] {}, ArrayUtils.concat(new byte[] {}));
}
public void testConcat_twoByteArrays() {
assertArrayEquals(new byte[] { 1 }, ArrayUtils.concat(new byte[] { 1 }, new byte[] {}));
assertArrayEquals(new byte[] { 1 }, ArrayUtils.concat(new byte[] {}, new byte[] { 1 }));
assertArrayEquals(new byte[] { 1, 2 },
ArrayUtils.concat(new byte[] { 1 }, new byte[] { 2 }));
assertArrayEquals(new byte[] { 1, 2, 3, 4 },
ArrayUtils.concat(new byte[] { 1, 2 }, new byte[] { 3, 4 }));
}
public void testConcat_twoEmptyByteArrays() {
assertArrayEquals(new byte[] {}, ArrayUtils.concat((byte[]) null, null));
assertArrayEquals(new byte[] {}, ArrayUtils.concat(new byte[] {}, null));
assertArrayEquals(new byte[] {}, ArrayUtils.concat((byte[]) null, new byte[] {}));
assertArrayEquals(new byte[] {}, ArrayUtils.concat(new byte[] {}, new byte[] {}));
}
public void testConcat_threeByteArrays() {
byte[] array1 = { 1, 2 };
byte[] array2 = { 3, 4 };
byte[] array3 = { 5, 6 };
byte[] expectation = { 1, 2, 3, 4, 5, 6 };
assertArrayEquals(expectation, ArrayUtils.concat(array1, array2, array3));
}
public void testConcat_threeByteArraysWithNull() {
byte[] array1 = { 1, 2 };
byte[] array2 = null;
byte[] array3 = { 5, 6 };
byte[] expectation = { 1, 2, 5, 6 };
assertArrayEquals(expectation, ArrayUtils.concat(array1, array2, array3));
}
} }

View File

@@ -875,16 +875,16 @@ public class SettingsBackupAgent extends BackupAgentHelper {
String[] whitelist; String[] whitelist;
Map<String, Validator> validators = null; Map<String, Validator> validators = null;
if (contentUri.equals(Settings.Secure.CONTENT_URI)) { if (contentUri.equals(Settings.Secure.CONTENT_URI)) {
whitelist = ArrayUtils.concatElements(String.class, SecureSettings.SETTINGS_TO_BACKUP, whitelist = ArrayUtils.concat(String.class, SecureSettings.SETTINGS_TO_BACKUP,
Settings.Secure.LEGACY_RESTORE_SETTINGS, Settings.Secure.LEGACY_RESTORE_SETTINGS,
DeviceSpecificSettings.DEVICE_SPECIFIC_SETTINGS_TO_BACKUP); DeviceSpecificSettings.DEVICE_SPECIFIC_SETTINGS_TO_BACKUP);
validators = SecureSettingsValidators.VALIDATORS; validators = SecureSettingsValidators.VALIDATORS;
} else if (contentUri.equals(Settings.System.CONTENT_URI)) { } else if (contentUri.equals(Settings.System.CONTENT_URI)) {
whitelist = ArrayUtils.concatElements(String.class, SystemSettings.SETTINGS_TO_BACKUP, whitelist = ArrayUtils.concat(String.class, SystemSettings.SETTINGS_TO_BACKUP,
Settings.System.LEGACY_RESTORE_SETTINGS); Settings.System.LEGACY_RESTORE_SETTINGS);
validators = SystemSettingsValidators.VALIDATORS; validators = SystemSettingsValidators.VALIDATORS;
} else if (contentUri.equals(Settings.Global.CONTENT_URI)) { } else if (contentUri.equals(Settings.Global.CONTENT_URI)) {
whitelist = ArrayUtils.concatElements(String.class, GlobalSettings.SETTINGS_TO_BACKUP, whitelist = ArrayUtils.concat(String.class, GlobalSettings.SETTINGS_TO_BACKUP,
Settings.Global.LEGACY_RESTORE_SETTINGS); Settings.Global.LEGACY_RESTORE_SETTINGS);
validators = GlobalSettingsValidators.VALIDATORS; validators = GlobalSettingsValidators.VALIDATORS;
} else { } else {