storj/examples/grpc-debug/main.go

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

package main

import (
	"context"
	"fmt"

	"github.com/spf13/pflag"
	"google.golang.org/grpc"

	"storj.io/storj/pkg/cfgstruct"
	"storj.io/storj/pkg/identity"
	"storj.io/storj/pkg/peertls/tlsopts"
	"storj.io/storj/pkg/storj"
)

var (
	targetAddr = pflag.String("target", "satellite.staging.storj.io:7777", "address of target")

	identityConfig identity.Config
)

func init() {
	cfgstruct.Bind(pflag.CommandLine, &identityConfig, cfgstruct.ConfDir("$HOME/.storj/gw"))
}

func main() {
	ctx := context.Background()
	pflag.Parse()
	identity, err := identityConfig.Load()
	if err != nil {
		panic(err)
	}
	clientOptions, err := tlsopts.NewOptions(identity, tlsopts.Config{})
	if err != nil {
		panic(err)
	}

	dialOption := clientOptions.DialOption(storj.NodeID{})

	conn, err := grpc.Dial(*targetAddr, dialOption)
	if err != nil {
		panic(err)
	}
	fmt.Println(conn.GetState())
	err = conn.Invoke(ctx, "NonExistentMethod", nil, nil)
	if err != nil && err.Error() != `rpc error: code = ResourceExhausted desc = malformed method name: "NonExistentMethod"` {
		fmt.Println(err)
	}
	fmt.Println(conn.GetState())
	err = conn.Close()
	if err != nil {
		fmt.Println(err)
	}
}