65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package psclient
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"storj.io/storj/pkg/pb"
|
||
|
|
"storj.io/storj/pkg/provider"
|
||
|
|
"storj.io/storj/pkg/transport"
|
||
|
|
)
|
||
|
|
|
||
|
|
// LiteClient is the lightweight client for getting stats
|
||
|
|
type LiteClient interface {
|
||
|
|
Stats(ctx context.Context) (*pb.StatSummary, error)
|
||
|
|
Dashboard(ctx context.Context) (pb.PieceStoreRoutes_DashboardClient, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// PieceStoreLite is the struct that holds the client
|
||
|
|
type PieceStoreLite struct {
|
||
|
|
client pb.PieceStoreRoutesClient
|
||
|
|
}
|
||
|
|
|
||
|
|
// Dashboard returns a simple terminal dashboard displaying info
|
||
|
|
func (psl *PieceStoreLite) Dashboard(ctx context.Context) (pb.PieceStoreRoutes_DashboardClient, error) {
|
||
|
|
return psl.client.Dashboard(ctx, &pb.DashboardReq{})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Stats will retrieve stats about a piece storage node
|
||
|
|
func (psl *PieceStoreLite) Stats(ctx context.Context) (*pb.StatSummary, error) {
|
||
|
|
return psl.client.Stats(ctx, &pb.StatsReq{})
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewLiteClient returns a new LiteClient
|
||
|
|
func NewLiteClient(ctx context.Context, addr string) (LiteClient, error) {
|
||
|
|
clientIdent, err := provider.NewFullIdentity(ctx, 12, 4)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// address of node to create client connection
|
||
|
|
if addr == "" {
|
||
|
|
addr = ":7777"
|
||
|
|
}
|
||
|
|
|
||
|
|
tc := transport.NewClient(clientIdent)
|
||
|
|
n := &pb.Node{
|
||
|
|
Address: &pb.NodeAddress{
|
||
|
|
Address: addr,
|
||
|
|
Transport: 0,
|
||
|
|
},
|
||
|
|
Type: pb.NodeType_STORAGE,
|
||
|
|
}
|
||
|
|
|
||
|
|
conn, err := tc.DialNode(ctx, n)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &PieceStoreLite{
|
||
|
|
client: pb.NewPieceStoreRoutesClient(conn),
|
||
|
|
}, nil
|
||
|
|
}
|