You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.0 KiB
Lua
43 lines
1.0 KiB
Lua
local Observable = require "rx.observable"
|
|
|
|
--- Returns a new Observable that only produces the first n results of the original.
|
|
-- @arg {number=1} n - The number of elements to produce before completing.
|
|
-- @returns {Observable}
|
|
function Observable:take(n)
|
|
n = n or 1
|
|
|
|
return Observable.create(function(observer)
|
|
local subscription
|
|
if n <= 0 then
|
|
observer:onCompleted()
|
|
return
|
|
end
|
|
|
|
local i = 1
|
|
|
|
local function onNext(...)
|
|
observer:onNext(...)
|
|
|
|
i = i + 1
|
|
|
|
if i > n then
|
|
if subscription then
|
|
subscription:unsubscribe()
|
|
end
|
|
observer:onCompleted()
|
|
end
|
|
end
|
|
|
|
local function onError(e)
|
|
return observer:onError(e)
|
|
end
|
|
|
|
local function onCompleted()
|
|
return observer:onCompleted()
|
|
end
|
|
|
|
subscription = self:subscribe(onNext, onError, onCompleted)
|
|
return subscription
|
|
end)
|
|
end
|