Skip to content

Commit b24b3ff

Browse files
committed
Remove dead updatePoloScores, add nameserver cleanup
- Remove unused Daemon.updatePoloScores (replaced by inline logic in handleTaskResults) - Add defer ns.Close() to all nameserver integration tests
1 parent df69aea commit b24b3ff

2 files changed

Lines changed: 6 additions & 33 deletions

File tree

pkg/daemon/services.go

Lines changed: 0 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -1113,36 +1113,3 @@ func (d *Daemon) handleTaskResults(adapter *connAdapter, conn *Connection, frame
11131113
}
11141114
}
11151115

1116-
// updatePoloScores updates polo scores for task processing.
1117-
func (d *Daemon) updatePoloScores(receiverNode, submitterNode uint32) {
1118-
if d.regConn == nil {
1119-
slog.Warn("tasksubmit: cannot update polo score, no registry connection")
1120-
return
1121-
}
1122-
1123-
// Receiver gets +1 polo score
1124-
if _, err := d.regConn.UpdatePoloScore(receiverNode, 1); err != nil {
1125-
slog.Warn("tasksubmit: failed to update receiver polo score",
1126-
"node_id", receiverNode,
1127-
"error", err,
1128-
)
1129-
} else {
1130-
slog.Info("tasksubmit: polo score updated",
1131-
"node_id", receiverNode,
1132-
"delta", 1,
1133-
)
1134-
}
1135-
1136-
// Submitter gets -1 polo score
1137-
if _, err := d.regConn.UpdatePoloScore(submitterNode, -1); err != nil {
1138-
slog.Warn("tasksubmit: failed to update submitter polo score",
1139-
"node_id", submitterNode,
1140-
"error", err,
1141-
)
1142-
} else {
1143-
slog.Info("tasksubmit: polo score updated",
1144-
"node_id", submitterNode,
1145-
"delta", -1,
1146-
)
1147-
}
1148-
}

tests/nameserver_test.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ func TestNameserver(t *testing.T) {
3535
ns := nameserver.New(a.Driver, "")
3636
go ns.ListenAndServe()
3737
waitNSReady(t, ns)
38+
defer ns.Close()
3839

3940
// Pre-register some records directly
4041
ns.Store().RegisterA("agent-alpha", a.Daemon.Addr())
@@ -114,6 +115,7 @@ func TestNameserverSRecord(t *testing.T) {
114115
ns := nameserver.New(a.Driver, "")
115116
go ns.ListenAndServe()
116117
waitNSReady(t, ns)
118+
defer ns.Close()
117119

118120
client := nameserver.NewClient(b.Driver, a.Daemon.Addr())
119121

@@ -157,6 +159,7 @@ func TestNameserverRegisterN(t *testing.T) {
157159
ns := nameserver.New(a.Driver, "")
158160
go ns.ListenAndServe()
159161
waitNSReady(t, ns)
162+
defer ns.Close()
160163

161164
client := nameserver.NewClient(b.Driver, a.Daemon.Addr())
162165

@@ -188,6 +191,7 @@ func TestNameserverOverwriteA(t *testing.T) {
188191
ns := nameserver.New(a.Driver, "")
189192
go ns.ListenAndServe()
190193
waitNSReady(t, ns)
194+
defer ns.Close()
191195

192196
client := nameserver.NewClient(b.Driver, a.Daemon.Addr())
193197

@@ -228,6 +232,7 @@ func TestNameserverPersistence(t *testing.T) {
228232
ns1 := nameserver.New(a.Driver, storePath)
229233
go ns1.ListenAndServe()
230234
waitNSReady(t, ns1)
235+
defer ns1.Close()
231236

232237
client := nameserver.NewClient(b.Driver, a.Daemon.Addr())
233238

@@ -280,6 +285,7 @@ func TestNameserverMultipleClients(t *testing.T) {
280285
ns := nameserver.New(a.Driver, "")
281286
go ns.ListenAndServe()
282287
waitNSReady(t, ns)
288+
defer ns.Close()
283289

284290
// Pre-register
285291
ns.Store().RegisterA("target", a.Daemon.Addr())

0 commit comments

Comments
 (0)