0

我正在开发一个库，我在其中对我的服务进行 Http 调用，如果我的服务机器没有响应（有套接字超时或连接超时），我将它们添加到我的本地blockList，如果机器被阻塞 5 次，那么我不要给他们打电话。

因此，假设如果 machineA没有响应(throwing RestClientException)，我将onFailure每次调用方法并继续增加计数器，然后在再次调用时，我通过作为主机名和 5 作为阈值传递来machineA检查方法，所以如果已被阻止 5 次，那么我不会打电话给他们。我的库是多线程的，所以这就是我在这里使用 volatile 的原因，因为我希望所有线程都看到相同的值。isBlockedmachineAmachineA

以下是我在DataMapping课堂上的内容：

public static volatile ConcurrentHashMap<String, AtomicInteger> blockedHosts =
      new ConcurrentHashMap<String, AtomicInteger>();

boolean isBlocked(String hostname, int threshold) {
    AtomicInteger count = blockedHosts.get(hostname);
    return count != null && count.get() >= threshold;
}

void onFailure(String hostname) {
    AtomicInteger newValue = new AtomicInteger();
    AtomicInteger val = blockedHosts.putIfAbsent(hostname, newValue);
    // no need to care about over-reaching 5 here
    (val == null ? newValue : val).incrementAndGet();
}

void onSuccess(String hostname) {
    blockedHosts.remove(hostname);
}

问题陈述：-

现在我想再添加一个特性——如果machineA被阻塞（因为它的阻塞计数>= 5），那么我想让它阻塞 x 间隔。我将有另一个参数(key.getInterval())，它会告诉我们我想让这台机器阻塞多长时间，在这个间隔过去之后，只有我会开始打电话给他们。我无法理解如何添加此功能？

下面是我的主线程代码，我在其中使用DataMapping方法来检查主机名是否被阻止以及阻止主机名。

@Override
public DataResponse call() {
    ResponseEntity<String> response = null;

    List<String> hostnames = some_code_here;

    for (String hostname : hostnames) {
        // If hostname is in block list, skip sending request to this host
        if (DataMapping.isBlocked(hostname)) {
            continue;
        }
        try {
            String url = createURL(hostname);
            response = restTemplate.exchange(url, HttpMethod.GET, key.getEntity(), String.class);
            DataMapping.onSuccess(hostname);

            // some code here to return the response if successful
        } catch (RestClientException ex) {
            // adding to block list
            DataMapping.onFailure(hostname);
        }
    }

    return new DataResponse(DataErrorEnum.SERVER_UNAVAILABLE, DataStatusEnum.ERROR);        
}

如何在特定时间段内阻止特定机器​​，并且一旦该时间间隔过去，然后才开始调用它们？

4

1 回答 1

1

您可以使用ScheduledExecutorService并schedule在特定超时后重置计数器。

你可以在你的DataMapping类中声明它：

private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor(); // or perhaps the thread pool version ?

在您的onFailure()方法中，您可以决定是要重置还是在某个超时后减少计数器：

void onFailure(String hostname) {
    // you can use `computeIfAbsent` in java8
    AtomicInteger val = blockedHosts.computeIfAbsent(hostname, key -> new AtomicInteger());
    int count = val.incrementAndGet();
    // the test here is `==` to make sure the task is scheduled only once
    if (count == threshold) {
        scheduler.schedule(() -> blockedHosts.remove(hostname), 5L, TimeUnit.MINUTES);  // or you may choose to just decrement the counter
    }
}

作为旁注，没有理由制作blockedHosts volatile. 该参考永远不会改变；应该final改为；并且可能private。

在 java7 中，上面的代码如下所示：

void onFailure(String hostname) {
    AtomicInteger newValue = new AtomicInteger();
    AtomicInteger val = blockedHosts.putIfAbsent(hostname, newValue);
    int count = (val == null ? newValue : val).incrementAndGet();
    // the test here is `==` to make sure the task is scheduled only once
    if (count == threshold) {
        scheduler.schedule(new Runnable() {
            @Override public void run() {
                blockedHosts.remove(hostname);  // or you may choose to just decrement the counter
            }
        }, 5L, TimeUnit.MINUTES);
    }
}
于 2016-05-16T00:07:28.833 回答