在作性能測試過程當中,遇到一個棘手的問題,開發讓咱們復現幾個請求時間較長的請求,他們看日誌進行鏈路追蹤,查找瓶頸所在。java
這裏說一下框架中的處理邏輯:每一個請求有一個惟一的requestid,由幾部分組成,還有一些算法保證其惟一性。而後這個requestID貫穿整個請求過程的日誌,服務間的相互調用,與數據庫中間件的交互都依賴於這個requestID。算法
以往壓測都是寫了一個請求ID,並未對這個header作處理,如今得搞起來了。數據庫
首先我先新建了一個Java interface,用於使用閉包直接完成這個功能,還有就是其餘標記方法:apache
package com.fun.base.interfaces; import org.apache.http.client.methods.HttpRequestBase; import java.io.Serializable; /** * 用來標記request,爲了記錄超時的請求 */ public interface MarkRequest extends Serializable { /** * 用來標記base,刪除header其中一項,添加一項 * * @param base * @return */ public String mark(HttpRequestBase base); }
而後我再ThreadLimitTimesCount
和ThreadLimitTimeCount
實現類中使用這個接口對象,兩個實現類的代碼已經發過了性能測試框架第二版,這裏只寫一個:編程
mark
對象,而不是多線程共享,下面會看到效果。package com.fun.frame.thead; import com.fun.base.constaint.ThreadLimitTimesCount; import com.fun.base.interfaces.MarkRequest; import com.fun.config.Constant; import com.fun.config.HttpClientConstant; import com.fun.frame.Save; import com.fun.frame.excute.Concurrent; import com.fun.frame.httpclient.FanLibrary; import com.fun.frame.httpclient.FunRequest; import com.fun.frame.httpclient.GCThread; import com.fun.utils.Time; import org.apache.http.client.methods.HttpRequestBase; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.util.ArrayList; import java.util.List; import java.util.Vector; /** * http請求多線程類 */ public class RequestThreadTimes extends ThreadLimitTimesCount { private static final long serialVersionUID = -2751325651625435070L; static Logger logger = LoggerFactory.getLogger(RequestThreadTimes.class); /** * 記錄總的請求超時的狀況 */ public static Vector<String> requestMark = new Vector<>(); /** * 請求 */ public HttpRequestBase request; /** * 標記對象 */ public MarkRequest mark; /** * 記錄當前線程超時請求 */ public List<String> marks = new ArrayList<>(); /** * 單請求多線程屢次任務構造方法 * * @param request 被執行的請求 * @param times 每一個線程運行的次數 */ public RequestThreadTimes(HttpRequestBase request, int times) { this.request = request; this.times = times; this.mark = new MarkRequest() { private static final long serialVersionUID = 5599842482575655279L; @Override public String mark(HttpRequestBase base) { return EMPTY; } }; } /** * 應對對每一個請求進行標記的狀況 * * @param request * @param times * @param mark */ public RequestThreadTimes(HttpRequestBase request, int times, MarkRequest mark) { this(request, times); this.mark = mark; } protected RequestThreadTimes() { super(); } @Override public void before() { super.before(); GCThread.starts(); } /** * @throws Exception */ @Override protected void doing() throws Exception { FanLibrary.excuteSimlple(request); } @Override protected void after() { requestMark.addAll(marks); GCThread.stop(); synchronized (RequestThreadTimes.class) { if (countDownLatch.getCount() == 0) Save.saveStringList(requestMark, Constant.DEFAULT_STRING); } } @Override public void run() { try { before(); List<Long> t = new ArrayList<>(); long ss = Time.getTimeStamp(); for (int i = 0; i < times; i++) { try { String m = this.mark.mark(request); long s = Time.getTimeStamp(); doing(); long e = Time.getTimeStamp(); long diff = e - s; t.add(diff); if (diff > HttpClientConstant.MAX_ACCEPT_TIME) marks.add(diff + CONNECTOR + m); excuteNum++; if (status()) break; } catch (Exception e) { logger.warn("執行任務失敗!", e); errorNum++; } } long ee = Time.getTimeStamp(); logger.info("執行次數:{},錯誤次數: {},總耗時:{} s", times, errorNum, (ee - ss) / 1000 + 1); Concurrent.allTimes.addAll(t); } catch (Exception e) { logger.warn("執行任務失敗!", e); } finally { if (countDownLatch != null) countDownLatch.countDown(); after(); } } @Override public RequestThreadTimes clone() { RequestThreadTimes threadTimes = new RequestThreadTimes(); threadTimes.times = this.times; threadTimes.request = FunRequest.cloneRequest(request); threadTimes.mark = deepClone(mark); return threadTimes; } }
我本身寫了一個使用Demo:瀏覽器
def "測試併發狀況下記錄響應標記符的"() { given: HttpGet httpGet = FanLibrary.getHttpGet("https://cn.bing.com/"); MarkRequest mark = new MarkRequest() { String m; @Override public String mark(HttpRequestBase base) { base.removeHeaders("requestid"); m = m == null ? RString.getStringWithoutNum(4) : m String value = "fun_" + m + CONNECTOR + Time.getTimeStamp(); base.addHeader("requestid", value); return value; } }; FanLibrary.getHttpResponse(httpGet); HttpClientConstant.MAX_ACCEPT_TIME = -1 RequestThreadTimes threadTimes = new RequestThreadTimes(httpGet, 2, mark); new Concurrent(threadTimes, 2).start(); output(RequestThreadTimes.requestMark) }
下面是記錄的結果以下,能夠看到,一共出現了兩個m
的值,後面跟的是時間戳,這樣既保證了requestID惟一性,也能夠對線程進行歸類。多線程
80_fun_QkhQ_1578367527661 103_fun_QkhQ_1578367527742 101_fun_zwtk_1578367527661 107_fun_zwtk_1578367527763