commit fb29275dc023b839997eb25db363e70ed0e57aa2 · kitten.sh/wonka

+25 -4
src/operators/wonka_operator_tapAll.re
···

       2
       2
        
       

     

       3
       3
        
       let tapAll = (~onStart, ~onPush, ~onEnd) => {

     

       4
       4
        
         curry(source => curry(sink => {

     

       5
       5
       +
           let ended = ref(false);

     

       6
       6
       +
       

     

       5
       7
        
           source((.signal) => {

     

       6
       8
        
             switch (signal) {

     

       7
       7
       -
             | Start(_) => onStart(.)

     

       8
       8
       -
             | Push(x) => onPush(.x)

     

       9
       9
       -
             | End => onEnd(.)

     

       9
       9
       +
             | Start(talkback) => {

     

       10
       10
       +
               onStart(.)

     

       11
       11
       +
       

     

       12
       12
       +
               sink(.Start((.signal) => {

     

       13
       13
       +
                 switch (signal) {

     

       14
       14
       +
                 | Close when !ended^ => {

     

       15
       15
       +
                   ended := true;

     

       16
       16
       +
                   onEnd(.);

     

       17
       17
       +
                 }

     

       18
       18
       +
                 | Close => ()

     

       19
       19
       +
                 | _ => talkback(.signal)

     

       20
       20
       +
                 }

     

       21
       21
       +
               }));

     

       22
       22
       +
             }

     

       23
       23
       +
             | Push(x) => {

     

       24
       24
       +
               onPush(.x)

     

       25
       25
       +
               sink(.signal);

     

       26
       26
       +
             }

     

       27
       27
       +
             | End when !ended^ => {

     

       28
       28
       +
               ended := true;

     

       29
       29
       +
               onEnd(.)

     

       30
       30
       +
             }

     

       31
       31
       +
             | _ => ()

     

       10
       32
        
             };

     

       11
       33
        
       

     

       12
       12
       -
             sink(.signal);

     

       13
       34
        
           });

     

       14
       35
        
         }));

     

       15
       36
        
       };