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>
82 lines
3.2 KiB
Go
82 lines
3.2 KiB
Go
// Copyright IBM Corp. 2020, 2026
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package static
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/globals"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/errors"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/host"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/listtoken"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/pagination"
|
|
"repo.anhonesthost.net/CyberCoveLLC/threshold/internal/types/resource"
|
|
)
|
|
|
|
// ListHostsRefresh lists hosts according to the page size
|
|
// and list token, filtering out entries that do not
|
|
// pass the filter item fn. It returns a new list token
|
|
// based on the old one, the grants hash, and the returned
|
|
// hosts. It also returns the plugin associated with the host catalog.
|
|
func ListHostsRefresh(
|
|
ctx context.Context,
|
|
grantsHash []byte,
|
|
pageSize int,
|
|
filterItemFn pagination.ListFilterFunc[host.Host],
|
|
tok *listtoken.Token,
|
|
repo *Repository,
|
|
hostCatalogId string,
|
|
) (*pagination.ListResponse[host.Host], error) {
|
|
const op = "plugin.ListHostsRefresh"
|
|
|
|
switch {
|
|
case len(grantsHash) == 0:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing grants hash")
|
|
case pageSize < 1:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "page size must be at least 1")
|
|
case filterItemFn == nil:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing filter item callback")
|
|
case tok == nil:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing token")
|
|
case repo == nil:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing repo")
|
|
case hostCatalogId == "":
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "missing host catalog ID")
|
|
case tok.ResourceType != resource.Host:
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "token did not have a host resource type")
|
|
}
|
|
rt, ok := tok.Subtype.(*listtoken.StartRefreshToken)
|
|
if !ok {
|
|
return nil, errors.New(ctx, errors.InvalidParameter, op, "token did not have a start-refresh token component")
|
|
}
|
|
|
|
listItemsFn := func(ctx context.Context, lastPageItem host.Host, limit int) ([]host.Host, time.Time, error) {
|
|
opts := []Option{
|
|
WithLimit(limit),
|
|
}
|
|
if lastPageItem != nil {
|
|
opts = append(opts, WithStartPageAfterItem(lastPageItem))
|
|
}
|
|
// Add the database read timeout to account for any creations missed due to concurrent
|
|
// transactions in the initial pagination phase.
|
|
sHosts, listTime, err := repo.listHostsRefresh(ctx, hostCatalogId, rt.PreviousPhaseUpperBound.Add(-globals.RefreshReadLookbackDuration), opts...)
|
|
if err != nil {
|
|
return nil, time.Time{}, err
|
|
}
|
|
var hosts []host.Host
|
|
for _, host := range sHosts {
|
|
hosts = append(hosts, host)
|
|
}
|
|
return hosts, listTime, nil
|
|
}
|
|
listDeletedIdsFn := func(ctx context.Context, since time.Time) ([]string, time.Time, error) {
|
|
// Add the database read timeout to account for any deletes missed due to concurrent
|
|
// transactions in the original list pagination phase.
|
|
return repo.listDeletedHostIds(ctx, since.Add(-globals.RefreshReadLookbackDuration))
|
|
}
|
|
|
|
return pagination.ListRefresh(ctx, grantsHash, pageSize, filterItemFn, listItemsFn, repo.estimatedHostCount, listDeletedIdsFn, tok)
|
|
}
|