Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Java 8 fill array with supplier

Is there a way to fill an array using java 8 Supplier ?

I would like to write:

Supplier<Object> supplier = () -> new Object();
Object[] array = new Object[size];
Arrays.fill(array, supplier);

Note: I know i could write my own method.

like image 500
gontard Avatar asked Aug 01 '14 09:08

gontard


2 Answers

In case you want to create new array filled with results generated by Supplier you can use

Object[] array = Stream.generate(supplier)
                       .limit(arraySize)
                       .toArray(); // will generate new *Object[]* array

For different types than Object[] you can use toArray(IntFunction<YourType[]> generator); like toArray(YourType[]::new) (credits to @Holger).

String[] array  = Stream.generate(supplier)
                        .limit(10)
                        .toArray(String[]::new); //now *String[]* array will be returned

Streams also allow us to work with most "popular" primitive types which are int long and double. For instance we can use IntStream#toArray to create int[] holding elements from IntStream. To "fill" IntStream with elements from supplier we can use IntStream.generate(intSupplier) like

int[] array = IntStream.generate(()->1)
                       .limit(5)
                       .toArray(); //returns `new Int[]{1,1,1,1,1}

In case when you want to fill already existing array with data from Supplier see answer posted by Stuart Marks based on Arrays.setAll(array, supplier) which aside from handling arrays of objects also supports some arrays of primitive types: double[] int[] and long[] .

Other alternative is to use use creative solution from @Hogler's comment:

Arrays.asList(array).replaceAll(x -> supplier.get()); 
//you can even overwrite a range using `subList`

just be aware of its problems with array of primitive types explained by Jon Skeet at https://stackoverflow.com/a/1467940.

like image 186
Pshemo Avatar answered Nov 03 '22 05:11

Pshemo


In java.util.Arrays there is

<T> void Arrays.setAll(T[] array, IntFunction<T> generator)

This doesn't take a supplier; instead it takes an IntFunction whose input argument is the array index being filled. If your objects aren't dependent upon the destination array index, you can disregard the parameter and call a supplier like this:

Arrays.setAll(array, i -> supplier.get());

There are overloads for arrays of primitives as well as arrays of reference type. There is also a corresponding family of methods parallelSetAll() that does the same thing, except in parallel. (It uses streams internally.)

like image 31
Stuart Marks Avatar answered Nov 03 '22 05:11

Stuart Marks