FilesExpand file tree

 2.x

/

GroupByTests.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

87 lines (79 loc) · 2.85 KB

 2.x

/

GroupByTests.java

Copy path

Top

File metadata and controls

• Code
• Blame

87 lines (79 loc) · 2.85 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

/**

* Copyright 2016 Netflix, Inc.

*

* Licensed 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 io.reactivex;

import org.junit.Test;

import org.reactivestreams.Publisher;

import io.reactivex.Observable;

import io.reactivex.EventStream.Event;

import io.reactivex.functions.*;

import io.reactivex.observables.GroupedObservable;

public class GroupByTests {

@Test

public void testTakeUnsubscribesOnGroupBy() {

Observable.merge(

EventStream.getEventStream("HTTP-ClusterA", 50),

EventStream.getEventStream("HTTP-ClusterB", 20)

)

// group by type (2 clusters)

.groupBy(new Function<Event, Object>() {

@Override

public Object apply(Event event) {

return event.type;

}

})

.take(1)

.toBlocking()

.forEach(new Consumer<GroupedObservable<Object, Event>>() {

@Override

public void accept(GroupedObservable<Object, Event> v) {

System.out.println(v);

v.take(1).subscribe(); // FIXME groups need consumption to a certain degree to cancel upstream

}

});

System.out.println("**** finished");

}

@Test

public void testTakeUnsubscribesOnFlatMapOfGroupBy() {

Observable.merge(

EventStream.getEventStream("HTTP-ClusterA", 50),

EventStream.getEventStream("HTTP-ClusterB", 20)

)

// group by type (2 clusters)

.groupBy(new Function<Event, Object>() {

@Override

public Object apply(Event event) {

return event.type;

}

})

.flatMap(new Function<GroupedObservable<Object, Event>, Publisher<Object>>() {

@Override

public Publisher<Object> apply(GroupedObservable<Object, Event> g) {

return g.map(new Function<Event, Object>() {

@Override

public Object apply(Event event) {

return event.instanceId + " - " + event.values.get("count200");

}

});

}

})

.take(20)

.toBlocking()

.forEach(new Consumer<Object>() {

@Override

public void accept(Object v) {

System.out.println(v);

}

});

System.out.println("**** finished");

}

}