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 +}