/* | |
* Copyright 1999-2011 Alibaba Group. | |
* | |
* Licensed under the Apache License, Version 2.0 (the "License"); | |
* you may not use this file except in compliance with the License. | |
* You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, software | |
* distributed under the License is distributed on an "AS IS" BASIS, | |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
* See the License for the specific language governing permissions and | |
* limitations under the License. | |
*/ | |
package com.alibaba.dubbo.config.spring.status; | |
import java.sql.Connection; | |
import java.sql.DatabaseMetaData; | |
import java.sql.ResultSet; | |
import java.util.Map; | |
import javax.sql.DataSource; | |
import org.springframework.context.ApplicationContext; | |
import com.alibaba.dubbo.common.Extension; | |
import com.alibaba.dubbo.common.logger.Logger; | |
import com.alibaba.dubbo.common.logger.LoggerFactory; | |
import com.alibaba.dubbo.common.status.Status; | |
import com.alibaba.dubbo.common.status.StatusChecker; | |
import com.alibaba.dubbo.config.spring.ServiceBean; | |
/** | |
* DataSourceStatusChecker | |
* | |
* @author william.liangf | |
*/ | |
@Extension("datasource") | |
public class DataSourceStatusChecker implements StatusChecker { | |
private static final Logger logger = LoggerFactory.getLogger(DataSourceStatusChecker.class); | |
@SuppressWarnings("unchecked") | |
public Status check() { | |
ApplicationContext context = ServiceBean.getSpringContext(); | |
if (context == null) { | |
return new Status(Status.Level.UNKNOWN); | |
} | |
Map<String, DataSource> dataSources = context.getBeansOfType(DataSource.class, false, false); | |
if (dataSources == null || dataSources.size() == 0) { | |
return new Status(Status.Level.UNKNOWN); | |
} | |
Status.Level level = Status.Level.OK; | |
StringBuilder buf = new StringBuilder(); | |
for (Map.Entry<String, DataSource> entry : dataSources.entrySet()) { | |
DataSource dataSource = entry.getValue(); | |
if (buf.length() > 0) { | |
buf.append(", "); | |
} | |
buf.append(entry.getKey()); | |
try { | |
Connection connection = dataSource.getConnection(); | |
try { | |
DatabaseMetaData metaData = connection.getMetaData(); | |
ResultSet resultSet = metaData.getTypeInfo(); | |
try { | |
if (! resultSet.next()) { | |
level = Status.Level.ERROR; | |
} | |
} finally { | |
resultSet.close(); | |
} | |
buf.append(metaData.getURL()); | |
buf.append("("); | |
buf.append(metaData.getDatabaseProductName()); | |
buf.append("-"); | |
buf.append(metaData.getDatabaseProductVersion()); | |
buf.append(")"); | |
} finally { | |
connection.close(); | |
} | |
} catch (Throwable e) { | |
logger.warn(e.getMessage(), e); | |
return new Status(level, e.getMessage()); | |
} | |
} | |
return new Status(level, buf.toString()); | |
} | |
} |