Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

support monitor

Browse Source
This commit is contained in:
wenshao
2013-08-14 21:32:26 +08:00
parent 7f255fa494
commit a16e7ec5da
5 changed files with 258 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/support/monitor/MonitorClient.java
+6
View File
@@ -18,6 +18,7 @@ package com.alibaba.druid.support.monitor;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;
@@ -26,6 +27,7 @@ import java.util.concurrent.TimeUnit;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceStatValue;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcSqlStatValue;
import com.alibaba.druid.support.http.stat.WebAppStat;
import com.alibaba.druid.support.http.stat.WebAppStatManager;
import com.alibaba.druid.support.http.stat.WebAppStatValue;
@@ -153,6 +155,10 @@ public class MonitorClient {
dao.saveWebURI(ctx, webURIValueList);
dao.saveWebApp(ctx, webAppStatValueList);
}
public List<JdbcSqlStatValue> loadSqlList(Map<String, Object> filters) {
return dao.loadSqlList(filters);
}
public MonitorDao getDao() {
return dao;
src/main/java/com/alibaba/druid/support/monitor/MonitorContext.java
+4 -4
View File
@@ -21,11 +21,11 @@ import java.util.Map;
public class MonitorContext {
private final Map<String, Object> atrributes = new HashMap<String, Object>();
private final Map<String, Object> atrributes = new HashMap<String, Object>();
private String domainName;
private String appName;
private String clusterName;
private String domainName = "defaultDomain";
private String appName = "defaultApp";
private String clusterName = "defaultCluster";
private String host;
private int pid;
src/main/java/com/alibaba/druid/support/monitor/dao/MonitorDao.java
+4
View File
@@ -16,8 +16,10 @@
package com.alibaba.druid.support.monitor.dao;
import java.util.List;
import java.util.Map;
import com.alibaba.druid.pool.DruidDataSourceStatValue;
import com.alibaba.druid.stat.JdbcSqlStatValue;
import com.alibaba.druid.support.http.stat.WebAppStatValue;
import com.alibaba.druid.support.http.stat.WebURIStatValue;
import com.alibaba.druid.support.monitor.MonitorContext;
@@ -32,4 +34,6 @@ public interface MonitorDao {
void saveWebURI(MonitorContext ctx, List<WebURIStatValue> uriList);
void saveWebApp(MonitorContext ctx, List<WebAppStatValue> uriList);
List<JdbcSqlStatValue> loadSqlList(Map<String, Object> filters);
}
src/main/java/com/alibaba/druid/support/monitor/dao/MonitorDaoJdbcImpl.java
+224
View File
@@ -18,12 +18,16 @@ package com.alibaba.druid.support.monitor.dao;
import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.sql.Types;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
@@ -88,6 +92,216 @@ public class MonitorDaoJdbcImpl implements MonitorDao {
save(webAppStatBeanInfo, ctx, list);
}
@SuppressWarnings("unchecked")
public List<JdbcSqlStatValue> loadSqlList(Map<String, Object> filters) {
return ((List<JdbcSqlStatValue>) load(sqlStatBeanInfo, filters));
}
static Integer getInteger(Map<String, Object> filters, String key) {
Object value = filters.get(key);
if (value == null) {
return null;
}
if (value instanceof Integer) {
return (Integer) value;
}
if (value instanceof Number) {
return ((Number) value).intValue();
}
if (value instanceof String) {
String text = (String) value;
if (StringUtils.isEmpty(text)) {
return null;
}
return Integer.parseInt(text);
}
return null;
}
static Date getDate(Map<String, Object> filters, String key) {
Object value = filters.get(key);
if (value == null) {
return null;
}
if (value instanceof Number) {
long millis = ((Number) value).longValue();
return new Date(millis);
}
if (value instanceof String) {
String text = (String) value;
if (StringUtils.isEmpty(text)) {
return null;
}
try {
return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(text);
} catch (ParseException e) {
LOG.error("parse filter error", e);
return null;
}
}
return null;
}
private List<?> load(BeanInfo beanInfo, Map<String, Object> filters) {
List<Object> list = new ArrayList<Object>();
StringBuilder buf = new StringBuilder();
buf.append("SELECT ");
List<FieldInfo> fields = beanInfo.getFields();
for (int i = 0; i < fields.size(); ++i) {
FieldInfo field = fields.get(i);
if (i != 0) {
buf.append(", ");
}
buf.append(field.getColumnName());
}
buf.append("\nFROM ");
buf.append(getTableName(beanInfo));
buf.append("\nWHERE collectTime >= ? AND collectTime <= ? AND domain = ? AND app = ? AND cluster = ?");
Date startTime = getDate(filters, "startTime");
if (startTime == null) {
long now = System.currentTimeMillis();
startTime = new Date(now - 1000 * 60 * 30); // 3 hours
}
Date endTime = getDate(filters, "endTime");
if (endTime == null) {
endTime = new Date(); // now
}
String domain = (String) filters.get("domain");
if (StringUtils.isEmpty(domain)) {
domain = "defaultDomain";
}
String app = (String) filters.get("app");
if (StringUtils.isEmpty(app)) {
app = "defaultApp";
}
String cluster = (String) filters.get("cluster");
if (StringUtils.isEmpty(cluster)) {
cluster = "defaultCluster";
}
String host = (String) filters.get("host");
if (!StringUtils.isEmpty(host)) {
buf.append("\nAND host = ?");
}
Integer pid = getInteger(filters, "pid");
if (pid != null) {
buf.append("\nAND pid = ?");
}
{
Integer offset = (Integer) filters.get("offset");
Integer limit = (Integer) filters.get("limit");
if (limit == null) {
limit = 1000;
}
buf.append("\nLIMIT ");
if (offset != null) {
buf.append(offset);
buf.append(", ");
}
buf.append(limit);
}
String sql = buf.toString();
Connection conn = null;
PreparedStatement stmt = null;
ResultSet rs = null;
try {
conn = dataSource.getConnection();
stmt = conn.prepareStatement(sql);
int paramIndex = 1;
stmt.setTimestamp(paramIndex++, new Timestamp(startTime.getTime()));
stmt.setTimestamp(paramIndex++, new Timestamp(endTime.getTime()));
stmt.setString(paramIndex++, domain);
stmt.setString(paramIndex++, app);
stmt.setString(paramIndex++, cluster);
if (!StringUtils.isEmpty(host)) {
stmt.setString(paramIndex++, host);
}
if (pid != null) {
stmt.setInt(paramIndex++, pid);
}
rs = stmt.executeQuery();
while (rs.next()) {
Object object = createInstance(beanInfo);
for (int i = 0; i < fields.size(); ++i) {
FieldInfo field = fields.get(i);
readFieldValue(object, field, rs, i + 1);
}
list.add(object);
}
stmt.close();
} catch (SQLException ex) {
LOG.error("save sql error", ex);
} finally {
JdbcUtils.close(rs);
JdbcUtils.close(stmt);
JdbcUtils.close(conn);
}
return list;
}
protected void readFieldValue(Object object, FieldInfo field, ResultSet rs, int paramIndex) throws SQLException {
Class<?> fieldType = field.getFieldType();
Object fieldValue = null;
if (fieldType.equals(int.class) || fieldType.equals(Integer.class)) {
fieldValue = rs.getInt(paramIndex);
} else if (fieldType.equals(long.class) || fieldType.equals(Long.class)) {
fieldValue = rs.getLong(paramIndex);
} else if (fieldType.equals(String.class)) {
fieldValue = rs.getString(paramIndex);
} else if (fieldType.equals(Date.class)) {
Timestamp timestamp = rs.getTimestamp(paramIndex);
if (timestamp != null) {
fieldValue = new Date(timestamp.getTime());
}
} else {
throw new UnsupportedOperationException();
}
try {
field.getField().set(object, fieldValue);
} catch (IllegalArgumentException e) {
throw new DruidRuntimeException("set field error" + field.getField(), e);
} catch (IllegalAccessException e) {
throw new DruidRuntimeException("set field error" + field.getField(), e);
}
}
private void save(BeanInfo beanInfo, MonitorContext ctx, List<?> list) {
String sql = buildInsertSql(beanInfo);
Connection conn = null;
@@ -151,6 +365,16 @@ public class MonitorDaoJdbcImpl implements MonitorDao {
}
}
public Object createInstance(BeanInfo beanInfo) {
try {
return beanInfo.getClazz().newInstance();
} catch (InstantiationException ex) {
throw new DruidRuntimeException("create instance error", ex);
} catch (IllegalAccessException ex) {
throw new DruidRuntimeException("create instance error", ex);
}
}
public String buildInsertSql(BeanInfo beanInfo) {
String sql = beanInfo.insertSql;
src/test/java/com/alibaba/druid/bvt/support/monitor/MonitorDaoJdbcImplTest.java
+20 -1
View File
@@ -6,6 +6,8 @@ import java.util.List;
import junit.framework.TestCase;
import org.junit.Assert;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.stat.JdbcSqlStatValue;
@@ -41,7 +43,7 @@ public class MonitorDaoJdbcImplTest extends TestCase {
MonitorClient client = new MonitorClient();
client.setDao(dao);
{
String sql = buildCreateSql(dao, new BeanInfo(JdbcSqlStatValue.class));
System.out.println(SQLUtils.format(sql, JdbcConstants.MYSQL));
@@ -50,6 +52,23 @@ public class MonitorDaoJdbcImplTest extends TestCase {
client.collectSql();
{
List<JdbcSqlStatValue> sqlList = client.loadSqlList(Collections.<String, Object> emptyMap());
for (JdbcSqlStatValue sqlStatValue : sqlList) {
System.out.println(sqlStatValue.getData());
}
Assert.assertEquals(1, sqlList.size());
}
client.collectSql();
{
List<JdbcSqlStatValue> sqlList = client.loadSqlList(Collections.<String, Object> emptyMap());
for (JdbcSqlStatValue sqlStatValue : sqlList) {
System.out.println(sqlStatValue.getData());
}
Assert.assertEquals(3, sqlList.size());
}
}
public String buildCreateSql(MonitorDaoJdbcImpl dao, BeanInfo beanInfo) {