001    /**
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.activemq.broker.jmx;
018    
019    import java.util.concurrent.Callable;
020    import org.apache.activemq.store.PersistenceAdapter;
021    
022    public class PersistenceAdapterView implements PersistenceAdapterViewMBean {
023    
024        private final String name;
025        private final PersistenceAdapter persistenceAdapter;
026    
027        private Callable<String> inflightTransactionViewCallable;
028        private Callable<String> dataViewCallable;
029    
030        public PersistenceAdapterView(PersistenceAdapter adapter) {
031            this.name = adapter.toString();
032            this.persistenceAdapter = adapter;
033        }
034    
035        @Override
036        public String getName() {
037            return name;
038        }
039    
040        @Override
041        public String getTransactions() {
042            return invoke(inflightTransactionViewCallable);
043        }
044    
045        @Override
046        public String getData() {
047            return invoke(dataViewCallable);
048        }
049    
050        @Override
051        public long getSize() {
052            return persistenceAdapter.size();
053        }
054    
055        private String invoke(Callable<String> callable) {
056            String result = null;
057            if (callable != null) {
058                try {
059                    result = callable.call();
060                } catch (Exception e) {
061                    result = e.toString();
062                }
063            }
064            return result;
065        }
066    
067        public void setDataViewCallable(Callable<String> dataViewCallable) {
068            this.dataViewCallable = dataViewCallable;
069        }
070    
071        public void setInflightTransactionViewCallable(Callable<String> inflightTransactionViewCallable) {
072            this.inflightTransactionViewCallable = inflightTransactionViewCallable;
073        }
074    }