Extended socket protocol

[oleeks]

Hello,
How to expand the socket protocol? There are lots of connection errors in the following ways.
thaks.

from locust import Locust, events, task, TaskSet
import socket
import time

class TcpSocketClient(socket.socket):
    def __init__(self, af_inet, socket_type):
        super(TcpSocketClient, self).__init__(af_inet, socket_type)

    def connect(self, address):
        start_time = time.time()
        try:
            super(TcpSocketClient, self).connect(address)
        except Exception as e:
            total_time = int((time.time() - start_time) * 1000)
            events.request_failure.fire(request_type="tcpsocket", name="connect", response_time=total_time, exception=e)
        else:
            total_time = int((time.time() - start_time) * 1000)
            events.request_success.fire(request_type="tcpsocket", name="connect", response_time=total_time,
                                        response_length=0)

    def send(self, data, flags=None):
        start_time = time.time()
        try:
            super(TcpSocketClient, self).send(data)
        except Exception as e:
            total_time = int((time.time() - start_time) * 1000)
            events.request_failure.fire(request_type="tcpsocket", name="send", response_time=total_time, exception=e)
        else:
            total_time = int((time.time() - start_time) * 1000)
            events.request_success.fire(request_type="tcpsocket", name="send", response_time=total_time,
                                        response_length=0)

    def recv(self, buffersize, flags=None):
        start_time = time.time()
        try:
            r = super(TcpSocketClient, self).recv(buffersize)
        except Exception as e:
            total_time = int((time.time() - start_time) * 1000)
            events.request_failure.fire(request_type="tcpsocket", name="recv", response_time=total_time, exception=e)
        else:
            total_time = int((time.time() - start_time) * 1000)
            events.request_success.fire(request_type="tcpsocket", name="recv", response_time=total_time,
                                        response_length=0)
            return r


class TcpSocketLocust(Locust):
    """
    This is the abstract Locust class which should be subclassed. It provides an TCP socket client
    that can be used to make TCP socket requests that will be tracked in Locust's statistics.
    """
    def __init__(self):
        super(TcpSocketLocust, self).__init__()
        self.client = TcpSocketClient(socket.AF_INET, socket.SOCK_STREAM)



class TcpTestUser(TcpSocketLocust):
    min_wait = 100
    max_wait = 1000

    class task_set(TaskSet):
        @task
        def send_data(self):
            ADDR = ("127.0.0.1", 8000)
            self.client.connect(ADDR)
            send_data = 'abcd'.encode('utf-8')
            self.client.send(send_data)
            data = self.client.recv(1024).decode('utf-8')
            print(data)

Environment settings (for bug reports)

• OS: win10
• Python version:3.6.4
• Locust version:0.8.1

Steps to reproduce (for bug reports)

TBD - example code appreciated

[aldenpeterson-wf]

@guifa2015 do you still need help with this issue?

[oleeks]

ÿes

…

On Sat, Jul 14, 2018 at 6:17 AM aldenpeterson-wf ***@***.***> wrote: @guifa2015 <https://github.com/guifa2015> do you still need help with this issue? — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#776 (comment)>, or mute the thread <https://github.com/notifications/unsubscribe-auth/ANSOazv_Mv9y6TEC8YwNlnnstPs_VSEoks5uGRyIgaJpZM4TYhGl> .

[cgoldberg]

closing for lack of detail.

If you found an issue with Locust, please submit a new report containing a clear description of the problem.
Make sure to include a concise test case that reproduces the issue, including expected and actual results.