/*
* Copyright 2008-2009 the original author or authors.
*
* Licensed 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.
*/
package net.hasor.rsf.address.route.flowcontrol.speed;
import net.hasor.core.Settings;
import net.hasor.rsf.InterAddress;
import net.hasor.rsf.RsfEnvironment;
import net.hasor.rsf.RsfSettings;
import net.hasor.rsf.address.route.rule.AbstractRule;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
/**
* 基于QoS的速率控制规则。
* <pre>
* 配置实例:
* <flowControl enable="true|false" type="speed">
* <action>service|method|address</action>
* <rate>20</rate> <!-- 稳态速率 -->
* <peak>100</peak> <!-- 峰值速率 -->
* <timeWindow>10</timeWindow> <!-- 时间窗口 -->
* </flowControl>
* </pre>
* 解释:根据action的配置决定RPC调用速率。
*/
public class SpeedFlowControl extends AbstractRule {
private QoSActionEnum action;
private int rate = 20;
private int peak = 200;
private int timeWindow = 10;
private QoSBucket defaultQoSBucket;
private ConcurrentMap<String, QoSBucket> qosBucketMap;
//
public void paserControl(Settings settings) {
this.enable(settings.getBoolean("flowControl.enable"));
this.action = settings.getEnum("flowControl.action", QoSActionEnum.class);
this.rate = settings.getInteger("flowControl.rate");
this.peak = settings.getInteger("flowControl.peak");
this.timeWindow = settings.getInteger("flowControl.timeWindow");
this.qosBucketMap = new ConcurrentHashMap<String, QoSBucket>();
//
if (this.action == null) {
this.enable(false);
logger.info("action fail. config is null.");
}
if (!this.enable()) {
return;
}
logger.info("init default QoS.");
QoSBucket qosBucket = this.createQoSBucket();
if (!qosBucket.validate()) {
this.enable(false);
logger.info("QoS config validate fail. -> %s", this.defaultQoSBucket);
return;
}
defaultQoSBucket = qosBucket;
}
//
public boolean callCheck(String serviceID, String methodName, InterAddress doCallAddress) {
if (!this.enable()) {
return true;
}
//
String key = null;
switch (this.action) {
case Address:
key = doCallAddress.toString();
break;
case Method:
key = methodName;
break;
case Service:
key = serviceID;
break;
}
//
if (key == null) {
return true;
}
QoSBucket qos = this.qosBucketMap.get(key);
if (qos == null) {
qos = this.qosBucketMap.putIfAbsent(key, this.createQoSBucket());
qos = this.qosBucketMap.get(key);
}
return qos.check();
}
//
protected QoSBucket createQoSBucket() {
QoSBucket qosBucket = new QoSBucket(this.rate, this.peak, this.timeWindow);
logger.info("create {}", qosBucket);
return qosBucket;
}
//
public static SpeedFlowControl defaultControl(RsfEnvironment rsfEnvironment) {
SpeedFlowControl flowControl = new SpeedFlowControl();
RsfSettings rsfSettings = rsfEnvironment.getSettings();
flowControl.action = rsfSettings.getEnum("hasor.rsfConfig.defaultSpeedFlowControl.action", QoSActionEnum.class);
flowControl.rate = rsfSettings.getInteger("hasor.rsfConfig.defaultSpeedFlowControl.rate");
flowControl.peak = rsfSettings.getInteger("hasor.rsfConfig.defaultSpeedFlowControl.peak");
flowControl.timeWindow = rsfSettings.getInteger("hasor.rsfConfig.defaultSpeedFlowControl.timeWindow");
return flowControl;
}
}