26 lines
503 B
Go
26 lines
503 B
Go
package rpc
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
_ "github.com/iwind/TeaGo/bootstrap"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestRPC_Dial(t *testing.T) {
|
|
client, err := SharedRPC()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var before = time.Now()
|
|
for i := 0; i < 1_000_000; i++ {
|
|
_ = client.pickConn()
|
|
}
|
|
_, err = client.NSNodeRPC.FindCurrentNSNodeConfig(client.Context(), &pb.FindCurrentNSNodeConfigRequest{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(time.Since(before).Seconds()*1000, "ms")
|
|
}
|