added StreamObject interface

proxima-one · Sep 15, 2022 · 2c3ebf4 · 2c3ebf4
1 parent 6479713
commit 2c3ebf4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/client/stream_reader.go b/client/stream_reader.go
@@ -348,7 +348,7 @@ func (reader *StreamReader) Stop() {
 	reader.stopSync <- struct{}{}
 }
 
-func (reader *StreamReader) ReadNext() (*model.ProximaStreamObject, error) {
+func (reader *StreamReader) ReadNext() (model.StreamObject, error) {
 	select {
 	case <-reader.outputCtx.Done():
 		return nil, reader.outputCtx.Err()

diff --git a/model/stream_object.go b/model/stream_object.go
@@ -2,6 +2,13 @@ package model
 
 import "time"
 
+type StreamObject interface {
+	GetTimestamp() time.Time
+	GetUndo() bool
+	GetTransition() *Transition
+	GetPreprocess() *TransitionPreprocessingResult
+}
+
 type ProximaStreamObject struct {
 	Transition *Transition `json:"transition"`
 	Preprocess *TransitionPreprocessingResult