View Javadoc
1   /* 
2    * Licensed under the Apache License, Version 2.0 (the "License");
3    * you may not use this file except in compliance with the License.
4    * You may obtain a copy of the License at
5    *
6    * http://www.apache.org/licenses/LICENSE-2.0
7    *
8    * Unless required by applicable law or agreed to in writing, software
9    * distributed under the License is distributed on an "AS IS" BASIS,
10   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11   * See the License for the specific language governing permissions and
12   * limitations under the License.
13   *
14   */
15  
16  package org.esigate.url;
17  
18  import java.util.concurrent.atomic.AtomicInteger;
19  
20  import org.esigate.api.BaseUrlRetrieveStrategy;
21  import org.esigate.http.IncomingRequest;
22  
23  public class RoundRobinBaseUrlRetrieveStrategy implements BaseUrlRetrieveStrategy {
24      private final AtomicInteger counter = new AtomicInteger(0);
25      private final String[] urls;
26  
27      public RoundRobinBaseUrlRetrieveStrategy(String[] urls) {
28          this.urls = urls;
29      }
30  
31      @Override
32      public String getBaseURL(IncomingRequest originalRequest) {
33          int incremented = this.counter.incrementAndGet();
34          int index = incremented % this.urls.length;
35          return this.urls[Math.abs(index)];
36      }
37  }