it's coming along

internal/processing/timeline/manager.go

@@ -19,35 +19,158 @@
 package timeline
 
 import (
+	"sync"
+
+	"github.com/sirupsen/logrus"
 	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
 	"github.com/superseriousbusiness/gotosocial/internal/config"
 	"github.com/superseriousbusiness/gotosocial/internal/db"
 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+	"github.com/superseriousbusiness/gotosocial/internal/typeutils"
+)
+
+const (
+	preparedPostsMinLength = 80
+	desiredPostIndexLength = 400
 )
 
 type Manager interface {
-	Ingest(status *gtsmodel.Status) error
-	HomeTimelineGet(account *gtsmodel.Account, maxID string, sinceID string, minID string, limit int, local bool) ([]apimodel.Status, error)
+	Ingest(status *gtsmodel.Status, timelineAccountID string) error
+	HomeTimeline(timelineAccountID string, maxID string, sinceID string, minID string, limit int, local bool) ([]*apimodel.Status, error)
+	GetIndexedLength(timelineAccountID string) int
+	GetDesiredIndexLength() int
+	GetOldestIndexedID(timelineAccountID string) (string, error)
+	PrepareXFromTop(timelineAccountID string, limit int) error
 }
 
-func NewManager(db db.DB, config *config.Config) Manager {
+func NewManager(db db.DB, tc typeutils.TypeConverter, config *config.Config, log *logrus.Logger) Manager {
 	return &manager{
-		accountTimelines: make(map[string]*timeline),
+		accountTimelines: sync.Map{},
 		db:               db,
+		tc:               tc,
 		config:           config,
+		log:              log,
 	}
 }
 
 type manager struct {
-	accountTimelines map[string]*timeline
+	accountTimelines sync.Map
 	db               db.DB
+	tc               typeutils.TypeConverter
 	config           *config.Config
+	log              *logrus.Logger
 }
 
-func (m *manager) Ingest(status *gtsmodel.Status) error {
-	return nil
+func (m *manager) Ingest(status *gtsmodel.Status, timelineAccountID string) error {
+	l := m.log.WithFields(logrus.Fields{
+		"func":              "Ingest",
+		"timelineAccountID": timelineAccountID,
+		"statusID":          status.ID,
+	})
+
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	l.Trace("ingesting status")
+
+	return t.IndexOne(status.CreatedAt, status.ID)
 }
 
-func (m *manager) HomeTimelineGet(account *gtsmodel.Account, maxID string, sinceID string, minID string, limit int, local bool) ([]apimodel.Status, error) {
-	return nil, nil
+func (m *manager) Remove(statusID string, timelineAccountID string) error {
+	l := m.log.WithFields(logrus.Fields{
+		"func":              "Remove",
+		"timelineAccountID": timelineAccountID,
+		"statusID":          statusID,
+	})
+
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	l.Trace("removing status")
+
+	return t.Remove(statusID)
+}
+
+func (m *manager) HomeTimeline(timelineAccountID string, maxID string, sinceID string, minID string, limit int, local bool) ([]*apimodel.Status, error) {
+	l := m.log.WithFields(logrus.Fields{
+		"func":              "HomeTimelineGet",
+		"timelineAccountID": timelineAccountID,
+	})
+
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	var err error
+	var statuses []*apimodel.Status
+	if maxID != "" {
+		statuses, err = t.GetXBehindID(limit, maxID)
+	} else {
+		statuses, err = t.GetXFromTop(limit)
+	}
+
+	if err != nil {
+		l.Errorf("error getting statuses: %s", err)
+	}
+	return statuses, nil
+}
+
+func (m *manager) GetIndexedLength(timelineAccountID string) int {
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	return t.PostIndexLength()
+}
+
+func (m *manager) GetDesiredIndexLength() int {
+	return desiredPostIndexLength
+}
+
+func (m *manager) GetOldestIndexedID(timelineAccountID string) (string, error) {
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	return t.OldestIndexedPostID()
+}
+
+func (m *manager) PrepareXFromTop(timelineAccountID string, limit int) error {
+	var t Timeline
+	i, ok := m.accountTimelines.Load(timelineAccountID)
+	if !ok {
+		t = NewTimeline(timelineAccountID, m.db, m.tc)
+		m.accountTimelines.Store(timelineAccountID, t)
+	} else {
+		t = i.(Timeline)
+	}
+
+	return t.PrepareXFromTop(limit)
 }