267506bb20
metabase has become a central concept and it's more suitable for it to be directly nested under satellite rather than being part of metainfo. metainfo is going to be the "endpoint" logic for handling requests. Change-Id: I53770d6761ac1e9a1283b5aa68f471b21e784198
131 lines
3.4 KiB
Go
131 lines
3.4 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package metabase
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/common/storj"
|
|
"storj.io/storj/private/dbutil/pgutil"
|
|
"storj.io/storj/private/tagsql"
|
|
)
|
|
|
|
// NodeAlias is a metabase local alias for NodeID-s to reduce segment table size.
|
|
type NodeAlias int32
|
|
|
|
// NodeAliasEntry is a mapping between NodeID and NodeAlias.
|
|
type NodeAliasEntry struct {
|
|
ID storj.NodeID
|
|
Alias NodeAlias
|
|
}
|
|
|
|
// EnsureNodeAliases contains arguments necessary for creating NodeAlias-es.
|
|
type EnsureNodeAliases struct {
|
|
Nodes []storj.NodeID
|
|
}
|
|
|
|
// EnsureNodeAliases ensures that the supplied node ID-s have a alias.
|
|
// It's safe to concurrently try and create node ID-s for the same NodeID.
|
|
func (db *DB) EnsureNodeAliases(ctx context.Context, opts EnsureNodeAliases) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
for _, node := range opts.Nodes {
|
|
if node.IsZero() {
|
|
return Error.New("tried to add alias to zero node")
|
|
}
|
|
}
|
|
|
|
_, err = db.db.ExecContext(ctx, `
|
|
INSERT INTO node_aliases(node_id)
|
|
SELECT unnest($1::BYTEA[])
|
|
ON CONFLICT DO NOTHING
|
|
`, pgutil.NodeIDArray(opts.Nodes))
|
|
return Error.Wrap(err)
|
|
}
|
|
|
|
// ListNodeAliases lists all node alias mappings.
|
|
func (db *DB) ListNodeAliases(ctx context.Context) (_ []NodeAliasEntry, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
var aliases []NodeAliasEntry
|
|
rows, err := db.db.Query(ctx, `
|
|
SELECT node_id, node_alias
|
|
FROM node_aliases
|
|
`)
|
|
if err != nil {
|
|
return nil, Error.New("ListNodeAliases query: %w", err)
|
|
}
|
|
defer func() { err = errs.Combine(err, rows.Close()) }()
|
|
|
|
for rows.Next() {
|
|
var entry NodeAliasEntry
|
|
err := rows.Scan(&entry.ID, &entry.Alias)
|
|
if err != nil {
|
|
return nil, Error.New("ListNodeAliases scan failed: %w", err)
|
|
}
|
|
aliases = append(aliases, entry)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, Error.New("ListNodeAliases scan failed: %w", err)
|
|
}
|
|
|
|
return aliases, nil
|
|
}
|
|
|
|
// txNodeAliases is used inside a migration.
|
|
// This will be removed once the migration has been completed.
|
|
type txNodeAliases struct {
|
|
db tagsql.Tx
|
|
}
|
|
|
|
// EnsureNodeAliases ensures that the supplied node ID-s have a alias.
|
|
// It's safe to concurrently try and create node ID-s for the same NodeID.
|
|
func (db *txNodeAliases) EnsureNodeAliases(ctx context.Context, opts EnsureNodeAliases) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
for _, node := range opts.Nodes {
|
|
if node.IsZero() {
|
|
return Error.New("tried to add alias to zero node")
|
|
}
|
|
}
|
|
|
|
_, err = db.db.ExecContext(ctx, `
|
|
INSERT INTO node_aliases(node_id)
|
|
SELECT unnest($1::BYTEA[])
|
|
ON CONFLICT DO NOTHING
|
|
`, pgutil.NodeIDArray(opts.Nodes))
|
|
return Error.Wrap(err)
|
|
}
|
|
|
|
// ListNodeAliases lists all node alias mappings.
|
|
func (db *txNodeAliases) ListNodeAliases(ctx context.Context) (_ []NodeAliasEntry, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
var aliases []NodeAliasEntry
|
|
rows, err := db.db.Query(ctx, `
|
|
SELECT node_id, node_alias
|
|
FROM node_aliases
|
|
`)
|
|
if err != nil {
|
|
return nil, Error.New("ListNodeAliases query: %w", err)
|
|
}
|
|
defer func() { err = errs.Combine(err, rows.Close()) }()
|
|
|
|
for rows.Next() {
|
|
var entry NodeAliasEntry
|
|
err := rows.Scan(&entry.ID, &entry.Alias)
|
|
if err != nil {
|
|
return nil, Error.New("ListNodeAliases scan failed: %w", err)
|
|
}
|
|
aliases = append(aliases, entry)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, Error.New("ListNodeAliases scan failed: %w", err)
|
|
}
|
|
|
|
return aliases, nil
|
|
}
|