go/service/tracker_loader.go - master - githubmirror/client

go/service/tracker_loader.go
156 строк4 KB
chrisnojima-zoom
Version 670 - clean2 (#29122)
08 июн 2026, 19:31
08 июн 2026, 19:311943197
package service

import (
	"context"
	"errors"
	"fmt"
	"sync"
	"time"

	"github.com/keybase/client/go/engine"
	"github.com/keybase/client/go/libkb"
	"github.com/keybase/client/go/protocol/keybase1"
	"golang.org/x/sync/errgroup"
)

type TrackerLoader struct {
	libkb.Contextified
	sync.Mutex

	eg         errgroup.Group
	started    bool
	shutdownCh chan struct{}
	queueCh    chan keybase1.UID

	cancel context.CancelFunc
}

func NewTrackerLoader(g *libkb.GlobalContext) *TrackerLoader {
	l := &TrackerLoader{
		Contextified: libkb.NewContextified(g),
		shutdownCh:   make(chan struct{}),
		queueCh:      make(chan keybase1.UID, 100),
	}

	g.PushShutdownHook(func(mctx libkb.MetaContext) error {
		<-l.Shutdown(mctx.Ctx())
		return nil
	})
	return l
}

func (l *TrackerLoader) debug(ctx context.Context, msg string, args ...any) {
	l.G().Log.CDebugf(ctx, "TrackerLoader: %s", fmt.Sprintf(msg, args...))
}

func (l *TrackerLoader) Run(ctx context.Context) {
	defer l.G().CTrace(ctx, "TrackerLoader.Run", nil)()
	l.Lock()
	defer l.Unlock()
	if l.started {
		return
	}
	l.started = true
	l.shutdownCh = make(chan struct{})
	lctx, lcancel := context.WithCancel(ctx)
	l.cancel = lcancel
	l.eg.Go(func() error { return l.loadLoop(lctx, l.shutdownCh) })
}

func (l *TrackerLoader) Shutdown(ctx context.Context) chan struct{} {
	defer l.G().CTrace(ctx, "TrackerLoader.Shutdown", nil)()
	l.Lock()
	defer l.Unlock()
	ch := make(chan struct{})
	if l.started {
		l.cancel()
		close(l.shutdownCh)
		l.started = false
		go func() {
			_ = l.eg.Wait()
			close(ch)
		}()
	} else {
		close(ch)
	}
	return ch
}

func (l *TrackerLoader) Queue(ctx context.Context, uid keybase1.UID) (err error) {
	defer l.G().CTrace(ctx, fmt.Sprintf("TrackerLoader.Queue: uid: %s", uid), &err)()
	select {
	case l.queueCh <- uid:
	default:
		return errors.New("queue full")
	}
	return nil
}

var cachedOnlyStalenessWindow = time.Hour * 24 * 7

func (l *TrackerLoader) trackingArg(mctx libkb.MetaContext, uid keybase1.UID, withNetwork bool) *engine.ListTrackingEngineArg {
	arg := &engine.ListTrackingEngineArg{UID: uid, CachedOnly: !withNetwork}
	if !withNetwork {
		arg.CachedOnlyStalenessWindow = &cachedOnlyStalenessWindow
	}
	return arg
}

func (l *TrackerLoader) trackersArg(uid keybase1.UID, withNetwork bool) engine.ListTrackersUnverifiedEngineArg {
	return engine.ListTrackersUnverifiedEngineArg{UID: uid, CachedOnly: !withNetwork}
}

func (l *TrackerLoader) loadInner(mctx libkb.MetaContext, uid keybase1.UID, withNetwork bool) error {
	eng := engine.NewListTrackingEngine(mctx.G(), l.trackingArg(mctx, uid, withNetwork))
	err := engine.RunEngine2(mctx, eng)
	if err != nil {
		return err
	}
	following := eng.TableResult()

	eng2 := engine.NewListTrackersUnverifiedEngine(mctx.G(), l.trackersArg(uid, withNetwork))
	err = engine.RunEngine2(mctx, eng2)
	if err != nil {
		return err
	}
	followers := eng2.GetResults()

	l.debug(mctx.Ctx(), "loadInner: loaded %d followers, %d following", len(followers.Users), len(following.Users))
	l.G().NotifyRouter.HandleTrackingInfo(keybase1.TrackingInfoArg{
		Uid:       uid,
		Followees: following.Usernames(),
		Followers: followers.Usernames(),
	})

	return nil
}

func (l *TrackerLoader) load(ctx context.Context, uid keybase1.UID) error {
	defer l.G().CTrace(ctx, "TrackerLoader.load", nil)()

	mctx := libkb.NewMetaContext(ctx, l.G())

	withNetwork := false
	if err := l.loadInner(mctx, uid, withNetwork); err != nil {
		l.debug(ctx, "load: failed to load from local storage: %s", err)
	}

	withNetwork = true
	return l.loadInner(mctx, uid, withNetwork)
}

func (l *TrackerLoader) loadLoop(ctx context.Context, stopCh chan struct{}) error {
	for {
		select {
		case uid := <-l.queueCh:
			err := l.load(ctx, uid)
			if err != nil {
				l.debug(ctx, "loadLoop: failed to load: %s", err)
			} else {
				l.debug(ctx, "loadLoop: load tracks for %s successfully", uid)
			}
		case <-stopCh:
			return nil
		}
	}
}