shadowdaoandClaude Sonnet 5.5
962d708fdd
Rename root, api and sdk module paths (imports, go.mod, go_package, ldflags, build scripts, buf configs) and regenerate protobuf/gateway code with the project's pinned toolchain. Wire identifiers unchanged. api gets a local replace for sdk; root/api/sdk require v0.1.0. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
48 lines
1.8 KiB
Go
48 lines
1.8 KiB
Go
// Copyright IBM Corp. 2020, 2026
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package session
|
|
|
|
import (
|
|
"context"
|
|
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/errors"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/event"
|
|
)
|
|
|
|
// CloseOrphanedConnections closes the orphaned connections for a given worker.
|
|
func CloseOrphanedConnections(ctx context.Context, repo *ConnectionRepository, workerId string, connections []*Connection) ([]string, error) {
|
|
const op = "session.CloseOrphanedConnections"
|
|
if repo == nil {
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing connection repository")
|
|
}
|
|
if workerId == "" {
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing worker id")
|
|
}
|
|
connectionIds := []string{}
|
|
for _, c := range connections {
|
|
connectionIds = append(connectionIds, c.GetPublicId())
|
|
}
|
|
closed, err := repo.closeOrphanedConnections(ctx, workerId, connectionIds)
|
|
if len(closed) > 0 {
|
|
event.WriteSysEvent(ctx, op, "marked unclaimed connections as closed", "worker_id", workerId, "count", len(closed))
|
|
}
|
|
if err != nil {
|
|
return nil, errors.Wrap(ctx, err, op, errors.WithMsg("failed to close orphaned connections for worker %q", workerId))
|
|
}
|
|
return closed, nil
|
|
}
|
|
|
|
// UpdateConnectionBytesUpDown processes the connection statistics for a given worker.
|
|
func UpdateConnectionBytesUpDown(ctx context.Context, repo *ConnectionRepository, connections []*Connection) error {
|
|
const op = "session.UpdateConnectionBytesUpDown"
|
|
if repo == nil {
|
|
return errors.New(ctx, errors.InvalidParameter, op, "missing connection repository")
|
|
}
|
|
err := repo.updateBytesUpBytesDown(ctx, connections...)
|
|
if err != nil {
|
|
return errors.Wrap(ctx, err, op, errors.WithMsg("failed to update bytes up and down for worker connection statistics"))
|
|
}
|
|
return nil
|
|
}
|