如何在locust中调用GRPC python客户端？

pip install locust grpcio-tools

import grpc
import your_pb2
import your_pb2_grpc

class YourClient:
    def __init__(self):
        channel = grpc.insecure_channel('your_grpc_server_address')
        self.stub = your_pb2_grpc.YourServiceStub(channel)

    def call_your_method(self, request_data):
        response = self.stub.YourMethod(your_pb2.YourRequest(data=request_data))
        return response.data

from locust import HttpUser, task
from your_grpc_client import YourClient

class GrpcUser(HttpUser):
    def on_start(self):
        self.grpc_client = YourClient()

    @task
    def call_grpc_method(self):
        request_data = "your_request_data"
        response = self.grpc_client.call_your_method(request_data)
        print(response)

locust -f your_locust_file.py --users 10 --spawn-rate 1