Skip to content

Commit 3fa6919

Browse files
committed
Adicionado drift modifier
1 parent e433b8a commit 3fa6919

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

src/modifiers.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,5 +34,17 @@ function apply!(modifier::FilterModifier, data::DataFrame, event::Int)
3434
end
3535

3636
select!(data, columns)
37+
return nothing
38+
end
39+
struct DriftModifier <: Modifier
40+
column::Symbol
41+
filter::Function
42+
drift::Function
43+
end
44+
45+
function apply!(modifier::DriftModifier, data::DataFrame, event::Int)
46+
elements = modifier.filter(data)
47+
data[elements, modifier.column] = data[elements, modifier.column] .+ modifier.drift(event)
48+
3749
return nothing
3850
end

src/stream.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ function increment(stream::AbstractStream)
1515
stream.events += 1
1616
return nothing
1717
end
18+
1819
mutable struct BatchStream <: AbstractStream
1920
connector::AbstractConnector
2021
batch::Int
@@ -36,6 +37,7 @@ function listen(stream::BatchStream)::DataFrame
3637
end
3738

3839
increment(stream)
40+
3941
values = DataFrame[]
4042

4143
for i = 1:stream.batch

0 commit comments

Comments
 (0)