storj/storagenode/storagenodedb/reputation.go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package storagenodedb

import (
	"context"

	"github.com/zeebo/errs"

	"storj.io/storj/pkg/storj"
	"storj.io/storj/storagenode/reputation"
)

// ErrReputation represents errors from the reputation database.
var ErrReputation = errs.Class("reputation error")

// reputation works with node reputation DB
type reputationDB struct {
	location string
	SQLDB
}

// newReputationDB returns a new instance of reputationDB initialized with the specified database.
func newReputationDB(db SQLDB, location string) *reputationDB {
	return &reputationDB{
		location: location,
		SQLDB:    db,
	}
}

// Store inserts or updates reputation stats into the db
func (db *reputationDB) Store(ctx context.Context, stats reputation.Stats) (err error) {
	defer mon.Task()(&ctx)(&err)

	query := `INSERT OR REPLACE INTO reputation (
			satellite_id, 
			uptime_success_count,
			uptime_total_count,
			uptime_reputation_alpha,
			uptime_reputation_beta,
			uptime_reputation_score,
			audit_success_count,
			audit_total_count,
			audit_reputation_alpha,
			audit_reputation_beta,
			audit_reputation_score,
			disqualified,
			updated_at
		) VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?)`

	// ensure we insert utc
	if stats.Disqualified != nil {
		utc := stats.Disqualified.UTC()
		stats.Disqualified = &utc
	}

	_, err = db.ExecContext(ctx, query,
		stats.SatelliteID,
		stats.Uptime.SuccessCount,
		stats.Uptime.TotalCount,
		stats.Uptime.Alpha,
		stats.Uptime.Beta,
		stats.Uptime.Score,
		stats.Audit.SuccessCount,
		stats.Audit.TotalCount,
		stats.Audit.Alpha,
		stats.Audit.Beta,
		stats.Audit.Score,
		stats.Disqualified,
		stats.UpdatedAt.UTC(),
	)

	return ErrReputation.Wrap(err)
}

// Get retrieves stats for specific satellite
func (db *reputationDB) Get(ctx context.Context, satelliteID storj.NodeID) (_ *reputation.Stats, err error) {
	defer mon.Task()(&ctx)(&err)

	var stats reputation.Stats

	row := db.QueryRowContext(ctx,
		`SELECT satellite_id, 
			uptime_success_count,
			uptime_total_count,
			uptime_reputation_alpha,
			uptime_reputation_beta,
			uptime_reputation_score,
			audit_success_count,
			audit_total_count,
			audit_reputation_alpha,
			audit_reputation_beta,
			audit_reputation_score,
			disqualified,
			updated_at
		FROM reputation WHERE satellite_id = ?`,
		satelliteID,
	)

	err = row.Scan(&stats.SatelliteID,
		&stats.Uptime.SuccessCount,
		&stats.Uptime.TotalCount,
		&stats.Uptime.Alpha,
		&stats.Uptime.Beta,
		&stats.Uptime.Score,
		&stats.Audit.SuccessCount,
		&stats.Audit.TotalCount,
		&stats.Audit.Alpha,
		&stats.Audit.Beta,
		&stats.Audit.Score,
		&stats.Disqualified,
		&stats.UpdatedAt,
	)

	return &stats, ErrReputation.Wrap(err)
}

// All retrieves all stats from DB
func (db *reputationDB) All(ctx context.Context) (_ []reputation.Stats, err error) {
	defer mon.Task()(&ctx)(&err)

	query := `SELECT satellite_id, 
			uptime_success_count,
			uptime_total_count,
			uptime_reputation_alpha,
			uptime_reputation_beta,
			uptime_reputation_score,
			audit_success_count,
			audit_total_count,
			audit_reputation_alpha,
			audit_reputation_beta,
			audit_reputation_score,
			disqualified,
			updated_at
		FROM reputation`

	rows, err := db.QueryContext(ctx, query)
	if err != nil {
		return nil, err
	}

	defer func() { err = errs.Combine(err, rows.Close()) }()

	var statsList []reputation.Stats
	for rows.Next() {
		var stats reputation.Stats

		err := rows.Scan(&stats.SatelliteID,
			&stats.Uptime.SuccessCount,
			&stats.Uptime.TotalCount,
			&stats.Uptime.Alpha,
			&stats.Uptime.Beta,
			&stats.Uptime.Score,
			&stats.Audit.SuccessCount,
			&stats.Audit.TotalCount,
			&stats.Audit.Alpha,
			&stats.Audit.Beta,
			&stats.Audit.Score,
			&stats.Disqualified,
			&stats.UpdatedAt,
		)

		if err != nil {
			return nil, ErrReputation.Wrap(err)
		}

		statsList = append(statsList, stats)
	}

	return statsList, nil
}