#!/usr/bin/python

# -*- coding: utf-8 -*-

import time

import socket

from locust import Locust, TaskSet, events, task, seq_task,TaskSequence

import login

import mylib

import sys

import struct

#cmd = "PlayerVerifyVerLoginClientCmd_C"

#cmd = "PlayerVerifyVerLoginClientCmd_S"

#print(Command_py.Command_pb2.CmdNumber.Name(1))

#print(Command_py.Command_pb2.CmdNumber.Value('ClientCmdNone'))

#print(Command_py.Command_pb2.CmdNumber.Value(cmd))

#print(cmd.split('_')[0])

#print(cmd)

defaultencoding = 'utf-8'

if sys.getdefaultencoding() != defaultencoding:

reload(sys)

sys.setdefaultencoding(defaultencoding)

class TcpSocketClient(socket.socket):

def __init__(self, af_inet, socket_type):

super(TcpSocketClient, self).__init__(af_inet, socket_type)

def connect(self, addr):

start_time = time.time()

try:

super(TcpSocketClient, self).connect(addr)

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 close(self):

start_time = time.time()

try:

super(TcpSocketClient, self).close()

except Exception as e:

total_time = int((time.time() - start_time) * 1000)

events.request_failure.fire(request_type="tcpsocket", name="close", response_time=total_time, exception=e)

else:

total_time = int((time.time() - start_time) * 1000)

events.request_success.fire(request_type="tcpsocket", name="close", response_time=total_time,response_length=0)

def send(self, msg):

start_time = time.time()

try:

super(TcpSocketClient, self).send(msg)

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, bufsize):

recv_data = ''

start_time = time.time()

try:

recv_data = super(TcpSocketClient, self).recv(bufsize)

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 recv_data

def setlogininfo(self,account_id,login_id,version):

self.account_id = account_id

self.login_id = login_id

self.version = version

def getlogininfo(self):

return self.account_id,self.login_id,self.version

class VerifyVerLogin(TaskSequence):

def on_start(self):

print("------start------")

@seq_task(1)

def test1(self):

print("-------1-------")

@seq_task(2)

def test2(self):

print("-------2-------")

@seq_task(3)

@task

def test3(self):

print("-------3------")

@seq_task(3)

@task

def test4(self):

print("-------4-------")

@seq_task(3)

@task

def test5(self):

print("-------5-------")

class PlayerVerifyVerLogin(Locust):

min_wait = 100

max_wait = 1000

task_set = VerifyVerLogin

if __name__ == "__main__":

user = PlayerVerifyVerLogin()

user.run()