2 * Copyright (C) 2016 ZTE, Inc. and others. All rights reserved. (ZTE)
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.openo.msb.wrapper.consul.option;
19 import static com.google.common.base.Preconditions.checkArgument;
20 import static org.openo.msb.wrapper.consul.option.Options.optionallyAdd;
22 import java.math.BigInteger;
24 import javax.ws.rs.client.WebTarget;
26 import com.google.common.base.Optional;
29 * Container for common query options used by the Consul API.
32 public abstract class QueryOptions implements ParamAdder {
34 public static final QueryOptions BLANK = ImmutableQueryOptions.builder().build();
36 public abstract Optional<String> getWait();
37 public abstract Optional<String> getToken();
38 public abstract Optional<BigInteger> getIndex();
39 public abstract Optional<String> getNear();
42 public ConsistencyMode getConsistencyMode() {
43 return ConsistencyMode.DEFAULT;
47 public boolean isBlocking() {
48 return getWait().isPresent();
52 public boolean hasToken() {
53 return getToken().isPresent();
59 checkArgument(getIndex().isPresent(), "If wait is specified, index must also be specified");
63 public static ImmutableQueryOptions.Builder blockSeconds(int seconds, BigInteger index) {
64 return blockBuilder("s", seconds, index);
67 public static ImmutableQueryOptions.Builder blockMinutes(int minutes, BigInteger index) {
68 return blockBuilder("m", minutes, index);
71 private static ImmutableQueryOptions.Builder blockBuilder(String identifier, int qty, BigInteger index) {
72 return ImmutableQueryOptions.builder()
73 .wait(String.format("%s%s", qty, identifier))
78 public WebTarget apply(WebTarget input) {
80 WebTarget added = input;
81 switch (getConsistencyMode()) {
83 added = added.queryParam("consistent", "");
86 added = added.queryParam("stale", "");
91 added = added.queryParam("wait", getWait().get())
92 .queryParam("index", String.valueOf(getIndex().get()));
95 added = optionallyAdd(added, "token", getToken());
96 added = optionallyAdd(added, "near", getToken());