Akka Stream Create Partition

CreatePartition.java

Graph<UniformFanOutShape<EventMessage, EventMessage>, NotUsed> eventTypePartition = Partition.create(3, 
                                                                                   (Function<EventMessage, Object>) tuple -> {
            switch (tuple.message) {
                case "EVENT_TYPE_1":
                    return 1;
                case "EVENT_TYPE_2":
                    return 2;
                default:
                    return 0;
            }
        });

This is a 3 partition example.