refactor

Author: PaulisMatrix

1-producer-consumer/main.go

@@ -66,7 +66,6 @@ func main() {
 	stream := GetMockStream()
 
 	// use a chan for producer to add tweet to it and consumer to consume from
-
 	tweetChan := make(chan *Tweet)
 
 	// Producer

1-producer-consumer/mockstream.go

@@ -12,20 +12,28 @@ import (
 	"time"
 )
 
-// GetMockStream is a blackbox function which returns a mock stream for
-// demonstration purposes
-func GetMockStream() Stream {
-	return Stream{0, mockdata}
+// ErrEOF returns on End of File error
+var ErrEOF = errors.New("End of File")
+
+type Streaming interface{
+	Next() (*Tweet, error)
 }
 
+var _ Streaming = (*Stream)(nil)
+
+
 // Stream is a mock stream for demonstration purposes, not threadsafe
 type Stream struct {
 	pos    int
 	tweets []Tweet
 }
 
-// ErrEOF returns on End of File error
-var ErrEOF = errors.New("End of File")
+// GetMockStream is a blackbox function which returns a mock stream for
+// demonstration purposes
+func GetMockStream() Stream {
+	return Stream{0, mockdata}
+}
+
 
 // Next returns the next Tweet in the stream, returns EOF error if
 // there are no more tweets