Google Git
Sign in
apache / hbase-site / 5a6938ba8875b19a3d96e84a37f1016c53778aec / . / devapidocs / src-html / org / apache / hadoop / hbase / HBaseRpcServicesBase.html
blob: 076a00b4d590988e32f17a47e4b3992bc7733cf4 [file] [log] [blame]
<!DOCTYPE HTML>
<html lang="en">
<head>
<!-- Generated by javadoc (17) -->
<title>Source code</title>
<meta name="viewport" content="width=device-width, initial-scale=1">
<meta name="description" content="source: package: org.apache.hadoop.hbase, class: HBaseRpcServicesBase">
<meta name="generator" content="javadoc/SourceToHTMLConverter">
<link rel="stylesheet" type="text/css" href="../../../../../stylesheet.css" title="Style">
</head>
<body class="source-page">
<main role="main">
<div class="source-container">
<pre><span class="source-line-no">001</span><span id="line-1">/*</span>
<span class="source-line-no">002</span><span id="line-2"> * Licensed to the Apache Software Foundation (ASF) under one</span>
<span class="source-line-no">003</span><span id="line-3"> * or more contributor license agreements. See the NOTICE file</span>
<span class="source-line-no">004</span><span id="line-4"> * distributed with this work for additional information</span>
<span class="source-line-no">005</span><span id="line-5"> * regarding copyright ownership. The ASF licenses this file</span>
<span class="source-line-no">006</span><span id="line-6"> * to you under the Apache License, Version 2.0 (the</span>
<span class="source-line-no">007</span><span id="line-7"> * "License"); you may not use this file except in compliance</span>
<span class="source-line-no">008</span><span id="line-8"> * with the License. You may obtain a copy of the License at</span>
<span class="source-line-no">009</span><span id="line-9"> *</span>
<span class="source-line-no">010</span><span id="line-10"> * http://www.apache.org/licenses/LICENSE-2.0</span>
<span class="source-line-no">011</span><span id="line-11"> *</span>
<span class="source-line-no">012</span><span id="line-12"> * Unless required by applicable law or agreed to in writing, software</span>
<span class="source-line-no">013</span><span id="line-13"> * distributed under the License is distributed on an "AS IS" BASIS,</span>
<span class="source-line-no">014</span><span id="line-14"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
<span class="source-line-no">015</span><span id="line-15"> * See the License for the specific language governing permissions and</span>
<span class="source-line-no">016</span><span id="line-16"> * limitations under the License.</span>
<span class="source-line-no">017</span><span id="line-17"> */</span>
<span class="source-line-no">018</span><span id="line-18">package org.apache.hadoop.hbase;</span>
<span class="source-line-no">019</span><span id="line-19"></span>
<span class="source-line-no">020</span><span id="line-20">import java.io.IOException;</span>
<span class="source-line-no">021</span><span id="line-21">import java.lang.reflect.InvocationTargetException;</span>
<span class="source-line-no">022</span><span id="line-22">import java.lang.reflect.Method;</span>
<span class="source-line-no">023</span><span id="line-23">import java.net.BindException;</span>
<span class="source-line-no">024</span><span id="line-24">import java.net.InetSocketAddress;</span>
<span class="source-line-no">025</span><span id="line-25">import java.util.Collections;</span>
<span class="source-line-no">026</span><span id="line-26">import java.util.List;</span>
<span class="source-line-no">027</span><span id="line-27">import java.util.Optional;</span>
<span class="source-line-no">028</span><span id="line-28">import org.apache.hadoop.conf.Configuration;</span>
<span class="source-line-no">029</span><span id="line-29">import org.apache.hadoop.hbase.client.ConnectionUtils;</span>
<span class="source-line-no">030</span><span id="line-30">import org.apache.hadoop.hbase.conf.ConfigurationObserver;</span>
<span class="source-line-no">031</span><span id="line-31">import org.apache.hadoop.hbase.io.ByteBuffAllocator;</span>
<span class="source-line-no">032</span><span id="line-32">import org.apache.hadoop.hbase.ipc.HBaseRPCErrorHandler;</span>
<span class="source-line-no">033</span><span id="line-33">import org.apache.hadoop.hbase.ipc.PriorityFunction;</span>
<span class="source-line-no">034</span><span id="line-34">import org.apache.hadoop.hbase.ipc.QosPriority;</span>
<span class="source-line-no">035</span><span id="line-35">import org.apache.hadoop.hbase.ipc.RpcScheduler;</span>
<span class="source-line-no">036</span><span id="line-36">import org.apache.hadoop.hbase.ipc.RpcServer;</span>
<span class="source-line-no">037</span><span id="line-37">import org.apache.hadoop.hbase.ipc.RpcServer.BlockingServiceAndInterface;</span>
<span class="source-line-no">038</span><span id="line-38">import org.apache.hadoop.hbase.ipc.RpcServerFactory;</span>
<span class="source-line-no">039</span><span id="line-39">import org.apache.hadoop.hbase.ipc.RpcServerInterface;</span>
<span class="source-line-no">040</span><span id="line-40">import org.apache.hadoop.hbase.namequeues.NamedQueuePayload;</span>
<span class="source-line-no">041</span><span id="line-41">import org.apache.hadoop.hbase.namequeues.NamedQueueRecorder;</span>
<span class="source-line-no">042</span><span id="line-42">import org.apache.hadoop.hbase.namequeues.RpcLogDetails;</span>
<span class="source-line-no">043</span><span id="line-43">import org.apache.hadoop.hbase.namequeues.request.NamedQueueGetRequest;</span>
<span class="source-line-no">044</span><span id="line-44">import org.apache.hadoop.hbase.namequeues.response.NamedQueueGetResponse;</span>
<span class="source-line-no">045</span><span id="line-45">import org.apache.hadoop.hbase.net.Address;</span>
<span class="source-line-no">046</span><span id="line-46">import org.apache.hadoop.hbase.regionserver.RpcSchedulerFactory;</span>
<span class="source-line-no">047</span><span id="line-47">import org.apache.hadoop.hbase.security.User;</span>
<span class="source-line-no">048</span><span id="line-48">import org.apache.hadoop.hbase.security.access.AccessChecker;</span>
<span class="source-line-no">049</span><span id="line-49">import org.apache.hadoop.hbase.security.access.NoopAccessChecker;</span>
<span class="source-line-no">050</span><span id="line-50">import org.apache.hadoop.hbase.security.access.Permission;</span>
<span class="source-line-no">051</span><span id="line-51">import org.apache.hadoop.hbase.security.access.ZKPermissionWatcher;</span>
<span class="source-line-no">052</span><span id="line-52">import org.apache.hadoop.hbase.util.DNS;</span>
<span class="source-line-no">053</span><span id="line-53">import org.apache.hadoop.hbase.util.OOMEChecker;</span>
<span class="source-line-no">054</span><span id="line-54">import org.apache.hadoop.hbase.util.ReservoirSample;</span>
<span class="source-line-no">055</span><span id="line-55">import org.apache.hadoop.hbase.zookeeper.ZKWatcher;</span>
<span class="source-line-no">056</span><span id="line-56">import org.apache.yetus.audience.InterfaceAudience;</span>
<span class="source-line-no">057</span><span id="line-57">import org.apache.zookeeper.KeeperException;</span>
<span class="source-line-no">058</span><span id="line-58">import org.slf4j.Logger;</span>
<span class="source-line-no">059</span><span id="line-59">import org.slf4j.LoggerFactory;</span>
<span class="source-line-no">060</span><span id="line-60"></span>
<span class="source-line-no">061</span><span id="line-61">import org.apache.hbase.thirdparty.com.google.protobuf.ByteString;</span>
<span class="source-line-no">062</span><span id="line-62">import org.apache.hbase.thirdparty.com.google.protobuf.Message;</span>
<span class="source-line-no">063</span><span id="line-63">import org.apache.hbase.thirdparty.com.google.protobuf.RpcController;</span>
<span class="source-line-no">064</span><span id="line-64">import org.apache.hbase.thirdparty.com.google.protobuf.ServiceException;</span>
<span class="source-line-no">065</span><span id="line-65"></span>
<span class="source-line-no">066</span><span id="line-66">import org.apache.hadoop.hbase.shaded.protobuf.ProtobufUtil;</span>
<span class="source-line-no">067</span><span id="line-67">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.AdminService;</span>
<span class="source-line-no">068</span><span id="line-68">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.ClearSlowLogResponseRequest;</span>
<span class="source-line-no">069</span><span id="line-69">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.ClearSlowLogResponses;</span>
<span class="source-line-no">070</span><span id="line-70">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.SlowLogResponseRequest;</span>
<span class="source-line-no">071</span><span id="line-71">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.SlowLogResponses;</span>
<span class="source-line-no">072</span><span id="line-72">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.UpdateConfigurationRequest;</span>
<span class="source-line-no">073</span><span id="line-73">import org.apache.hadoop.hbase.shaded.protobuf.generated.AdminProtos.UpdateConfigurationResponse;</span>
<span class="source-line-no">074</span><span id="line-74">import org.apache.hadoop.hbase.shaded.protobuf.generated.HBaseProtos;</span>
<span class="source-line-no">075</span><span id="line-75">import org.apache.hadoop.hbase.shaded.protobuf.generated.RPCProtos.RequestHeader;</span>
<span class="source-line-no">076</span><span id="line-76">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.ClientMetaService;</span>
<span class="source-line-no">077</span><span id="line-77">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetActiveMasterRequest;</span>
<span class="source-line-no">078</span><span id="line-78">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetActiveMasterResponse;</span>
<span class="source-line-no">079</span><span id="line-79">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetBootstrapNodesRequest;</span>
<span class="source-line-no">080</span><span id="line-80">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetBootstrapNodesResponse;</span>
<span class="source-line-no">081</span><span id="line-81">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetClusterIdRequest;</span>
<span class="source-line-no">082</span><span id="line-82">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetClusterIdResponse;</span>
<span class="source-line-no">083</span><span id="line-83">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetMastersRequest;</span>
<span class="source-line-no">084</span><span id="line-84">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetMastersResponse;</span>
<span class="source-line-no">085</span><span id="line-85">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetMastersResponseEntry;</span>
<span class="source-line-no">086</span><span id="line-86">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetMetaRegionLocationsRequest;</span>
<span class="source-line-no">087</span><span id="line-87">import org.apache.hadoop.hbase.shaded.protobuf.generated.RegistryProtos.GetMetaRegionLocationsResponse;</span>
<span class="source-line-no">088</span><span id="line-88">import org.apache.hadoop.hbase.shaded.protobuf.generated.TooSlowLog.SlowLogPayload;</span>
<span class="source-line-no">089</span><span id="line-89"></span>
<span class="source-line-no">090</span><span id="line-90">/**</span>
<span class="source-line-no">091</span><span id="line-91"> * Base class for Master and RegionServer RpcServices.</span>
<span class="source-line-no">092</span><span id="line-92"> */</span>
<span class="source-line-no">093</span><span id="line-93">@InterfaceAudience.Private</span>
<span class="source-line-no">094</span><span id="line-94">public abstract class HBaseRpcServicesBase&lt;S extends HBaseServerBase&lt;?&gt;&gt;</span>
<span class="source-line-no">095</span><span id="line-95"> implements ClientMetaService.BlockingInterface, AdminService.BlockingInterface,</span>
<span class="source-line-no">096</span><span id="line-96"> HBaseRPCErrorHandler, PriorityFunction, ConfigurationObserver {</span>
<span class="source-line-no">097</span><span id="line-97"></span>
<span class="source-line-no">098</span><span id="line-98"> private static final Logger LOG = LoggerFactory.getLogger(HBaseRpcServicesBase.class);</span>
<span class="source-line-no">099</span><span id="line-99"></span>
<span class="source-line-no">100</span><span id="line-100"> public static final String CLIENT_BOOTSTRAP_NODE_LIMIT = "hbase.client.bootstrap.node.limit";</span>
<span class="source-line-no">101</span><span id="line-101"></span>
<span class="source-line-no">102</span><span id="line-102"> public static final int DEFAULT_CLIENT_BOOTSTRAP_NODE_LIMIT = 10;</span>
<span class="source-line-no">103</span><span id="line-103"></span>
<span class="source-line-no">104</span><span id="line-104"> protected final S server;</span>
<span class="source-line-no">105</span><span id="line-105"></span>
<span class="source-line-no">106</span><span id="line-106"> // Server to handle client requests.</span>
<span class="source-line-no">107</span><span id="line-107"> protected final RpcServer rpcServer;</span>
<span class="source-line-no">108</span><span id="line-108"></span>
<span class="source-line-no">109</span><span id="line-109"> private final InetSocketAddress isa;</span>
<span class="source-line-no">110</span><span id="line-110"></span>
<span class="source-line-no">111</span><span id="line-111"> protected final PriorityFunction priority;</span>
<span class="source-line-no">112</span><span id="line-112"></span>
<span class="source-line-no">113</span><span id="line-113"> private AccessChecker accessChecker;</span>
<span class="source-line-no">114</span><span id="line-114"></span>
<span class="source-line-no">115</span><span id="line-115"> private ZKPermissionWatcher zkPermissionWatcher;</span>
<span class="source-line-no">116</span><span id="line-116"></span>
<span class="source-line-no">117</span><span id="line-117"> protected HBaseRpcServicesBase(S server, String processName) throws IOException {</span>
<span class="source-line-no">118</span><span id="line-118"> this.server = server;</span>
<span class="source-line-no">119</span><span id="line-119"> Configuration conf = server.getConfiguration();</span>
<span class="source-line-no">120</span><span id="line-120"> final RpcSchedulerFactory rpcSchedulerFactory;</span>
<span class="source-line-no">121</span><span id="line-121"> try {</span>
<span class="source-line-no">122</span><span id="line-122"> rpcSchedulerFactory = getRpcSchedulerFactoryClass(conf).asSubclass(RpcSchedulerFactory.class)</span>
<span class="source-line-no">123</span><span id="line-123"> .getDeclaredConstructor().newInstance();</span>
<span class="source-line-no">124</span><span id="line-124"> } catch (NoSuchMethodException | InvocationTargetException | InstantiationException</span>
<span class="source-line-no">125</span><span id="line-125"> | IllegalAccessException e) {</span>
<span class="source-line-no">126</span><span id="line-126"> throw new IllegalArgumentException(e);</span>
<span class="source-line-no">127</span><span id="line-127"> }</span>
<span class="source-line-no">128</span><span id="line-128"> String hostname = DNS.getHostname(conf, getDNSServerType());</span>
<span class="source-line-no">129</span><span id="line-129"> int port = conf.getInt(getPortConfigName(), getDefaultPort());</span>
<span class="source-line-no">130</span><span id="line-130"> // Creation of a HSA will force a resolve.</span>
<span class="source-line-no">131</span><span id="line-131"> final InetSocketAddress initialIsa = new InetSocketAddress(hostname, port);</span>
<span class="source-line-no">132</span><span id="line-132"> final InetSocketAddress bindAddress = new InetSocketAddress(getHostname(conf, hostname), port);</span>
<span class="source-line-no">133</span><span id="line-133"> if (initialIsa.getAddress() == null) {</span>
<span class="source-line-no">134</span><span id="line-134"> throw new IllegalArgumentException("Failed resolve of " + initialIsa);</span>
<span class="source-line-no">135</span><span id="line-135"> }</span>
<span class="source-line-no">136</span><span id="line-136"> priority = createPriority();</span>
<span class="source-line-no">137</span><span id="line-137"> // Using Address means we don't get the IP too. Shorten it more even to just the host name</span>
<span class="source-line-no">138</span><span id="line-138"> // w/o the domain.</span>
<span class="source-line-no">139</span><span id="line-139"> final String name = processName + "/"</span>
<span class="source-line-no">140</span><span id="line-140"> + Address.fromParts(initialIsa.getHostName(), initialIsa.getPort()).toStringWithoutDomain();</span>
<span class="source-line-no">141</span><span id="line-141"> server.setName(name);</span>
<span class="source-line-no">142</span><span id="line-142"> // Set how many times to retry talking to another server over Connection.</span>
<span class="source-line-no">143</span><span id="line-143"> ConnectionUtils.setServerSideHConnectionRetriesConfig(conf, name, LOG);</span>
<span class="source-line-no">144</span><span id="line-144"> boolean reservoirEnabled =</span>
<span class="source-line-no">145</span><span id="line-145"> conf.getBoolean(ByteBuffAllocator.ALLOCATOR_POOL_ENABLED_KEY, defaultReservoirEnabled());</span>
<span class="source-line-no">146</span><span id="line-146"> try {</span>
<span class="source-line-no">147</span><span id="line-147"> // use final bindAddress for this server.</span>
<span class="source-line-no">148</span><span id="line-148"> rpcServer = RpcServerFactory.createRpcServer(server, name, getServices(), bindAddress, conf,</span>
<span class="source-line-no">149</span><span id="line-149"> rpcSchedulerFactory.create(conf, this, server), reservoirEnabled);</span>
<span class="source-line-no">150</span><span id="line-150"> } catch (BindException be) {</span>
<span class="source-line-no">151</span><span id="line-151"> throw new IOException(be.getMessage() + ". To switch ports use the '" + getPortConfigName()</span>
<span class="source-line-no">152</span><span id="line-152"> + "' configuration property.", be.getCause() != null ? be.getCause() : be);</span>
<span class="source-line-no">153</span><span id="line-153"> }</span>
<span class="source-line-no">154</span><span id="line-154"> final InetSocketAddress address = rpcServer.getListenerAddress();</span>
<span class="source-line-no">155</span><span id="line-155"> if (address == null) {</span>
<span class="source-line-no">156</span><span id="line-156"> throw new IOException("Listener channel is closed");</span>
<span class="source-line-no">157</span><span id="line-157"> }</span>
<span class="source-line-no">158</span><span id="line-158"> // Set our address, however we need the final port that was given to rpcServer</span>
<span class="source-line-no">159</span><span id="line-159"> isa = new InetSocketAddress(initialIsa.getHostName(), address.getPort());</span>
<span class="source-line-no">160</span><span id="line-160"> rpcServer.setErrorHandler(this);</span>
<span class="source-line-no">161</span><span id="line-161"> }</span>
<span class="source-line-no">162</span><span id="line-162"></span>
<span class="source-line-no">163</span><span id="line-163"> protected abstract boolean defaultReservoirEnabled();</span>
<span class="source-line-no">164</span><span id="line-164"></span>
<span class="source-line-no">165</span><span id="line-165"> protected abstract DNS.ServerType getDNSServerType();</span>
<span class="source-line-no">166</span><span id="line-166"></span>
<span class="source-line-no">167</span><span id="line-167"> protected abstract String getHostname(Configuration conf, String defaultHostname);</span>
<span class="source-line-no">168</span><span id="line-168"></span>
<span class="source-line-no">169</span><span id="line-169"> protected abstract String getPortConfigName();</span>
<span class="source-line-no">170</span><span id="line-170"></span>
<span class="source-line-no">171</span><span id="line-171"> protected abstract int getDefaultPort();</span>
<span class="source-line-no">172</span><span id="line-172"></span>
<span class="source-line-no">173</span><span id="line-173"> protected abstract PriorityFunction createPriority();</span>
<span class="source-line-no">174</span><span id="line-174"></span>
<span class="source-line-no">175</span><span id="line-175"> protected abstract Class&lt;?&gt; getRpcSchedulerFactoryClass(Configuration conf);</span>
<span class="source-line-no">176</span><span id="line-176"></span>
<span class="source-line-no">177</span><span id="line-177"> protected abstract List&lt;BlockingServiceAndInterface&gt; getServices();</span>
<span class="source-line-no">178</span><span id="line-178"></span>
<span class="source-line-no">179</span><span id="line-179"> protected final void internalStart(ZKWatcher zkWatcher) {</span>
<span class="source-line-no">180</span><span id="line-180"> if (AccessChecker.isAuthorizationSupported(getConfiguration())) {</span>
<span class="source-line-no">181</span><span id="line-181"> accessChecker = new AccessChecker(getConfiguration());</span>
<span class="source-line-no">182</span><span id="line-182"> } else {</span>
<span class="source-line-no">183</span><span id="line-183"> accessChecker = new NoopAccessChecker(getConfiguration());</span>
<span class="source-line-no">184</span><span id="line-184"> }</span>
<span class="source-line-no">185</span><span id="line-185"> zkPermissionWatcher =</span>
<span class="source-line-no">186</span><span id="line-186"> new ZKPermissionWatcher(zkWatcher, accessChecker.getAuthManager(), getConfiguration());</span>
<span class="source-line-no">187</span><span id="line-187"> try {</span>
<span class="source-line-no">188</span><span id="line-188"> zkPermissionWatcher.start();</span>
<span class="source-line-no">189</span><span id="line-189"> } catch (KeeperException e) {</span>
<span class="source-line-no">190</span><span id="line-190"> LOG.error("ZooKeeper permission watcher initialization failed", e);</span>
<span class="source-line-no">191</span><span id="line-191"> }</span>
<span class="source-line-no">192</span><span id="line-192"> rpcServer.start();</span>
<span class="source-line-no">193</span><span id="line-193"> }</span>
<span class="source-line-no">194</span><span id="line-194"></span>
<span class="source-line-no">195</span><span id="line-195"> protected final void requirePermission(String request, Permission.Action perm)</span>
<span class="source-line-no">196</span><span id="line-196"> throws IOException {</span>
<span class="source-line-no">197</span><span id="line-197"> if (accessChecker != null) {</span>
<span class="source-line-no">198</span><span id="line-198"> accessChecker.requirePermission(RpcServer.getRequestUser().orElse(null), request, null, perm);</span>
<span class="source-line-no">199</span><span id="line-199"> }</span>
<span class="source-line-no">200</span><span id="line-200"> }</span>
<span class="source-line-no">201</span><span id="line-201"></span>
<span class="source-line-no">202</span><span id="line-202"> public AccessChecker getAccessChecker() {</span>
<span class="source-line-no">203</span><span id="line-203"> return accessChecker;</span>
<span class="source-line-no">204</span><span id="line-204"> }</span>
<span class="source-line-no">205</span><span id="line-205"></span>
<span class="source-line-no">206</span><span id="line-206"> public ZKPermissionWatcher getZkPermissionWatcher() {</span>
<span class="source-line-no">207</span><span id="line-207"> return zkPermissionWatcher;</span>
<span class="source-line-no">208</span><span id="line-208"> }</span>
<span class="source-line-no">209</span><span id="line-209"></span>
<span class="source-line-no">210</span><span id="line-210"> protected final void internalStop() {</span>
<span class="source-line-no">211</span><span id="line-211"> if (zkPermissionWatcher != null) {</span>
<span class="source-line-no">212</span><span id="line-212"> zkPermissionWatcher.close();</span>
<span class="source-line-no">213</span><span id="line-213"> }</span>
<span class="source-line-no">214</span><span id="line-214"> rpcServer.stop();</span>
<span class="source-line-no">215</span><span id="line-215"> }</span>
<span class="source-line-no">216</span><span id="line-216"></span>
<span class="source-line-no">217</span><span id="line-217"> public Configuration getConfiguration() {</span>
<span class="source-line-no">218</span><span id="line-218"> return server.getConfiguration();</span>
<span class="source-line-no">219</span><span id="line-219"> }</span>
<span class="source-line-no">220</span><span id="line-220"></span>
<span class="source-line-no">221</span><span id="line-221"> public S getServer() {</span>
<span class="source-line-no">222</span><span id="line-222"> return server;</span>
<span class="source-line-no">223</span><span id="line-223"> }</span>
<span class="source-line-no">224</span><span id="line-224"></span>
<span class="source-line-no">225</span><span id="line-225"> public InetSocketAddress getSocketAddress() {</span>
<span class="source-line-no">226</span><span id="line-226"> return isa;</span>
<span class="source-line-no">227</span><span id="line-227"> }</span>
<span class="source-line-no">228</span><span id="line-228"></span>
<span class="source-line-no">229</span><span id="line-229"> public RpcServerInterface getRpcServer() {</span>
<span class="source-line-no">230</span><span id="line-230"> return rpcServer;</span>
<span class="source-line-no">231</span><span id="line-231"> }</span>
<span class="source-line-no">232</span><span id="line-232"></span>
<span class="source-line-no">233</span><span id="line-233"> public RpcScheduler getRpcScheduler() {</span>
<span class="source-line-no">234</span><span id="line-234"> return rpcServer.getScheduler();</span>
<span class="source-line-no">235</span><span id="line-235"> }</span>
<span class="source-line-no">236</span><span id="line-236"></span>
<span class="source-line-no">237</span><span id="line-237"> @Override</span>
<span class="source-line-no">238</span><span id="line-238"> public int getPriority(RequestHeader header, Message param, User user) {</span>
<span class="source-line-no">239</span><span id="line-239"> return priority.getPriority(header, param, user);</span>
<span class="source-line-no">240</span><span id="line-240"> }</span>
<span class="source-line-no">241</span><span id="line-241"></span>
<span class="source-line-no">242</span><span id="line-242"> @Override</span>
<span class="source-line-no">243</span><span id="line-243"> public long getDeadline(RequestHeader header, Message param) {</span>
<span class="source-line-no">244</span><span id="line-244"> return priority.getDeadline(header, param);</span>
<span class="source-line-no">245</span><span id="line-245"> }</span>
<span class="source-line-no">246</span><span id="line-246"></span>
<span class="source-line-no">247</span><span id="line-247"> /**</span>
<span class="source-line-no">248</span><span id="line-248"> * Check if an OOME and, if so, abort immediately to avoid creating more objects.</span>
<span class="source-line-no">249</span><span id="line-249"> * @return True if we OOME'd and are aborting.</span>
<span class="source-line-no">250</span><span id="line-250"> */</span>
<span class="source-line-no">251</span><span id="line-251"> @Override</span>
<span class="source-line-no">252</span><span id="line-252"> public boolean checkOOME(Throwable e) {</span>
<span class="source-line-no">253</span><span id="line-253"> return OOMEChecker.exitIfOOME(e, getClass().getSimpleName());</span>
<span class="source-line-no">254</span><span id="line-254"> }</span>
<span class="source-line-no">255</span><span id="line-255"></span>
<span class="source-line-no">256</span><span id="line-256"> @Override</span>
<span class="source-line-no">257</span><span id="line-257"> public void onConfigurationChange(Configuration conf) {</span>
<span class="source-line-no">258</span><span id="line-258"> rpcServer.onConfigurationChange(conf);</span>
<span class="source-line-no">259</span><span id="line-259"> }</span>
<span class="source-line-no">260</span><span id="line-260"></span>
<span class="source-line-no">261</span><span id="line-261"> @Override</span>
<span class="source-line-no">262</span><span id="line-262"> public GetClusterIdResponse getClusterId(RpcController controller, GetClusterIdRequest request)</span>
<span class="source-line-no">263</span><span id="line-263"> throws ServiceException {</span>
<span class="source-line-no">264</span><span id="line-264"> return GetClusterIdResponse.newBuilder().setClusterId(server.getClusterId()).build();</span>
<span class="source-line-no">265</span><span id="line-265"> }</span>
<span class="source-line-no">266</span><span id="line-266"></span>
<span class="source-line-no">267</span><span id="line-267"> @Override</span>
<span class="source-line-no">268</span><span id="line-268"> public GetActiveMasterResponse getActiveMaster(RpcController controller,</span>
<span class="source-line-no">269</span><span id="line-269"> GetActiveMasterRequest request) throws ServiceException {</span>
<span class="source-line-no">270</span><span id="line-270"> GetActiveMasterResponse.Builder builder = GetActiveMasterResponse.newBuilder();</span>
<span class="source-line-no">271</span><span id="line-271"> server.getActiveMaster()</span>
<span class="source-line-no">272</span><span id="line-272"> .ifPresent(name -&gt; builder.setServerName(ProtobufUtil.toServerName(name)));</span>
<span class="source-line-no">273</span><span id="line-273"> return builder.build();</span>
<span class="source-line-no">274</span><span id="line-274"> }</span>
<span class="source-line-no">275</span><span id="line-275"></span>
<span class="source-line-no">276</span><span id="line-276"> @Override</span>
<span class="source-line-no">277</span><span id="line-277"> public GetMastersResponse getMasters(RpcController controller, GetMastersRequest request)</span>
<span class="source-line-no">278</span><span id="line-278"> throws ServiceException {</span>
<span class="source-line-no">279</span><span id="line-279"> GetMastersResponse.Builder builder = GetMastersResponse.newBuilder();</span>
<span class="source-line-no">280</span><span id="line-280"> server.getActiveMaster()</span>
<span class="source-line-no">281</span><span id="line-281"> .ifPresent(activeMaster -&gt; builder.addMasterServers(GetMastersResponseEntry.newBuilder()</span>
<span class="source-line-no">282</span><span id="line-282"> .setServerName(ProtobufUtil.toServerName(activeMaster)).setIsActive(true)));</span>
<span class="source-line-no">283</span><span id="line-283"> server.getBackupMasters()</span>
<span class="source-line-no">284</span><span id="line-284"> .forEach(backupMaster -&gt; builder.addMasterServers(GetMastersResponseEntry.newBuilder()</span>
<span class="source-line-no">285</span><span id="line-285"> .setServerName(ProtobufUtil.toServerName(backupMaster)).setIsActive(false)));</span>
<span class="source-line-no">286</span><span id="line-286"> return builder.build();</span>
<span class="source-line-no">287</span><span id="line-287"> }</span>
<span class="source-line-no">288</span><span id="line-288"></span>
<span class="source-line-no">289</span><span id="line-289"> @Override</span>
<span class="source-line-no">290</span><span id="line-290"> public GetMetaRegionLocationsResponse getMetaRegionLocations(RpcController controller,</span>
<span class="source-line-no">291</span><span id="line-291"> GetMetaRegionLocationsRequest request) throws ServiceException {</span>
<span class="source-line-no">292</span><span id="line-292"> GetMetaRegionLocationsResponse.Builder builder = GetMetaRegionLocationsResponse.newBuilder();</span>
<span class="source-line-no">293</span><span id="line-293"> server.getMetaLocations()</span>
<span class="source-line-no">294</span><span id="line-294"> .forEach(location -&gt; builder.addMetaLocations(ProtobufUtil.toRegionLocation(location)));</span>
<span class="source-line-no">295</span><span id="line-295"> return builder.build();</span>
<span class="source-line-no">296</span><span id="line-296"> }</span>
<span class="source-line-no">297</span><span id="line-297"></span>
<span class="source-line-no">298</span><span id="line-298"> @Override</span>
<span class="source-line-no">299</span><span id="line-299"> public final GetBootstrapNodesResponse getBootstrapNodes(RpcController controller,</span>
<span class="source-line-no">300</span><span id="line-300"> GetBootstrapNodesRequest request) throws ServiceException {</span>
<span class="source-line-no">301</span><span id="line-301"> int maxNodeCount = server.getConfiguration().getInt(CLIENT_BOOTSTRAP_NODE_LIMIT,</span>
<span class="source-line-no">302</span><span id="line-302"> DEFAULT_CLIENT_BOOTSTRAP_NODE_LIMIT);</span>
<span class="source-line-no">303</span><span id="line-303"> ReservoirSample&lt;ServerName&gt; sample = new ReservoirSample&lt;&gt;(maxNodeCount);</span>
<span class="source-line-no">304</span><span id="line-304"> sample.add(server.getBootstrapNodes());</span>
<span class="source-line-no">305</span><span id="line-305"></span>
<span class="source-line-no">306</span><span id="line-306"> GetBootstrapNodesResponse.Builder builder = GetBootstrapNodesResponse.newBuilder();</span>
<span class="source-line-no">307</span><span id="line-307"> sample.getSamplingResult().stream().map(ProtobufUtil::toServerName)</span>
<span class="source-line-no">308</span><span id="line-308"> .forEach(builder::addServerName);</span>
<span class="source-line-no">309</span><span id="line-309"> return builder.build();</span>
<span class="source-line-no">310</span><span id="line-310"> }</span>
<span class="source-line-no">311</span><span id="line-311"></span>
<span class="source-line-no">312</span><span id="line-312"> @Override</span>
<span class="source-line-no">313</span><span id="line-313"> @QosPriority(priority = HConstants.ADMIN_QOS)</span>
<span class="source-line-no">314</span><span id="line-314"> public UpdateConfigurationResponse updateConfiguration(RpcController controller,</span>
<span class="source-line-no">315</span><span id="line-315"> UpdateConfigurationRequest request) throws ServiceException {</span>
<span class="source-line-no">316</span><span id="line-316"> try {</span>
<span class="source-line-no">317</span><span id="line-317"> requirePermission("updateConfiguration", Permission.Action.ADMIN);</span>
<span class="source-line-no">318</span><span id="line-318"> this.server.updateConfiguration();</span>
<span class="source-line-no">319</span><span id="line-319"> } catch (Exception e) {</span>
<span class="source-line-no">320</span><span id="line-320"> throw new ServiceException(e);</span>
<span class="source-line-no">321</span><span id="line-321"> }</span>
<span class="source-line-no">322</span><span id="line-322"> return UpdateConfigurationResponse.getDefaultInstance();</span>
<span class="source-line-no">323</span><span id="line-323"> }</span>
<span class="source-line-no">324</span><span id="line-324"></span>
<span class="source-line-no">325</span><span id="line-325"> @Override</span>
<span class="source-line-no">326</span><span id="line-326"> @QosPriority(priority = HConstants.ADMIN_QOS)</span>
<span class="source-line-no">327</span><span id="line-327"> public ClearSlowLogResponses clearSlowLogsResponses(final RpcController controller,</span>
<span class="source-line-no">328</span><span id="line-328"> final ClearSlowLogResponseRequest request) throws ServiceException {</span>
<span class="source-line-no">329</span><span id="line-329"> try {</span>
<span class="source-line-no">330</span><span id="line-330"> requirePermission("clearSlowLogsResponses", Permission.Action.ADMIN);</span>
<span class="source-line-no">331</span><span id="line-331"> } catch (IOException e) {</span>
<span class="source-line-no">332</span><span id="line-332"> throw new ServiceException(e);</span>
<span class="source-line-no">333</span><span id="line-333"> }</span>
<span class="source-line-no">334</span><span id="line-334"> final NamedQueueRecorder namedQueueRecorder = this.server.getNamedQueueRecorder();</span>
<span class="source-line-no">335</span><span id="line-335"> boolean slowLogsCleaned = Optional.ofNullable(namedQueueRecorder)</span>
<span class="source-line-no">336</span><span id="line-336"> .map(</span>
<span class="source-line-no">337</span><span id="line-337"> queueRecorder -&gt; queueRecorder.clearNamedQueue(NamedQueuePayload.NamedQueueEvent.SLOW_LOG))</span>
<span class="source-line-no">338</span><span id="line-338"> .orElse(false);</span>
<span class="source-line-no">339</span><span id="line-339"> ClearSlowLogResponses clearSlowLogResponses =</span>
<span class="source-line-no">340</span><span id="line-340"> ClearSlowLogResponses.newBuilder().setIsCleaned(slowLogsCleaned).build();</span>
<span class="source-line-no">341</span><span id="line-341"> return clearSlowLogResponses;</span>
<span class="source-line-no">342</span><span id="line-342"> }</span>
<span class="source-line-no">343</span><span id="line-343"></span>
<span class="source-line-no">344</span><span id="line-344"> private List&lt;SlowLogPayload&gt; getSlowLogPayloads(SlowLogResponseRequest request,</span>
<span class="source-line-no">345</span><span id="line-345"> NamedQueueRecorder namedQueueRecorder) {</span>
<span class="source-line-no">346</span><span id="line-346"> if (namedQueueRecorder == null) {</span>
<span class="source-line-no">347</span><span id="line-347"> return Collections.emptyList();</span>
<span class="source-line-no">348</span><span id="line-348"> }</span>
<span class="source-line-no">349</span><span id="line-349"> List&lt;SlowLogPayload&gt; slowLogPayloads;</span>
<span class="source-line-no">350</span><span id="line-350"> NamedQueueGetRequest namedQueueGetRequest = new NamedQueueGetRequest();</span>
<span class="source-line-no">351</span><span id="line-351"> namedQueueGetRequest.setNamedQueueEvent(RpcLogDetails.SLOW_LOG_EVENT);</span>
<span class="source-line-no">352</span><span id="line-352"> namedQueueGetRequest.setSlowLogResponseRequest(request);</span>
<span class="source-line-no">353</span><span id="line-353"> NamedQueueGetResponse namedQueueGetResponse =</span>
<span class="source-line-no">354</span><span id="line-354"> namedQueueRecorder.getNamedQueueRecords(namedQueueGetRequest);</span>
<span class="source-line-no">355</span><span id="line-355"> slowLogPayloads = namedQueueGetResponse != null</span>
<span class="source-line-no">356</span><span id="line-356"> ? namedQueueGetResponse.getSlowLogPayloads()</span>
<span class="source-line-no">357</span><span id="line-357"> : Collections.emptyList();</span>
<span class="source-line-no">358</span><span id="line-358"> return slowLogPayloads;</span>
<span class="source-line-no">359</span><span id="line-359"> }</span>
<span class="source-line-no">360</span><span id="line-360"></span>
<span class="source-line-no">361</span><span id="line-361"> @Override</span>
<span class="source-line-no">362</span><span id="line-362"> @QosPriority(priority = HConstants.ADMIN_QOS)</span>
<span class="source-line-no">363</span><span id="line-363"> public HBaseProtos.LogEntry getLogEntries(RpcController controller,</span>
<span class="source-line-no">364</span><span id="line-364"> HBaseProtos.LogRequest request) throws ServiceException {</span>
<span class="source-line-no">365</span><span id="line-365"> try {</span>
<span class="source-line-no">366</span><span id="line-366"> final String logClassName = request.getLogClassName();</span>
<span class="source-line-no">367</span><span id="line-367"> Class&lt;?&gt; logClass = Class.forName(logClassName).asSubclass(Message.class);</span>
<span class="source-line-no">368</span><span id="line-368"> Method method = logClass.getMethod("parseFrom", ByteString.class);</span>
<span class="source-line-no">369</span><span id="line-369"> if (logClassName.contains("SlowLogResponseRequest")) {</span>
<span class="source-line-no">370</span><span id="line-370"> SlowLogResponseRequest slowLogResponseRequest =</span>
<span class="source-line-no">371</span><span id="line-371"> (SlowLogResponseRequest) method.invoke(null, request.getLogMessage());</span>
<span class="source-line-no">372</span><span id="line-372"> final NamedQueueRecorder namedQueueRecorder = this.server.getNamedQueueRecorder();</span>
<span class="source-line-no">373</span><span id="line-373"> final List&lt;SlowLogPayload&gt; slowLogPayloads =</span>
<span class="source-line-no">374</span><span id="line-374"> getSlowLogPayloads(slowLogResponseRequest, namedQueueRecorder);</span>
<span class="source-line-no">375</span><span id="line-375"> SlowLogResponses slowLogResponses =</span>
<span class="source-line-no">376</span><span id="line-376"> SlowLogResponses.newBuilder().addAllSlowLogPayloads(slowLogPayloads).build();</span>
<span class="source-line-no">377</span><span id="line-377"> return HBaseProtos.LogEntry.newBuilder()</span>
<span class="source-line-no">378</span><span id="line-378"> .setLogClassName(slowLogResponses.getClass().getName())</span>
<span class="source-line-no">379</span><span id="line-379"> .setLogMessage(slowLogResponses.toByteString()).build();</span>
<span class="source-line-no">380</span><span id="line-380"> }</span>
<span class="source-line-no">381</span><span id="line-381"> } catch (ClassNotFoundException | NoSuchMethodException | IllegalAccessException</span>
<span class="source-line-no">382</span><span id="line-382"> | InvocationTargetException e) {</span>
<span class="source-line-no">383</span><span id="line-383"> LOG.error("Error while retrieving log entries.", e);</span>
<span class="source-line-no">384</span><span id="line-384"> throw new ServiceException(e);</span>
<span class="source-line-no">385</span><span id="line-385"> }</span>
<span class="source-line-no">386</span><span id="line-386"> throw new ServiceException("Invalid request params");</span>
<span class="source-line-no">387</span><span id="line-387"> }</span>
<span class="source-line-no">388</span><span id="line-388">}</span>
</pre>
</div>
</main>
</body>
</html>