001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 * 
010 *      http://www.apache.org/licenses/LICENSE-2.0
011 * 
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.oozie.util.db;
019
020import org.apache.commons.dbcp.BasicDataSource;
021import org.apache.oozie.service.InstrumentationService;
022import org.apache.oozie.service.Services;
023import org.apache.oozie.util.Instrumentation;
024
025public class InstrumentedBasicDataSource extends BasicDataSource {
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        InstrumentationService instrumentationService = Services.get().get(InstrumentationService.class);
034        if (instrumentationService != null) {
035            Instrumentation instr = instrumentationService.get();
036            instr.addSampler(INSTR_GROUP, INSTR_NAME, 60, 1, new Instrumentation.Variable<Long>() {
037                public Long getValue() {
038                    return (long) getNumActive();
039                }
040            });
041        }
042    }
043
044}