add reader stat to torrent stat

server/torr/state/state.go

@@ -1,5 +1,9 @@
 package state
 
+import (
+	"server/torr/storage/reader"
+)
+
 type TorrentStat int
 
 func (t TorrentStat) String() string {
@@ -62,6 +66,8 @@ type TorrentStatus struct {
 	PiecesDirtiedGood   int64       `json:"pieces_dirtied_good,omitempty"`
 	PiecesDirtiedBad    int64       `json:"pieces_dirtied_bad,omitempty"`
 
+	Readers []*reader.ReaderState `json:"readers,omitempty"`
+
 	FileStats []*TorrentFileStat `json:"file_stats,omitempty"`
 }
 

server/torr/storage/reader/state.go

@@ -0,0 +1,7 @@
+package reader
+
+type ReaderState struct {
+	Start  int
+	End    int
+	Reader int
+}

server/torr/storage/state/state.go

@@ -2,6 +2,7 @@ package state
 
 import (
 	"server/torr/state"
+	"server/torr/storage/reader"
 )
 
 type CacheState struct {
@@ -12,7 +13,7 @@ type CacheState struct {
 	PiecesCount  int
 	Torrent      *state.TorrentStatus
 	Pieces       map[int]ItemState
-	Readers      []*ReaderState
+	Readers      []*reader.ReaderState
 }
 
 type ItemState struct {
@@ -21,9 +22,3 @@ type ItemState struct {
 	Size      int64
 	Completed bool
 }
-
-type ReaderState struct {
-	Start  int
-	End    int
-	Reader int
-}

server/torr/storage/torrstor/cache.go

@@ -13,6 +13,7 @@ import (
 
 	"server/log"
 	"server/settings"
+	"server/torr/storage/reader"
 	"server/torr/storage/state"
 	"server/torr/utils"
 
@@ -149,12 +150,12 @@ func (c *Cache) GetState() *state.CacheState {
 		}
 	}
 
-	readersState := make([]*state.ReaderState, 0)
+	readersState := make([]*reader.ReaderState, 0)
 	c.muReaders.Lock()
 	for r, _ := range c.readers {
 		rng := r.getPiecesRange()
 		pc := r.getReaderPiece()
-		readersState = append(readersState, &state.ReaderState{
+		readersState = append(readersState, &reader.ReaderState{
 			Start:  rng.Start,
 			End:    rng.End,
 			Reader: pc,

server/torr/torrent.go

@@ -427,6 +427,11 @@ func (t *Torrent) Status() *state.TorrentStatus {
 			}
 		}
 	}
+
+	if t.cache != nil {
+		st.Readers = t.cache.GetState().Readers
+	}
+
 	return st
 }