11 changed files with 234 additions and 53 deletions
@ -0,0 +1,29 @@ |
|||
/** |
|||
* Copyright © 2016-2018 The Thingsboard 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 org.thingsboard.server.common.transport.service; |
|||
|
|||
import org.thingsboard.server.common.data.EntityType; |
|||
|
|||
/** |
|||
* Created by ashvayka on 22.10.18. |
|||
*/ |
|||
public class TbRateLimitsException extends Exception { |
|||
private final EntityType entityType; |
|||
|
|||
TbRateLimitsException(EntityType entityType) { |
|||
this.entityType = entityType; |
|||
} |
|||
} |
|||
@ -0,0 +1,53 @@ |
|||
/** |
|||
* Copyright © 2016-2018 The Thingsboard 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 org.thingsboard.server.common.transport.service; |
|||
|
|||
import io.github.bucket4j.Bandwidth; |
|||
import io.github.bucket4j.Bucket4j; |
|||
import io.github.bucket4j.local.LocalBucket; |
|||
import io.github.bucket4j.local.LocalBucketBuilder; |
|||
|
|||
import java.time.Duration; |
|||
|
|||
/** |
|||
* Created by ashvayka on 22.10.18. |
|||
*/ |
|||
class TbTransportRateLimits { |
|||
private final LocalBucket bucket; |
|||
|
|||
public TbTransportRateLimits(String limitsConfiguration) { |
|||
LocalBucketBuilder builder = Bucket4j.builder(); |
|||
boolean initialized = false; |
|||
for (String limitSrc : limitsConfiguration.split(",")) { |
|||
long capacity = Long.parseLong(limitSrc.split(":")[0]); |
|||
long duration = Long.parseLong(limitSrc.split(":")[1]); |
|||
builder.addLimit(Bandwidth.simple(capacity, Duration.ofSeconds(duration))); |
|||
initialized = true; |
|||
} |
|||
if (initialized) { |
|||
bucket = builder.build(); |
|||
} else { |
|||
throw new IllegalArgumentException("Failed to parse rate limits configuration: " + limitsConfiguration); |
|||
} |
|||
|
|||
|
|||
} |
|||
|
|||
boolean tryConsume() { |
|||
return bucket.tryConsume(1); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue