mirror of
https://gitee.com/rulego/streamsql.git
synced 2026-04-13 15:48:52 +00:00
rulego-team
126 lines
2.8 KiB
Go
126 lines
2.8 KiB
Go
/*
|
|
* Copyright 2024 The RuleGo Authors.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package window
|
|
|
|
import (
|
|
"github.com/rulego/streamsql/types"
|
|
queue2 "github.com/rulego/streamsql/utils/queue"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
type Window struct {
|
|
context types.SelectStreamSqlContext
|
|
observer types.WindowObserver
|
|
windowSize int32 // the size of the window
|
|
count int32
|
|
lastTime time.Time // the last time the window was updated
|
|
fieldName string // the name of the field that is being aggregated
|
|
//maxCapacity 最大容量
|
|
maxCapacity int
|
|
queue *queue2.Queue
|
|
startTime time.Time
|
|
endTime time.Time
|
|
//退出标志
|
|
quit chan struct{}
|
|
locker sync.Mutex
|
|
}
|
|
|
|
// 检查是否需要是下一个窗口
|
|
func (w *Window) checkNextWindow() {
|
|
w.locker.Lock()
|
|
defer w.locker.Unlock()
|
|
if atomic.LoadInt32(&w.count) >= w.windowSize {
|
|
//结束当前窗口
|
|
w.end()
|
|
//开始新的窗口
|
|
w.start()
|
|
}
|
|
}
|
|
|
|
// 开始窗口事件
|
|
func (w *Window) start() {
|
|
w.startTime = time.Now()
|
|
atomic.StoreInt32(&w.count, 0)
|
|
w.queue.Reset()
|
|
|
|
if w.observer.StartHandler != nil {
|
|
w.observer.StartHandler(w.context)
|
|
}
|
|
}
|
|
|
|
// 结束窗口事件
|
|
func (w *Window) end() {
|
|
w.endTime = time.Now()
|
|
if w.observer.EndHandler != nil {
|
|
w.observer.EndHandler(w.context, w.queue.PopAll())
|
|
}
|
|
}
|
|
|
|
// 队列满，触发full事件，并重置队列
|
|
func (w *Window) full() {
|
|
if w.observer.ArchiveHandler != nil {
|
|
w.observer.ArchiveHandler(w.context, w.queue.PopAll())
|
|
}
|
|
//重置队列
|
|
w.queue.Reset()
|
|
}
|
|
|
|
// Add 添加数据
|
|
func (w *Window) Add(data float64) {
|
|
if atomic.LoadInt32(&w.count) >= w.windowSize {
|
|
w.checkNextWindow()
|
|
}
|
|
if w.queue.IsFull() {
|
|
w.full()
|
|
}
|
|
atomic.AddInt32(&w.count, 1)
|
|
_ = w.queue.Push(data)
|
|
if w.observer.AddHandler != nil {
|
|
w.observer.AddHandler(w.context, data)
|
|
}
|
|
}
|
|
|
|
// FieldName 获取聚合运算字段名称
|
|
func (w *Window) FieldName() string {
|
|
return w.fieldName
|
|
}
|
|
|
|
// LastData 获取最后一条数据
|
|
func (w *Window) LastData() (float64, bool) {
|
|
return w.queue.Back()
|
|
}
|
|
|
|
// StartTime 获取窗口开始时间
|
|
func (w *Window) StartTime() time.Time {
|
|
return w.startTime
|
|
}
|
|
|
|
// EndTime 获取窗口结束时间
|
|
func (w *Window) EndTime() time.Time {
|
|
return w.endTime
|
|
}
|
|
|
|
// Archive 保存数据
|
|
func (w *Window) Archive() {
|
|
|
|
}
|
|
func (w *Window) Stop() {
|
|
w.quit <- struct{}{}
|
|
}
|