diff --git a/cat-client/src/main/java/com/dianping/cat/CatConstants.java b/cat-client/src/main/java/com/dianping/cat/CatConstants.java index 82971ef76441c063a10fe78a019ec7ab4bbec228..11d2d143eccfbe72c1cd8fa21799fb2c73f68bc3 100644 --- a/cat-client/src/main/java/com/dianping/cat/CatConstants.java +++ b/cat-client/src/main/java/com/dianping/cat/CatConstants.java @@ -15,6 +15,8 @@ public class CatConstants { public static final String CAT_PAGE_URI = "cat-page-uri"; + public static final String CAT_PAGE_TYPE = "cat-page-type"; + /** * Pigeon Transation Type */ diff --git a/cat-client/src/main/java/com/dianping/cat/servlet/CatFilter.java b/cat-client/src/main/java/com/dianping/cat/servlet/CatFilter.java index e34bbe08b5d72c98f6f3c7554e263b6149d6001a..509d48e71b2270df8d7a3ccf094f4710b6b5401a 100644 --- a/cat-client/src/main/java/com/dianping/cat/servlet/CatFilter.java +++ b/cat-client/src/main/java/com/dianping/cat/servlet/CatFilter.java @@ -268,10 +268,19 @@ public class CatFilter implements Filter { } private void customizeUri(Transaction t, HttpServletRequest req) { + if(!(t instanceof DefaultTransaction)) { + return; + } + Object catPageUri = req.getAttribute(CatConstants.CAT_PAGE_URI); + DefaultTransaction transaction = (DefaultTransaction) t; + if (catPageUri != null) { + transaction.setName(catPageUri.toString()); + } - if (t instanceof DefaultTransaction && catPageUri instanceof String) { - ((DefaultTransaction) t).setName(catPageUri.toString()); + Object catPageType = req.getAttribute(CatConstants.CAT_PAGE_TYPE); + if(catPageType != null) { + transaction.setType(catPageType.toString()); } }