GoReplay的middlware原理

中間件是一個(gè)程序，它在 STDIN 上接受請(qǐng)求和響應(yīng)負(fù)載，并在 STDOUT 上發(fā)出修改后的請(qǐng)求。您可以實(shí)現(xiàn)任何自定義邏輯，例如剝離私有數(shù)據(jù)、高級(jí)重寫、支持 oAuth 等。

中間件可以用任何語言編寫，請(qǐng)參閱examples/middleware文件夾中的示例。
中間件程序應(yīng)該接受這樣一個(gè)事實(shí)：所有與 Gor 的通信都是異步的，不能保證原始請(qǐng)求和響應(yīng)消息會(huì)相繼到來。如果邏輯依賴于原始或重播的響應(yīng)，您的應(yīng)用程序應(yīng)該處理狀態(tài)，請(qǐng)參見examples/middleware/token_modifier.go示例。

python版本環(huán)境搭建

python環(huán)境安裝

首先安裝python環(huán)境，由于版本 v0.2.x 不再支持 Python2.7 和 Python3.4，因此支持的最低 Python 版本為 3.5.2。此外，v0.1.x 版本仍然兼容 Python2.7 和 Python3.4。“
python版本，建議使用3.5.2以上版本。我這里演示用的Python 3.9.6

安裝好python環(huán)境后，使用python3 -V查看自己的版本號(hào)，出現(xiàn)版本號(hào)說明環(huán)境已經(jīng)安裝完成

gor庫環(huán)境安裝

在python環(huán)境中使用pip3命令安裝gor庫，gor版本建議下載最新版，我這里用的是gor 0.2.3版本

pip3 install gor

如果需要使用虛擬環(huán)境，需要先進(jìn)入虛擬環(huán)境再安裝，pycharm自帶了虛擬環(huán)境，如果當(dāng)面文件存在venv文件夾說明已有虛擬環(huán)境，進(jìn)入自帶的虛擬環(huán)境安裝

source venv/bin/activate

看到前面出現(xiàn)venv，說明成功進(jìn)入虛擬環(huán)境

在python中導(dǎo)入gor庫，沒有報(bào)錯(cuò)說明gor庫環(huán)境安裝完成

from gor.middleware import AsyncioGor

python代碼編寫

首先導(dǎo)入gor庫

from gor.middleware import AsyncioGor

然后在代碼內(nèi)定義三個(gè)函數(shù)，分別負(fù)責(zé)請(qǐng)求，響應(yīng)，重播的處理

以下是一個(gè)簡(jiǎn)單的demo，具體的使用可以查看官方文檔，gor官網(wǎng)文檔

# coding: utf-8
import sys
from gor.middleware import AsyncioGor
def on_request(proxy, msg, **kwargs):
    proxy.on('response', on_response, idx=msg.id, req=msg)
def on_response(proxy, msg, **kwargs):
    proxy.on('replay', on_replay, idx=kwargs['req'].id, req=kwargs['req'], resp=msg)
def on_replay(proxy, msg, **kwargs):    

    # do anything you want with msg
    # msg is a GorMessage object
    pass

if __name__ == '__main__':
    proxy = AsyncioGor()
    proxy.on('request', on_request)
    proxy.run()

具體代碼細(xì)節(jié)實(shí)現(xiàn)，代碼里實(shí)現(xiàn)了三個(gè)函數(shù)分別處理請(qǐng)求、響應(yīng)、重播等
請(qǐng)求部分，對(duì)請(qǐng)求進(jìn)行重寫消息頭，body，param參數(shù)等，把處理完的請(qǐng)求url的數(shù)量
響應(yīng)部分，對(duì)響應(yīng)進(jìn)行判斷，狀態(tài)碼是否等于200，等于200的記錄url的數(shù)量記錄到日志，不等于200的url，req，res記錄到日志
重播部分，對(duì)重播進(jìn)行判斷，狀態(tài)碼是否等于200，等于200的記錄url的數(shù)量記錄到日志，不等于200的url，req，res記錄到日志使用命令啟動(dòng)python的middleware

goreplay中間件使用

中間件跟命令結(jié)合使用，先后順序?yàn)?，先中間件處理，處理完再使用命令進(jìn)行過濾，過濾的命令應(yīng)該是中間件處理完的數(shù)據(jù)，具體順序如下

需要同時(shí)處理req、res、rep，可以使用如下命令

sudo gor --input-raw :5000 --output-file 'test.gor' --output-stdout --middleware 'python3 middleware.py' --output-http-track-response --input-raw-track-response

到此這篇關(guān)于GoReplay中間件python版本使用的文章就介紹到這了,更多相關(guān)GoReplay中間件python使用內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論