1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/java/sql-dk/src/main/java/info/globalcode/sql/dk/jmx/ConnectionManagement.java Mon Mar 04 20:15:24 2019 +0100
1.3 @@ -0,0 +1,97 @@
1.4 +/**
1.5 + * SQL-DK
1.6 + * Copyright © 2014 František Kučera (frantovo.cz)
1.7 + *
1.8 + * This program is free software: you can redistribute it and/or modify
1.9 + * it under the terms of the GNU General Public License as published by
1.10 + * the Free Software Foundation, either version 3 of the License, or
1.11 + * (at your option) any later version.
1.12 + *
1.13 + * This program is distributed in the hope that it will be useful,
1.14 + * but WITHOUT ANY WARRANTY; without even the implied warranty of
1.15 + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
1.16 + * GNU General Public License for more details.
1.17 + *
1.18 + * You should have received a copy of the GNU General Public License
1.19 + * along with this program. If not, see <http://www.gnu.org/licenses/>.
1.20 + */
1.21 +package info.globalcode.sql.dk.jmx;
1.22 +
1.23 +import java.util.EnumMap;
1.24 +import java.util.Map;
1.25 +
1.26 +/**
1.27 + * JMX management bean for progress reporting.
1.28 + *
1.29 + * @author Ing. František Kučera (frantovo.cz)
1.30 + */
1.31 +public class ConnectionManagement implements ConnectionManagementMBean {
1.32 +
1.33 + private final String databaseName;
1.34 + private final Map<COUNTER, Integer> counters = new EnumMap(COUNTER.class);
1.35 +
1.36 + public ConnectionManagement(String databaseName) {
1.37 + this.databaseName = databaseName;
1.38 + for (COUNTER c : COUNTER.values()) {
1.39 + counters.put(c, 0);
1.40 + }
1.41 + }
1.42 +
1.43 + public enum COUNTER {
1.44 +
1.45 + COMMAND,
1.46 + RECORD_CURRENT,
1.47 + RECORD_TOTAL
1.48 + };
1.49 +
1.50 + public void incrementCounter(COUNTER counter) {
1.51 + synchronized (counters) {
1.52 + int old = counters.get(counter);
1.53 + counters.put(counter, old + 1);
1.54 + }
1.55 + }
1.56 +
1.57 + public void resetCounter(COUNTER counter) {
1.58 + synchronized (counters) {
1.59 + counters.put(counter, 0);
1.60 + }
1.61 + }
1.62 +
1.63 + public static void incrementCounter(ConnectionManagement mbean, COUNTER counter) {
1.64 + if (mbean != null) {
1.65 + mbean.incrementCounter(counter);
1.66 + }
1.67 + }
1.68 +
1.69 + public static void resetCounter(ConnectionManagement mbean, COUNTER counter) {
1.70 + if (mbean != null) {
1.71 + mbean.resetCounter(counter);
1.72 + }
1.73 + }
1.74 +
1.75 + @Override
1.76 + public String getDatabaseName() {
1.77 + return databaseName;
1.78 + }
1.79 +
1.80 + @Override
1.81 + public int getCommandCount() {
1.82 + synchronized (counters) {
1.83 + return counters.get(COUNTER.COMMAND);
1.84 + }
1.85 + }
1.86 +
1.87 + @Override
1.88 + public int getCurrentRecordCount() {
1.89 + synchronized (counters) {
1.90 + return counters.get(COUNTER.RECORD_CURRENT);
1.91 + }
1.92 + }
1.93 +
1.94 + @Override
1.95 + public int getTotalRecordCount() {
1.96 + synchronized (counters) {
1.97 + return counters.get(COUNTER.RECORD_TOTAL);
1.98 + }
1.99 + }
1.100 +}