mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-10 11:52:10 +08:00
firestar5683
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from typing import Any, Callable, Dict, List, Optional, TypeVar
|
|
|
|
from reactivex import Observable, abc, typing
|
|
from reactivex.scheduler import TimeoutScheduler
|
|
|
|
_T = TypeVar("_T")
|
|
|
|
|
|
def skip_last_with_time_(
|
|
duration: typing.RelativeTime, scheduler: Optional[abc.SchedulerBase] = None
|
|
) -> Callable[[Observable[_T]], Observable[_T]]:
|
|
"""Skips elements for the specified duration from the end of the
|
|
observable source sequence.
|
|
|
|
Example:
|
|
>>> res = skip_last_with_time(5.0)
|
|
|
|
This operator accumulates a queue with a length enough to store
|
|
elements received during the initial duration window. As more
|
|
elements are received, elements older than the specified duration
|
|
are taken from the queue and produced on the result sequence. This
|
|
causes elements to be delayed with duration.
|
|
|
|
Args:
|
|
duration: Duration for skipping elements from the end of the
|
|
sequence.
|
|
scheduler: Scheduler to use for time handling.
|
|
|
|
Returns:
|
|
An observable sequence with the elements skipped during the
|
|
specified duration from the end of the source sequence.
|
|
"""
|
|
|
|
def skip_last_with_time(source: Observable[_T]) -> Observable[_T]:
|
|
def subscribe(
|
|
observer: abc.ObserverBase[_T],
|
|
scheduler_: Optional[abc.SchedulerBase] = None,
|
|
) -> abc.DisposableBase:
|
|
nonlocal duration
|
|
|
|
_scheduler: abc.SchedulerBase = (
|
|
scheduler or scheduler_ or TimeoutScheduler.singleton()
|
|
)
|
|
duration = _scheduler.to_timedelta(duration)
|
|
q: List[Dict[str, Any]] = []
|
|
|
|
def on_next(x: _T) -> None:
|
|
now = _scheduler.now
|
|
q.append({"interval": now, "value": x})
|
|
while q and now - q[0]["interval"] >= duration:
|
|
observer.on_next(q.pop(0)["value"])
|
|
|
|
def on_completed() -> None:
|
|
now = _scheduler.now
|
|
while q and now - q[0]["interval"] >= duration:
|
|
observer.on_next(q.pop(0)["value"])
|
|
|
|
observer.on_completed()
|
|
|
|
return source.subscribe(
|
|
on_next, observer.on_error, on_completed, scheduler=_scheduler
|
|
)
|
|
|
|
return Observable(subscribe)
|
|
|
|
return skip_last_with_time
|
|
|
|
|
|
__all__ = ["skip_last_with_time_"]
|