Ejemplos de _TimerCoder en Python

Ejemplo n.º 1

Archivo: coders_test_common.py Proyecto: eralmas7/beam

 def test_timer_coder(self):
   self.check_coder(coders._TimerCoder(coders.BytesCoder()),
                    *[{'timestamp': timestamp.Timestamp(micros=x),
                       'payload': b'xyz'}
                      for x in range(-3, 3)])
   self.check_coder(
       coders.TupleCoder((coders._TimerCoder(coders.VarIntCoder()),)),
       ({'timestamp': timestamp.Timestamp.of(37), 'payload': 389},))

Ejemplo n.º 2

Archivo: coders_test_common.py Proyecto: yjshen/beam

 def test_timer_coder(self):
   self.check_coder(coders._TimerCoder(coders.BytesCoder()),
                    *[{'timestamp': timestamp.Timestamp(micros=x),
                       'payload': b'xyz'}
                      for x in (-3000, 0, 3000)])
   self.check_coder(
       coders.TupleCoder((coders._TimerCoder(coders.VarIntCoder()),)),
       ({'timestamp': timestamp.Timestamp.of(37000), 'payload': 389},))

Ejemplo n.º 3

Archivo: userstate.py Proyecto: mjacoby/beam

 def to_runner_api(self, context):
     # type: (PipelineContext) -> beam_runner_api_pb2.TimerFamilySpec
     return beam_runner_api_pb2.TimerFamilySpec(
         time_domain=TimeDomain.to_runner_api(self.time_domain),
         timer_family_coder_id=context.coders.get_id(
             coders._TimerCoder(coders.StrUtf8Coder(),
                                coders.GlobalWindowCoder())))

Ejemplo n.º 4

 def test_timer_coder(self):
     self.check_coder(
         coders._TimerCoder(coders.StrUtf8Coder(),
                            coders.GlobalWindowCoder()),
         *[
             userstate.Timer(user_key="key",
                             dynamic_timer_tag="tag",
                             windows=(GlobalWindow(), ),
                             clear_bit=True,
                             fire_timestamp=None,
                             hold_timestamp=None,
                             paneinfo=None),
             userstate.Timer(user_key="key",
                             dynamic_timer_tag="tag",
                             windows=(GlobalWindow(), ),
                             clear_bit=False,
                             fire_timestamp=timestamp.Timestamp.of(123),
                             hold_timestamp=timestamp.Timestamp.of(456),
                             paneinfo=windowed_value.PANE_INFO_UNKNOWN)
         ])

Ejemplo n.º 5

Archivo: userstate.py Proyecto: yaoshi1994/beam

 def to_runner_api(self, context):
     # type: (PipelineContext) -> beam_runner_api_pb2.TimerSpec
     return beam_runner_api_pb2.TimerSpec(
         time_domain=TimeDomain.to_runner_api(self.time_domain),
         timer_coder_id=context.coders.get_id(
             coders._TimerCoder(coders.SingletonCoder(None))))

Ejemplo n.º 6

 def to_runner_api(self, context, key_coder, window_coder):
     return beam_runner_api_pb2.TimerFamilySpec(
         time_domain=TimeDomain.to_runner_api(self.time_domain),
         timer_family_coder_id=context.coders.get_id(
             coders._TimerCoder(key_coder, window_coder)))

Ejemplo n.º 7

Archivo: userstate.py Proyecto: mjacoby/beam

 def to_runner_api(self, context):
     return beam_runner_api_pb2.TimerFamilySpec(
         time_domain=TimeDomain.to_runner_api(self.time_domain),
         timer_family_coder_id=context.coders.get_id(
             coders._TimerCoder(coders.SingletonCoder(None))))

Ejemplo n.º 8

Archivo: userstate.py Proyecto: charlesccychen/incubator-beam

 def to_runner_api(self, context):
   return beam_runner_api_pb2.TimerSpec(
       time_domain=TimeDomain.to_runner_api(self.time_domain),
       timer_coder_id=context.coders.get_id(
           coders._TimerCoder(coders.SingletonCoder(None))))