forked from
tangled.org/core
Mirror of @tangled.org/core. Running on a Raspberry Pi Zero 2 (Please be gentle).
spindle: filter jetstream by wanted dids
Signed-off-by: oppiliappan <me@oppi.li>
+13
-6
+6
knotclient/events.go
+6
knotclient/events.go
···
150
150
}
151
151
152
152
func (c *EventConsumer) AddSource(ctx context.Context, s EventSource) {
153
+
// we are already listening to this source
154
+
if _, ok := c.cfg.Sources[s]; ok {
155
+
c.logger.Info("source already present", "source", s)
156
+
return
157
+
}
158
+
153
159
c.cfgMu.Lock()
154
160
c.cfg.Sources[s] = struct{}{}
155
161
c.wg.Add(1)
+7
-6
spindle/server.go
+7
-6
spindle/server.go
···
70
70
tangled.SpindleMemberNSID,
71
71
tangled.RepoNSID,
72
72
}
73
-
jc, err := jetstream.NewJetstreamClient(cfg.Server.JetstreamEndpoint, "spindle", collections, nil, logger, d, false, false)
73
+
jc, err := jetstream.NewJetstreamClient(cfg.Server.JetstreamEndpoint, "spindle", collections, nil, logger, d, true, true)
74
74
if err != nil {
75
75
return fmt.Errorf("failed to setup jetstream client: %w", err)
76
76
}
77
+
jc.AddDid(cfg.Server.Owner)
77
78
78
79
spindle := Spindle{
79
80
jc: jc,
···
119
118
ccfg.Dev = cfg.Server.Dev
120
119
ccfg.ProcessFunc = spindle.processPipeline
121
120
ccfg.CursorStore = cursorStore
122
-
knotstream := knotclient.NewEventConsumer(*ccfg)
123
121
knownKnots, err := d.Knots()
124
122
if err != nil {
125
123
return err
126
124
}
127
125
for _, knot := range knownKnots {
128
-
knotstream.AddSource(ctx, knotclient.NewEventSource(knot))
126
+
logger.Info("adding source start", "knot", knot)
127
+
ccfg.Sources[knotclient.EventSource{knot}] = struct{}{}
129
128
}
130
-
spindle.ks = knotstream
129
+
spindle.ks = knotclient.NewEventConsumer(*ccfg)
131
130
132
131
go func() {
133
-
logger.Info("starting knot event consumer", "knots")
134
-
knotstream.Start(ctx)
132
+
logger.Info("starting knot event consumer")
133
+
spindle.ks.Start(ctx)
135
134
}()
136
135
137
136
logger.Info("starting spindle server", "address", cfg.Server.ListenAddr)