zoukankan      html  css  js  c++  java
  • ngrinder 负载均衡脚本开发

    config.py

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

    cookies = []
    userids = []
    infile = open("./resources/test_auth_1W_1.txt", 'rb')
    for line in infile.readlines():
    userids.append(line.strip().split(",")[0])
    cookies.append(line.strip().split(",")[1])
    infile.close()


    putongGoodsIds = []
    infile = open("./resources/goods_putong1.txt", 'rb')
    for line in infile.readlines():
    putongGoodsIds.append(line.strip().split(",")[1])
    infile.close()

    putong2GoodsIds = []
    infile = open("./resources/goods_putong2.txt", 'rb')
    for line in infile.readlines():
    putong2GoodsIds.append(line.strip().split(",")[1])
    infile.close()


    # gateway 网关负载均衡 30台机器
    gateways = ["192.168.8.126:8080","192.168.8.127:8080","192.168.8.128:8080","192.168.102.6:8080","192.168.102.7:8080","192.168.102.8:8080","192.168.102.9:8080","192.168.102.10:8080","192.168.102.11:8080","192.168.102.12:8080","192.168.102.13:8080","192.168.102.14:8080","192.168.102.15:8080","192.168.102.16:8080","192.168.102.17:8080","192.168.102.18:8080","192.168.102.19:8080","192.168.102.20:8080","192.168.102.21:8080","192.168.102.22:8080","192.168.104.97:8080","192.168.104.98:8080","192.168.104.117:8080","192.168.104.139:8080","192.168.104.140:8080","192.168.104.141:8080","192.168.104.142:8080","192.168.104.143:8080","192.168.104.144:8080","192.168.104.145:8080"]

    # -*- coding: utf-8 -*-
    '''
    下单链路:确认订单、生成订单

    选取店铺 3381 3380 下面的商品(每个店铺160个普通商品,4种店铺优惠)
    下单链路_普通会员
    '''

    from net.grinder.script import Test
    from net.grinder.script.Grinder import grinder
    from HTTPClient import NVPair
    from net.grinder.plugin.http import HTTPRequest
    from net.grinder.plugin.http import HTTPPluginControl
    from java.util import Random
    from org.slf4j import LoggerFactory
    from ch.qos.logback.classic import Level
    from com.alibaba.fastjson import JSON
    from Config import cookies, gateways, putongGoodsIds, putong2GoodsIds

    case_no_1 = 1
    case_name_1 = "api_buy_confirmOrder"
    case_no_2 = 2
    case_name_2 = "api_buy_createOrder"

    connectionDefaults = HTTPPluginControl.getConnectionDefaults()
    connectionDefaults.timeout = 60000 # 超时时间设置 60秒
    httpUtilities = HTTPPluginControl.getHTTPUtilities()
    HTTPPluginControl.getConnectionDefaults().useCookies = 0

    totalAgents = int(grinder.getProperties().get('grinder.agents'))
    totalProcess = int(grinder.getProperties().get('grinder.processes'))
    totalThreads = int(grinder.getProperties().get('grinder.threads'))

    class TestRunner:
    def __init__(self):
    logger = LoggerFactory.getLogger("worker");
    logger.setLevel(Level.ERROR);
    grinder.statistics.delayReports=True
    pass

    def getIncrementId(self, runNumber):
    agentNum = grinder.agentNumber
    processNum = grinder.processNumber
    threadNum = grinder.threadNumber
    return agentNum * totalProcess * totalThreads + processNum * totalThreads + threadNum + totalAgents * totalProcess * totalThreads * runNumber

    def api_buy_confirmOrder(self, url, body, header):
    result = HTTPRequest(url=url, headers=header).POST(url, body)
    data = result.getText().encode('utf-8')
    #print data # 调试使用,压测时关闭
    try:
    addressId = JSON.parseObject(data).getJSONObject("data").getJSONObject("address").getString("addressId")
    idCardId = JSON.parseObject(data).getJSONObject("data").getJSONObject("identity").getString("idCardId")
    # 商品1的优惠信息-普通品
    promotionKey_putong_1 = JSON.parseObject(data).getJSONObject("data").getJSONArray("shopOrderList")[0]["selectedPromotionList"][0]["promotionKey"]
    SINGLE_ITEM_putong_1 = JSON.parseObject(data).getJSONObject("data").getJSONArray("shopOrderList")[0]["itemOrderList"][0]["selectedPromotionList"][0]["promotionKey"]
    logisticsCode_putong_1 = JSON.parseObject(data).getJSONObject("data").getJSONArray("shopOrderList")[0]["selectedLogistics"]["logisticsCode"]
    # 商品2的优惠信息-普通品
    MEMBER_PRICE_putong_2 = JSON.parseObject(data).getJSONObject("data").getJSONArray("shopOrderList")[1]["itemOrderList"][0]["selectedPromotionList"][0]["promotionKey"]
    logisticsCode_putong_2 = JSON.parseObject(data).getJSONObject("data").getJSONArray("shopOrderList")[1]["selectedLogistics"]["logisticsCode"]
    except:
    grinder.logger.error(case_name_1 + ':' + body + ':' + data.decode('utf-8'))

    if(result.getStatusCode() != 200 or data.find('"status":0') != -1 or data.find('"code":0') == -1):
    grinder.getStatistics().getForCurrentTest().setSuccess(False)
    grinder.logger.error(case_name_1 + ':' + body + ':' + data.decode('utf-8'))

    return addressId, idCardId, promotionKey_putong_1, SINGLE_ITEM_putong_1, logisticsCode_putong_1, MEMBER_PRICE_putong_2, logisticsCode_putong_2

    def api_buy_createOrder(self, url, body, header):
    result = HTTPRequest(url=url, headers=header).POST(url, body)
    data = result.getText().encode('utf-8')
    #print data # 调试使用,压测时关闭

    if(result.getStatusCode() != 200 or data.find('"status":0') != -1 or data.find('"code":0') == -1):
    grinder.getStatistics().getForCurrentTest().setSuccess(False)
    grinder.logger.error(case_name_2 + ':' + body + ':' + data.decode('utf-8'))

    request_1 = Test(case_no_1, case_name_1).wrap(api_buy_confirmOrder)
    request_2 = Test(case_no_2, case_name_2).wrap(api_buy_createOrder)

    def __call__(self):
    index_num = self.getIncrementId(grinder.runNumber)
    index = index_num % len(cookies)
    index_1 = index_num % len(putongGoodsIds)
    index_2 = index_num % len(putong2GoodsIds)

    header = [
    NVPair('Content-Type', 'application/x-www-form-urlencoded'),
    NVPair('cookie', cookies[index])
    ]

    url_1 = 'http://' + gateways[index_num % 30] + '/buy/api/buy/confirmOrder'
    body_1 = 'createOrderBizContent={"extensions":{"buySource":"itemDetail"},"addressId":"-1","createOrderShopGroupList":[{"logisticsCode":"","createOrderSkuGroupList":[{"extensions":{},"quantity":"1","selectedPromotionList":[],"skuId":"%s"},{"extensions":{},"quantity":1,"selectedPromotionList":[],"skuId":"%s"}],"comment":"perf_test","selectedPromotionList":[]}],"idCardId":"-1","selectedPromotionList":[],"coinToUse":"-1"}' %(putongGoodsIds[index_1], putong2GoodsIds[index_2])
    addressId, idCardId, promotionKey_putong_1, SINGLE_ITEM_putong_1, logisticsCode_putong_1, MEMBER_PRICE_putong_2, logisticsCode_putong_2 = self.request_1(url_1,body_1,header)

    url_2 = 'http://' + gateways[index_num % 30] + '/buy/api/buy/createOrder'
    body_2 = 'createOrderBizContent={"idCardId":%s,"addressId":%s,"selectedPromotionList":[],"extensions":{"buySource":"cart"},"createOrderShopGroupList":[{"comment":"","extensions":{},"logisticsCode":"%s","selectedPromotionList":["%s"],"createOrderSkuGroupList":[{"extensions":{},"selectedPromotionList":["%s"],"skuId":%s,"quantity":1}]},{"comment":"","extensions":{},"logisticsCode":"%s","selectedPromotionList":[],"createOrderSkuGroupList":[{"extensions":{},"selectedPromotionList":["%s"],"skuId":%s,"quantity":1}]}]}' %(idCardId, addressId, logisticsCode_putong_1, promotionKey_putong_1, SINGLE_ITEM_putong_1, putongGoodsIds[index_1], logisticsCode_putong_2, MEMBER_PRICE_putong_2, putong2GoodsIds[index_2])
    self.request_2(url_2,body_2,header)

  • 相关阅读:
    ubuntu python3 安装pip
    Windows远程桌面连接ubuntu 16
    Python 高级编程——单例模式
    学习资料推荐
    经典 测试开发面试题 (随时更新)
    mac上生成go文件失败报错,gRpc-- protoc-gen-go: program not found or is not executable
    小白从零开始学编程(五)--python数据类型--字符串
    小白从零开始学编程(三)--python基本概念
    小白从零开始学编程(二)--python虚拟环境和编辑器
    小白从零开始学编程--python安装与环境搭建
  • 原文地址:https://www.cnblogs.com/ceshi2016/p/13268221.html
Copyright © 2011-2022 走看看