separate func

mastodon.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"io"
 	"net/http"
 	"net/url"
 	"path"
@@ -318,6 +319,35 @@ type Event interface {
 	event()
 }
 
+func handleReader(q chan Event, r io.Reader) error {
+	name := ""
+	s := bufio.NewScanner(r)
+	for s.Scan() {
+		line := s.Text()
+		token := strings.SplitN(line, ":", 2)
+		if len(token) != 2 {
+			continue
+		}
+		switch strings.TrimSpace(token[0]) {
+		case "event":
+			name = strings.TrimSpace(token[1])
+		case "data":
+			switch name {
+			case "update":
+				var status Status
+				err = json.Unmarshal([]byte(token[1]), &status)
+				if err == nil {
+					q <- &UpdateEvent{&status}
+				}
+			case "notification":
+			case "delete":
+			}
+		default:
+		}
+	}
+	return ctx.Err()
+}
+
 // StreamingPublic return channel to read events.
 func (c *Client) StreamingPublic(ctx context.Context) (chan Event, error) {
 	url, err := url.Parse(c.config.Server)
@@ -339,36 +369,12 @@ func (c *Client) StreamingPublic(ctx context.Context) (chan Event, error) {
 				resp, err = c.Do(req)
 			}
 			if err == nil {
-				name := ""
+				err = handleReader(resp.Body)
-				s := bufio.NewScanner(resp.Body)
-				for s.Scan() {
-					line := s.Text()
-					token := strings.SplitN(line, ":", 2)
-					if len(token) != 2 {
-						continue
-					}
-					switch strings.TrimSpace(token[0]) {
-					case "event":
-						name = strings.TrimSpace(token[1])
-					case "data":
-						switch name {
-						case "update":
-							var status Status
-							err = json.Unmarshal([]byte(token[1]), &status)
-							if err == nil {
-								q <- &UpdateEvent{&status}
-							}
-						case "notification":
-						case "delete":
-						}
-					default:
-					}
-				}
-				resp.Body.Close()
-				err = ctx.Err()
 				if err == nil {
 					break
 				}
+				resp.Body.Close()
+				return err
 			} else {
 				q <- &ErrorEvent{err}
 			}