k6 grpc（基于wsl2环境）

//起个服务做测试
package server

import (
	"context"
	"fmt"
	"log"
	"net"

	pb "golearn/grpc/helloworld" // 替换为你的模块路径

	"google.golang.org/grpc"
	"google.golang.org/grpc/reflection"
)

const (
	port = ":50051"
)

// server 是用于实现 helloworld.Greeter 服务的服务器。
type server struct {
	pb.UnimplementedGreeterServer
}

// SayHello 实现了 Greeter 服务中的 SayHello 方法。
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
	log.Printf("Received: %v", in.GetName())
	return &pb.HelloReply{Message: "返回给你: " + in.GetName()}, nil
}

func Run() {
	lis, err := net.Listen("tcp", port)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}
	s := grpc.NewServer()
	pb.RegisterGreeterServer(s, &server{})

	// 在 gRPC 服务器上注册 reflection 服务.
	reflection.Register(s)

	fmt.Printf("Server listening at %v\n", lis.Addr())
	if err := s.Serve(lis); err != nil {
		log.Fatalf("failed to serve: %v", err)
	}
}

//helloworld.proto
syntax = "proto3";

package helloworld;
option go_package = "helloworld/helloworld";
service Greeter {
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}

//cilent.js
const client = createClient();
client.load(['./grpc'], 'helloworld.proto');
function createClient() {
    return new grpc.Client();
}

export function client_test (message='none message'){
  client.connect('192.168.31.179:50051',{plaintext: true, timeout: '5s' });
  
  console.log("connect success")
  const data = {'name': message};
  const response = client.invoke('helloworld.Greeter/SayHello', data);

  check(response, {
    'status is OK': (r) => r && r.status === grpc.StatusOK,
  });
  

  console.log(JSON.stringify(response.message));

  client.close();
  sleep(1);
};

一些注意

• client和load是要在init中的，因此不能写在function或者setup/teardown中。避免在 VU 运行期间发生文件 I/O 操作，从而提高性能和稳定性
• 尝试将client和load放在main中，尝试传参client又总是报错object，似乎未能正确识别
• 当模拟更多连接，似乎应该每个VU一个连接；不过服务端通常会有连接池概念。
• AI说每个VU多个迭代会共享一个连接池，但是我未证实，权且相信。需要进一步的监控测试才能确认。实际压测估计得调试调试。
• 本地机子太容易端口不够报错了，wsl2中用docker，几千都没啥问题。