gor引流測(cè)試中間件golang代碼添加注釋-創(chuàng)新互聯(lián)

package main

創(chuàng)新互聯(lián)憑借在網(wǎng)站建設(shè)、網(wǎng)站推廣領(lǐng)域領(lǐng)先的技術(shù)能力和多年的行業(yè)經(jīng)驗(yàn)，為客戶提供超值的營(yíng)銷型網(wǎng)站建設(shè)服務(wù)，我們始終認(rèn)為：好的營(yíng)銷型網(wǎng)站就是好的業(yè)務(wù)員。我們已成功為企業(yè)單位、個(gè)人等客戶提供了成都網(wǎng)站建設(shè)、網(wǎng)站建設(shè)服務(wù)，以良好的商業(yè)信譽(yù)，完善的服務(wù)及深厚的技術(shù)力量處于同行領(lǐng)先地位。

import (
"bufio"
"bytes"
"encoding/hex"
"fmt"
"github.com/buger/goreplay/proto"
"os"
)

// requestID -> originalToken
var originalTokens map[string][]byte

// originalToken -> replayedToken
var tokenAliases map[string][]byte

func main() {
originalTokens = make(map[string][]byte)
tokenAliases = make(map[string][]byte)

scanner := bufio.NewScanner(os.Stdin)

for scanner.Scan() {
    encoded := scanner.Bytes()
    buf := make([]byte, len(encoded)/2)
    hex.Decode(buf, encoded)

    process(buf)
}

}

func process(buf []byte) {
// First byte indicate payload type, possible values:
// 1 - Request
// 2 - Response
// 3 - ReplayedResponse
payloadType := buf[0]
headerSize := bytes.IndexByte(buf, '\n') + 1
header := buf[:headerSize-1]

// Header contains space separated values of: request type, request id, and request start time (or round-trip time for responses)
meta := bytes.Split(header, []byte(" "))
// For each request you should receive 3 payloads (request, response, replayed response) with same request id
reqID := string(meta[1])
payload := buf[headerSize:]

Debug("Received payload:", string(buf))

switch payloadType {
case '1': // Request
    if bytes.Equal(proto.Path(payload), []byte("/token")) {
        originalTokens[reqID] = []byte{}
        Debug("Found token request:", reqID)
    } else {
        token, vs, _ := proto.PathParam(payload, []byte("token"))//取到生產(chǎn)服務(wù)器的token值

        if vs != -1 { // If there is GET token param
            if alias, ok := tokenAliases[string(token)]; ok {  //檢查要替換的token值是否存在
                // Rewrite original token to alias
                payload = proto.SetPathParam(payload, []byte("token"), alias)//將生產(chǎn)的token替換成測(cè)服的token

                // Copy modified payload to our buffer
                buf = append(buf[:headerSize], payload...)
            }
        }
    }

    // Emitting data back
    os.Stdout.Write(encode(buf))//重寫(xiě)請(qǐng)求準(zhǔn)備發(fā)往測(cè)試服務(wù)器
case '2': // Original response
    if _, ok := originalTokens[reqID]; ok {
        // Token is inside response body
        secureToken := proto.Body(payload) //取到生產(chǎn)服務(wù)器中返回的token值
        originalTokens[reqID] = secureToken
        Debug("Remember origial token:", string(secureToken))
    }
case '3': // Replayed response
    if originalToken, ok := originalTokens[reqID]; ok {
        delete(originalTokens, reqID)
        secureToken := proto.Body(payload)
        tokenAliases[string(originalToken)] = secureToken//拿到測(cè)試服務(wù)器的token值用來(lái)替換掉正服的token值

        Debug("Create alias for new token token, was:", string(originalToken), "now:", string(secureToken))
    }
}

}

func encode(buf []byte) []byte {
dst := make([]byte, len(buf)*2+1)
hex.Encode(dst, buf)
dst[len(dst)-1] = '\n'

return dst

}

func Debug(args ...interface{}) {
fmt.Fprint(os.Stderr, "[DEBUG][TOKEN-MOD] ")
fmt.Fprintln(os.Stderr, args...)
}

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無(wú)理由+7*72小時(shí)售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性價(jià)比高”等特點(diǎn)與優(yōu)勢(shì)，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場(chǎng)景需求。

分享名稱：gor引流測(cè)試中間件golang代碼添加注釋-創(chuàng)新互聯(lián)
文章分享：http://bm7419.com/article36/goopg.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供面包屑導(dǎo)航、微信公眾號(hào)、企業(yè)網(wǎng)站制作、靜態(tài)網(wǎng)站、做網(wǎng)站、定制開(kāi)發(fā)

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源：創(chuàng)新互聯(lián)

猜你還喜歡下面的內(nèi)容