refactor: moved mirrorStatus back to manager

manager/status.go

@@ -0,0 +1,90 @@
+package manager
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"strconv"
+	"time"
+
+	. "github.com/tuna/tunasync/internal"
+)
+
+type mirrorStatus struct {
+	Name       string
+	Status     SyncStatus
+	LastUpdate time.Time
+	Upstream   string
+	Size       string // approximate size
+}
+
+func (s mirrorStatus) MarshalJSON() ([]byte, error) {
+	m := map[string]interface{}{
+		"name":           s.Name,
+		"status":         s.Status,
+		"last_update":    s.LastUpdate.Format("2006-01-02 15:04:05"),
+		"last_update_ts": fmt.Sprintf("%d", s.LastUpdate.Unix()),
+		"size":           s.Size,
+		"upstream":       s.Upstream,
+	}
+	return json.Marshal(m)
+}
+
+func (s *mirrorStatus) UnmarshalJSON(v []byte) error {
+	var m map[string]interface{}
+
+	err := json.Unmarshal(v, &m)
+	if err != nil {
+		return err
+	}
+
+	if name, ok := m["name"]; ok {
+		if s.Name, ok = name.(string); !ok {
+			return errors.New("name should be a string")
+		}
+	} else {
+		return errors.New("key `name` does not exist in the json")
+	}
+	if upstream, ok := m["upstream"]; ok {
+		if s.Upstream, ok = upstream.(string); !ok {
+			return errors.New("upstream should be a string")
+		}
+	} else {
+		return errors.New("key `upstream` does not exist in the json")
+	}
+	if size, ok := m["size"]; ok {
+		if s.Size, ok = size.(string); !ok {
+			return errors.New("size should be a string")
+		}
+	} else {
+		return errors.New("key `size` does not exist in the json")
+	}
+	// tricky: status
+	if status, ok := m["status"]; ok {
+		if ss, ok := status.(string); ok {
+			err := json.Unmarshal([]byte(`"`+ss+`"`), &(s.Status))
+			if err != nil {
+				return err
+			}
+		} else {
+			return errors.New("status should be a string")
+		}
+	} else {
+		return errors.New("key `status` does not exist in the json")
+	}
+	// tricky: last update
+	if lastUpdate, ok := m["last_update_ts"]; ok {
+		if sts, ok := lastUpdate.(string); ok {
+			ts, err := strconv.Atoi(sts)
+			if err != nil {
+				return fmt.Errorf("last_update_ts should be a interger, got: %s", sts)
+			}
+			s.LastUpdate = time.Unix(int64(ts), 0)
+		} else {
+			return fmt.Errorf("last_update_ts should be a string of integer, got: %s", lastUpdate)
+		}
+	} else {
+		return errors.New("key `last_update_ts` does not exist in the json")
+	}
+	return nil
+}

manager/status_test.go

@@ -0,0 +1,41 @@
+package manager
+
+import (
+	"encoding/json"
+	"testing"
+	"time"
+
+	tunasync "github.com/tuna/tunasync/internal"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestStatus(t *testing.T) {
+	Convey("status json ser-de should work", t, func() {
+		tz := "Asia/Shanghai"
+		loc, err := time.LoadLocation(tz)
+		So(err, ShouldBeNil)
+
+		m := mirrorStatus{
+			Name:       "tunalinux",
+			Status:     tunasync.Success,
+			LastUpdate: time.Date(2016, time.April, 16, 23, 8, 10, 0, loc),
+			Size:       "5GB",
+			Upstream:   "rsync://mirrors.tuna.tsinghua.edu.cn/tunalinux/",
+		}
+
+		b, err := json.Marshal(m)
+		So(err, ShouldBeNil)
+		// fmt.Println(string(b))
+		var m2 mirrorStatus
+		err = json.Unmarshal(b, &m2)
+		So(err, ShouldBeNil)
+		// fmt.Printf("%#v", m2)
+		So(m2.Name, ShouldEqual, m.Name)
+		So(m2.Status, ShouldEqual, m.Status)
+		So(m2.LastUpdate.Unix(), ShouldEqual, m.LastUpdate.Unix())
+		So(m2.LastUpdate.UnixNano(), ShouldEqual, m.LastUpdate.UnixNano())
+		So(m2.Size, ShouldEqual, m.Size)
+		So(m2.Upstream, ShouldEqual, m.Upstream)
+	})
+}