001 /** 002 * Copyright (c) 2010 Yahoo! Inc. All rights reserved. 003 * Licensed under the Apache License, Version 2.0 (the "License"); 004 * you may not use this file except in compliance with the License. 005 * You may obtain a copy of the License at 006 * 007 * http://www.apache.org/licenses/LICENSE-2.0 008 * 009 * Unless required by applicable law or agreed to in writing, software 010 * distributed under the License is distributed on an "AS IS" BASIS, 011 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 012 * See the License for the specific language governing permissions and 013 * limitations under the License. See accompanying LICENSE file. 014 */ 015 package org.apache.oozie.util.db; 016 017 import org.apache.commons.dbcp.BasicDataSource; 018 import org.apache.oozie.service.InstrumentationService; 019 import org.apache.oozie.service.Services; 020 import org.apache.oozie.util.Instrumentation; 021 import org.apache.openjpa.lib.util.Closeable; 022 023 import java.sql.SQLException; 024 025 public class InstrumentedBasicDataSource extends BasicDataSource implements Closeable { 026 public static final String INSTR_GROUP = "jdbc"; 027 public static final String INSTR_NAME = "connections.active"; 028 029 /** 030 * The created datasource instruments the active DB connections. 031 */ 032 public InstrumentedBasicDataSource() { 033 Instrumentation instr = Services.get().get(InstrumentationService.class).get(); 034 instr.addSampler(INSTR_GROUP, INSTR_NAME, 60, 1, new Instrumentation.Variable<Long>() { 035 public Long getValue() { 036 return (long) getNumActive(); 037 } 038 }); 039 } 040 041 @Override 042 public void close() { 043 try { 044 super.close(); 045 } 046 catch (SQLException ex) { 047 throw new RuntimeException(ex); 048 } 049 } 050 }