--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/java/sql-dk/src/info/globalcode/sql/dk/jmx/ConnectionManagement.java Thu Sep 25 17:50:40 2014 +0200
@@ -0,0 +1,97 @@
+/**
+ * SQL-DK
+ * Copyright © 2014 František Kučera (frantovo.cz)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package info.globalcode.sql.dk.jmx;
+
+import java.util.EnumMap;
+import java.util.Map;
+
+/**
+ * JMX management bean for progress reporting.
+ *
+ * @author Ing. František Kučera (frantovo.cz)
+ */
+public class ConnectionManagement implements ConnectionManagementMBean {
+
+ private final String databaseName;
+ private final Map<COUNTER, Integer> counters = new EnumMap(COUNTER.class);
+
+ public ConnectionManagement(String databaseName) {
+ this.databaseName = databaseName;
+ for (COUNTER c : COUNTER.values()) {
+ counters.put(c, 0);
+ }
+ }
+
+ public enum COUNTER {
+
+ COMMAND,
+ RECORD_CURRENT,
+ RECORD_TOTAL
+ };
+
+ public void incrementCounter(COUNTER counter) {
+ synchronized (counters) {
+ int old = counters.get(counter);
+ counters.put(counter, old + 1);
+ }
+ }
+
+ public void resetCounter(COUNTER counter) {
+ synchronized (counters) {
+ counters.put(counter, 0);
+ }
+ }
+
+ public static void incrementCounter(ConnectionManagement mbean, COUNTER counter) {
+ if (mbean != null) {
+ mbean.incrementCounter(counter);
+ }
+ }
+
+ public static void resetCounter(ConnectionManagement mbean, COUNTER counter) {
+ if (mbean != null) {
+ mbean.resetCounter(counter);
+ }
+ }
+
+ @Override
+ public String getDatabaseName() {
+ return databaseName;
+ }
+
+ @Override
+ public int getCommandCount() {
+ synchronized (counters) {
+ return counters.get(COUNTER.COMMAND);
+ }
+ }
+
+ @Override
+ public int getCurrentRecordCount() {
+ synchronized (counters) {
+ return counters.get(COUNTER.RECORD_CURRENT);
+ }
+ }
+
+ @Override
+ public int getTotalRecordCount() {
+ synchronized (counters) {
+ return counters.get(COUNTER.RECORD_TOTAL);
+ }
+ }
+}