Question

我正在使用Http调用我的服务的库，如果我的服务机器没有响应（有套接字超时或连接超时），我将它们添加到我的本地blockList和如果机器被阻挡5次，那么我就不打电话给他们。

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

以下是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被阻止（因为它的阻塞计数是＆gt; = 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);        
}

如何在特定时间段内阻止特定机器，并且只要该时间间隔过去，然后才开始拨打电话？

Answer 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);
    }
}

在特定时间间隔内阻止部分机器

1 个答案: