304 lines
10 KiB
Java
304 lines
10 KiB
Java
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
/*
|
|
* Copyright 2015 Cloudius Systems
|
|
*
|
|
* Modified by Cloudius Systems
|
|
*/
|
|
package org.apache.cassandra.service;
|
|
|
|
import static java.util.Collections.emptySet;
|
|
|
|
import java.lang.management.ManagementFactory;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
import javax.management.MBeanServer;
|
|
import javax.management.ObjectName;
|
|
import javax.ws.rs.core.MultivaluedHashMap;
|
|
import javax.ws.rs.core.MultivaluedMap;
|
|
|
|
import com.scylladb.jmx.api.APIClient;
|
|
|
|
public class StorageProxy implements StorageProxyMBean {
|
|
public static final String MBEAN_NAME = "org.apache.cassandra.db:type=StorageProxy";
|
|
private static final java.util.logging.Logger logger = java.util.logging.Logger
|
|
.getLogger(StorageProxy.class.getName());
|
|
|
|
private APIClient c = new APIClient();
|
|
|
|
public void log(String str) {
|
|
logger.finest(str);
|
|
}
|
|
|
|
private static final StorageProxy instance = new StorageProxy();
|
|
|
|
public static StorageProxy getInstance() {
|
|
return instance;
|
|
}
|
|
|
|
public static final String UNREACHABLE = "UNREACHABLE";
|
|
|
|
private StorageProxy() {
|
|
}
|
|
|
|
static {
|
|
MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
|
|
try {
|
|
mbs.registerMBean(instance, new ObjectName(MBEAN_NAME));
|
|
} catch (Exception e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
|
|
}
|
|
|
|
@Override
|
|
public long getTotalHints() {
|
|
log(" getTotalHints()");
|
|
return c.getLongValue("storage_proxy/total_hints");
|
|
}
|
|
|
|
@Override
|
|
public boolean getHintedHandoffEnabled() {
|
|
log(" getHintedHandoffEnabled()");
|
|
return c.getBooleanValue("storage_proxy/hinted_handoff_enabled");
|
|
}
|
|
|
|
@Override
|
|
public Set<String> getHintedHandoffEnabledByDC() {
|
|
log(" getHintedHandoffEnabledByDC()");
|
|
return c.getSetStringValue(
|
|
"storage_proxy/hinted_handoff_enabled_by_dc");
|
|
}
|
|
|
|
@Override
|
|
public void setHintedHandoffEnabled(boolean b) {
|
|
log(" setHintedHandoffEnabled(boolean b)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("enable", Boolean.toString(b));
|
|
c.post("storage_proxy/hinted_handoff_enabled", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public void setHintedHandoffEnabledByDCList(String dcs) {
|
|
log(" setHintedHandoffEnabledByDCList(String dcs)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("dcs", dcs);
|
|
c.post("storage_proxy/hinted_handoff_enabled_by_dc_list");
|
|
}
|
|
|
|
@Override
|
|
public int getMaxHintWindow() {
|
|
log(" getMaxHintWindow()");
|
|
return c.getIntValue("storage_proxy/max_hint_window");
|
|
}
|
|
|
|
@Override
|
|
public void setMaxHintWindow(int ms) {
|
|
log(" setMaxHintWindow(int ms)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("ms", Integer.toString(ms));
|
|
c.post("storage_proxy/max_hint_window", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public int getMaxHintsInProgress() {
|
|
log(" getMaxHintsInProgress()");
|
|
return c.getIntValue("storage_proxy/max_hints_in_progress");
|
|
}
|
|
|
|
@Override
|
|
public void setMaxHintsInProgress(int qs) {
|
|
log(" setMaxHintsInProgress(int qs)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("qs", Integer.toString(qs));
|
|
c.post("storage_proxy/max_hints_in_progress", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public int getHintsInProgress() {
|
|
log(" getHintsInProgress()");
|
|
return c.getIntValue("storage_proxy/hints_in_progress");
|
|
}
|
|
|
|
@Override
|
|
public Long getRpcTimeout() {
|
|
log(" getRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setRpcTimeout(Long timeoutInMillis) {
|
|
log(" setRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getReadRpcTimeout() {
|
|
log(" getReadRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/read_rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setReadRpcTimeout(Long timeoutInMillis) {
|
|
log(" setReadRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/read_rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getWriteRpcTimeout() {
|
|
log(" getWriteRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/write_rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setWriteRpcTimeout(Long timeoutInMillis) {
|
|
log(" setWriteRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/write_rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getCounterWriteRpcTimeout() {
|
|
log(" getCounterWriteRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/counter_write_rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setCounterWriteRpcTimeout(Long timeoutInMillis) {
|
|
log(" setCounterWriteRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/counter_write_rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getCasContentionTimeout() {
|
|
log(" getCasContentionTimeout()");
|
|
return c.getLongValue("storage_proxy/cas_contention_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setCasContentionTimeout(Long timeoutInMillis) {
|
|
log(" setCasContentionTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/cas_contention_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getRangeRpcTimeout() {
|
|
log(" getRangeRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/range_rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setRangeRpcTimeout(Long timeoutInMillis) {
|
|
log(" setRangeRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/range_rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public Long getTruncateRpcTimeout() {
|
|
log(" getTruncateRpcTimeout()");
|
|
return c.getLongValue("storage_proxy/truncate_rpc_timeout");
|
|
}
|
|
|
|
@Override
|
|
public void setTruncateRpcTimeout(Long timeoutInMillis) {
|
|
log(" setTruncateRpcTimeout(Long timeoutInMillis)");
|
|
MultivaluedMap<String, String> queryParams = new MultivaluedHashMap<String, String>();
|
|
queryParams.add("timeout", Long.toString(timeoutInMillis));
|
|
c.post("storage_proxy/truncate_rpc_timeout", queryParams);
|
|
}
|
|
|
|
@Override
|
|
public void reloadTriggerClasses() {
|
|
log(" reloadTriggerClasses()");
|
|
c.post("storage_proxy/reload_trigger_classes");
|
|
}
|
|
|
|
@Override
|
|
public long getReadRepairAttempted() {
|
|
log(" getReadRepairAttempted()");
|
|
return c.getLongValue("storage_proxy/read_repair_attempted");
|
|
}
|
|
|
|
@Override
|
|
public long getReadRepairRepairedBlocking() {
|
|
log(" getReadRepairRepairedBlocking()");
|
|
return c.getLongValue("storage_proxy/read_repair_repaired_blocking");
|
|
}
|
|
|
|
@Override
|
|
public long getReadRepairRepairedBackground() {
|
|
log(" getReadRepairRepairedBackground()");
|
|
return c.getLongValue("storage_proxy/read_repair_repaired_background");
|
|
}
|
|
|
|
/** Returns each live node's schema version */
|
|
@Override
|
|
public Map<String, List<String>> getSchemaVersions() {
|
|
log(" getSchemaVersions()");
|
|
return c.getMapStringListStrValue("storage_proxy/schema_versions");
|
|
}
|
|
|
|
@Override
|
|
public void setNativeTransportMaxConcurrentConnections(
|
|
Long nativeTransportMaxConcurrentConnections) {
|
|
// TODO Auto-generated method stub
|
|
log(" setNativeTransportMaxConcurrentConnections()");
|
|
|
|
}
|
|
|
|
@Override
|
|
public Long getNativeTransportMaxConcurrentConnections() {
|
|
// TODO Auto-generated method stub
|
|
log(" getNativeTransportMaxConcurrentConnections()");
|
|
return c.getLongValue("");
|
|
}
|
|
|
|
@Override
|
|
public void enableHintsForDC(String dc) {
|
|
// TODO if/when scylla uses hints
|
|
log(" enableHintsForDC()");
|
|
}
|
|
|
|
@Override
|
|
public void disableHintsForDC(String dc) {
|
|
// TODO if/when scylla uses hints
|
|
log(" disableHintsForDC()");
|
|
}
|
|
|
|
@Override
|
|
public Set<String> getHintedHandoffDisabledDCs() {
|
|
// TODO if/when scylla uses hints
|
|
log(" getHintedHandoffDisabledDCs()");
|
|
return emptySet();
|
|
}
|
|
}
|