mirror of
https://github.com/1f349/dendrite.git
synced 2024-12-23 16:54:08 +00:00
39 lines
640 B
Go
39 lines
640 B
Go
|
package streams
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/matrix-org/dendrite/syncapi/storage"
|
||
|
"github.com/matrix-org/dendrite/syncapi/types"
|
||
|
)
|
||
|
|
||
|
type StreamProvider struct {
|
||
|
DB storage.Database
|
||
|
latest types.StreamPosition
|
||
|
latestMutex sync.RWMutex
|
||
|
}
|
||
|
|
||
|
func (p *StreamProvider) Setup() {
|
||
|
}
|
||
|
|
||
|
func (p *StreamProvider) Advance(
|
||
|
latest types.StreamPosition,
|
||
|
) {
|
||
|
p.latestMutex.Lock()
|
||
|
defer p.latestMutex.Unlock()
|
||
|
|
||
|
if latest > p.latest {
|
||
|
p.latest = latest
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (p *StreamProvider) LatestPosition(
|
||
|
ctx context.Context,
|
||
|
) types.StreamPosition {
|
||
|
p.latestMutex.RLock()
|
||
|
defer p.latestMutex.RUnlock()
|
||
|
|
||
|
return p.latest
|
||
|
}
|