Merge remote-tracking branch 'upstream/feature_support_grpc_core' into 2.0.0
This commit is contained in:
commit
2c55f4f212
@ -51,16 +51,6 @@
|
||||
</exclusion>
|
||||
</exclusions>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.mockito</groupId>
|
||||
<artifactId>mockito-all</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.hamcrest</groupId>
|
||||
<artifactId>hamcrest-all</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
|
@ -60,9 +60,9 @@ public class ServerListManager implements ServerListFactory, Closeable {
|
||||
|
||||
private final AtomicInteger currentIndex = new AtomicInteger();
|
||||
|
||||
private List<String> serversFromEndpoint = new ArrayList<String>();
|
||||
private final List<String> serverList = new ArrayList<>();
|
||||
|
||||
private List<String> serverList = new ArrayList<String>();
|
||||
private List<String> serversFromEndpoint = new ArrayList<>();
|
||||
|
||||
private ScheduledExecutorService refreshServerListExecutor;
|
||||
|
||||
@ -85,12 +85,9 @@ public class ServerListManager implements ServerListFactory, Closeable {
|
||||
this.serversFromEndpoint = getServerListFromEndpoint();
|
||||
refreshServerListExecutor = new ScheduledThreadPoolExecutor(1,
|
||||
new NameThreadFactory("com.alibaba.nacos.client.naming.server.list.refresher"));
|
||||
refreshServerListExecutor.scheduleWithFixedDelay(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
refreshServerListIfNeed();
|
||||
}
|
||||
}, 0, refreshServerListInternal, TimeUnit.MILLISECONDS);
|
||||
refreshServerListExecutor
|
||||
.scheduleWithFixedDelay(this::refreshServerListIfNeed, 0, refreshServerListInternal,
|
||||
TimeUnit.MILLISECONDS);
|
||||
} else {
|
||||
String serverListFromProps = properties.getProperty(PropertyKeyConst.SERVER_ADDR);
|
||||
if (StringUtils.isNotEmpty(serverListFromProps)) {
|
||||
@ -138,7 +135,7 @@ public class ServerListManager implements ServerListFactory, Closeable {
|
||||
if (CollectionUtils.isEmpty(list)) {
|
||||
throw new Exception("Can not acquire Nacos list");
|
||||
}
|
||||
if (!CollectionUtils.isEqualCollection(list, serversFromEndpoint)) {
|
||||
if (null == serversFromEndpoint || !CollectionUtils.isEqualCollection(list, serversFromEndpoint)) {
|
||||
NAMING_LOGGER.info("[SERVER-LIST] server list is updated: " + list);
|
||||
}
|
||||
serversFromEndpoint = list;
|
||||
@ -156,6 +153,7 @@ public class ServerListManager implements ServerListFactory, Closeable {
|
||||
return nacosDomain;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getServerList() {
|
||||
return serverList.isEmpty() ? serversFromEndpoint : serverList;
|
||||
}
|
||||
|
@ -19,6 +19,7 @@ package com.alibaba.nacos.client.naming.utils;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.ThreadLocalRandom;
|
||||
|
||||
/**
|
||||
* Chooser.
|
||||
|
@ -1,223 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import java.util.Random;
|
||||
|
||||
/**
|
||||
* <p><code>JVMRandom</code> is a wrapper that supports all possible Random methods via the {@link
|
||||
* java.lang.Math#random()} method and its system-wide {@link Random} object.</p>
|
||||
* <p>
|
||||
* It does this to allow for a Random class in which the seed is shared between all members of the class - a better name
|
||||
* would have been SharedSeedRandom.</p>
|
||||
* <p>
|
||||
* <b>N.B.</b> the current implementation overrides the methods {@link Random#nextInt(int)} and {@link
|
||||
* Random#nextLong()} to produce positive numbers ranging from 0 (inclusive) to MAX_VALUE (exclusive).
|
||||
* </p>
|
||||
* @author unknown
|
||||
* @version $Id: JVMRandom.java 911986 2010-02-19 21:19:05Z niallp $
|
||||
* @since 2.0
|
||||
*/
|
||||
public final class JvmRandom extends Random {
|
||||
|
||||
/**
|
||||
* Required for serialization support.
|
||||
*
|
||||
* @see java.io.Serializable
|
||||
*/
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
private static final Random SHARED_RANDOM = new Random();
|
||||
|
||||
/**
|
||||
* Ensures that only the parent constructor can call reseed.
|
||||
*/
|
||||
private boolean constructed = false;
|
||||
|
||||
/**
|
||||
* Constructs a new instance.
|
||||
*/
|
||||
public JvmRandom() {
|
||||
this.constructed = true;
|
||||
}
|
||||
|
||||
/**
|
||||
* Unsupported in 2.0.
|
||||
*
|
||||
* @param seed ignored
|
||||
* @throws UnsupportedOperationException unsupported operation exception
|
||||
*/
|
||||
@Override
|
||||
public synchronized void setSeed(long seed) {
|
||||
if (this.constructed) {
|
||||
throw new UnsupportedOperationException();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Unsupported in 2.0.
|
||||
*
|
||||
* @return Nothing, this method always throws an UnsupportedOperationException.
|
||||
* @throws UnsupportedOperationException unsupported operation exception
|
||||
*/
|
||||
@Override
|
||||
public synchronized double nextGaussian() {
|
||||
throw new UnsupportedOperationException();
|
||||
}
|
||||
|
||||
/**
|
||||
* Unsupported in 2.0.
|
||||
*
|
||||
* @param byteArray ignored
|
||||
* @throws UnsupportedOperationException unsupported operation exception
|
||||
*/
|
||||
@Override
|
||||
public void nextBytes(byte[] byteArray) {
|
||||
throw new UnsupportedOperationException();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed int value from the Math.random() sequence.</p> Identical
|
||||
* to <code>nextInt(Integer.MAX_VALUE)</code> <p> <b>N.B. All values are >= 0.</b> </p>
|
||||
*
|
||||
* @return the random int
|
||||
*/
|
||||
@Override
|
||||
public int nextInt() {
|
||||
return nextInt(Integer.MAX_VALUE);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns a pseudorandom, uniformly distributed int value between <code>0</code> (inclusive) and the specified
|
||||
* value (exclusive), from the Math.random() sequence.</p>
|
||||
*
|
||||
* @param n the specified exclusive max-value
|
||||
* @return the random int
|
||||
* @throws IllegalArgumentException when <code>n <= 0</code>
|
||||
*/
|
||||
@Override
|
||||
public int nextInt(int n) {
|
||||
return SHARED_RANDOM.nextInt(n);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed long value from the Math.random() sequence.</p>
|
||||
* Identical
|
||||
* to <code>nextLong(Long.MAX_VALUE)</code> <p> <b>N.B. All values are >= 0.</b> </p>
|
||||
*
|
||||
* @return the random long
|
||||
*/
|
||||
@Override
|
||||
public long nextLong() {
|
||||
return nextLong(Long.MAX_VALUE);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns a pseudorandom, uniformly distributed long value between <code>0</code> (inclusive) and the specified
|
||||
* value (exclusive), from the Math.random() sequence.</p>
|
||||
*
|
||||
* @param n the specified exclusive max-value
|
||||
* @return the random long
|
||||
* @throws IllegalArgumentException when <code>n <= 0</code>
|
||||
*/
|
||||
public static long nextLong(long n) {
|
||||
if (n <= 0) {
|
||||
throw new IllegalArgumentException("Upper bound for nextInt must be positive");
|
||||
}
|
||||
// Code adapted from Harmony Random#nextInt(int)
|
||||
// n is power of 2
|
||||
if ((n & -n) == n) {
|
||||
// dropping lower order bits improves behaviour for low values of n
|
||||
return next63bits() >> 63 - bitsRequired(n - 1);
|
||||
}
|
||||
// Not a power of two
|
||||
long val;
|
||||
long bits;
|
||||
// reject some values to improve distribution
|
||||
do {
|
||||
bits = next63bits();
|
||||
val = bits % n;
|
||||
} while (bits - val + (n - 1) < 0);
|
||||
return val;
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed boolean value from the Math.random() sequence.</p>
|
||||
*
|
||||
* @return the random boolean
|
||||
*/
|
||||
@Override
|
||||
public boolean nextBoolean() {
|
||||
return SHARED_RANDOM.nextBoolean();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed float value between <code>0.0</code> and
|
||||
* <code>1.0</code>
|
||||
* from the Math.random() sequence.</p>
|
||||
*
|
||||
* @return the random float
|
||||
*/
|
||||
@Override
|
||||
public float nextFloat() {
|
||||
return SHARED_RANDOM.nextFloat();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Synonymous to the Math.random() call.</p>
|
||||
*
|
||||
* @return the random double
|
||||
*/
|
||||
@Override
|
||||
public double nextDouble() {
|
||||
return SHARED_RANDOM.nextDouble();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the next unsigned random long.
|
||||
*
|
||||
* @return unsigned random long
|
||||
*/
|
||||
private static long next63bits() {
|
||||
// drop the sign bit to leave 63 random bits
|
||||
return SHARED_RANDOM.nextLong() & 0x7fffffffffffffffL;
|
||||
}
|
||||
|
||||
/**
|
||||
* Count the number of bits required to represent a long number.
|
||||
*
|
||||
* @param num long number
|
||||
* @return number of bits required
|
||||
*/
|
||||
private static int bitsRequired(long num) {
|
||||
// Derived from Hacker's Delight, Figure 5-9
|
||||
long y = num;
|
||||
int n = 0;
|
||||
while (true) {
|
||||
// 64 = number of bits in a long
|
||||
if (num < 0) {
|
||||
return 64 - n;
|
||||
}
|
||||
if (y == 0) {
|
||||
return n;
|
||||
}
|
||||
n++;
|
||||
num = num << 1;
|
||||
y = y >> 1;
|
||||
}
|
||||
}
|
||||
}
|
@ -1,171 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import java.util.Random;
|
||||
|
||||
/**
|
||||
* <code>RandomUtils</code> is a wrapper that supports all possible {@link java.util.Random} methods via the {@link
|
||||
* java.lang.Math#random()} method and its system-wide <code>Random</code> object.
|
||||
*
|
||||
* @author Gary D. Gregory
|
||||
* @version $Id: RandomUtils.java 906320 2010-02-04 01:41:10Z sebb $
|
||||
* @since 2.0
|
||||
*/
|
||||
public class RandomUtils {
|
||||
|
||||
/**
|
||||
* An instance of {@link JvmRandom}.
|
||||
*/
|
||||
private static final Random JVM_RANDOM = new JvmRandom();
|
||||
|
||||
// should be possible for JVM_RANDOM?
|
||||
// public static void nextBytes(byte[]) {
|
||||
// public synchronized double nextGaussian();
|
||||
// }
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed int value from the Math.random() sequence.</p> <b>N.B.
|
||||
* All values are >= 0.</b>
|
||||
*
|
||||
* @return the random int
|
||||
*/
|
||||
public static int nextInt() {
|
||||
return nextInt(JVM_RANDOM);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed int value from the given <code>random</code>
|
||||
* sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @return the random int
|
||||
*/
|
||||
public static int nextInt(Random random) {
|
||||
return random.nextInt();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns a pseudorandom, uniformly distributed int value between <code>0</code> (inclusive) and the specified
|
||||
* value (exclusive), from the Math.random() sequence.</p>
|
||||
*
|
||||
* @param n the specified exclusive max-value
|
||||
* @return the random int
|
||||
*/
|
||||
public static int nextInt(int n) {
|
||||
return nextInt(JVM_RANDOM, n);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns a pseudorandom, uniformly distributed int value between <code>0</code> (inclusive) and the specified
|
||||
* value (exclusive), from the given Random sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @param n the specified exclusive max-value
|
||||
* @return the random int
|
||||
*/
|
||||
public static int nextInt(Random random, int n) {
|
||||
// check this cannot return 'n'
|
||||
return random.nextInt(n);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed long value from the Math.random() sequence.</p> <b>N.B.
|
||||
* All values are >= 0.</b>
|
||||
*
|
||||
* @return the random long
|
||||
*/
|
||||
public static long nextLong() {
|
||||
return nextLong(JVM_RANDOM);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed long value from the given Random sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @return the random long
|
||||
*/
|
||||
public static long nextLong(Random random) {
|
||||
return random.nextLong();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed boolean value from the Math.random() sequence.</p>
|
||||
*
|
||||
* @return the random boolean
|
||||
*/
|
||||
public static boolean nextBoolean() {
|
||||
return nextBoolean(JVM_RANDOM);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed boolean value from the given random sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @return the random boolean
|
||||
*/
|
||||
public static boolean nextBoolean(Random random) {
|
||||
return random.nextBoolean();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed float value between <code>0.0</code> and
|
||||
* <code>1.0</code>
|
||||
* from the Math.random() sequence.</p>
|
||||
*
|
||||
* @return the random float
|
||||
*/
|
||||
public static float nextFloat() {
|
||||
return nextFloat(JVM_RANDOM);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed float value between <code>0.0</code> and
|
||||
* <code>1.0</code>
|
||||
* from the given Random sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @return the random float
|
||||
*/
|
||||
public static float nextFloat(Random random) {
|
||||
return random.nextFloat();
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed float value between <code>0.0</code> and
|
||||
* <code>1.0</code>
|
||||
* from the Math.random() sequence.</p>
|
||||
*
|
||||
* @return the random double
|
||||
*/
|
||||
public static double nextDouble() {
|
||||
return nextDouble(JVM_RANDOM);
|
||||
}
|
||||
|
||||
/**
|
||||
* <p>Returns the next pseudorandom, uniformly distributed float value between <code>0.0</code> and
|
||||
* <code>1.0</code>
|
||||
* from the given Random sequence.</p>
|
||||
*
|
||||
* @param random the Random sequence generator.
|
||||
* @return the random double
|
||||
*/
|
||||
public static double nextDouble(Random random) {
|
||||
return random.nextDouble();
|
||||
}
|
||||
|
||||
}
|
@ -1,279 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import java.security.SecureRandom;
|
||||
import java.util.Random;
|
||||
import java.util.concurrent.BlockingQueue;
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.concurrent.atomic.AtomicLong;
|
||||
|
||||
/**
|
||||
* A random number generator isolated to the current thread. Like the global {@link java.util.Random} generator used by
|
||||
* the {@link java.lang.Math} class, a {@code ThreadLocalRandom} is initialized with an internally generated seed that
|
||||
* may not otherwise be modified. When applicable, use of {@code ThreadLocalRandom} rather than shared {@code Random}
|
||||
* objects in concurrent programs will typically encounter much less overhead and contention. Use of {@code
|
||||
* ThreadLocalRandom} is particularly appropriate when multiple tasks (for example, each a {@link
|
||||
* io.netty.util.internal.chmv8.ForkJoinTask}) use random numbers in parallel in thread pools.
|
||||
* <p>
|
||||
* Usages of this class should typically be of the form: {@code ThreadLocalRandom.current().nextX(...)} (where {@code X}
|
||||
* is {@code Int}, {@code Long}, etc). When all usages are of this form, it is never possible to accidently share a
|
||||
* {@code ThreadLocalRandom} across multiple threads.
|
||||
* </p>
|
||||
* <p>
|
||||
* This class also provides additional commonly used bounded random generation methods.
|
||||
* </p>
|
||||
* //since 1.7 //author Doug Lea
|
||||
*/
|
||||
@SuppressWarnings("all")
|
||||
public class ThreadLocalRandom extends Random {
|
||||
|
||||
private static final AtomicLong seedUniquifier = new AtomicLong();
|
||||
|
||||
private static volatile long initialSeedUniquifier;
|
||||
|
||||
public static void setInitialSeedUniquifier(long initialSeedUniquifier) {
|
||||
ThreadLocalRandom.initialSeedUniquifier = initialSeedUniquifier;
|
||||
}
|
||||
|
||||
public static synchronized long getInitialSeedUniquifier() {
|
||||
// Use the value set via the setter.
|
||||
long initialSeedUniquifier = ThreadLocalRandom.initialSeedUniquifier;
|
||||
// Otherwise, generate one.
|
||||
if (initialSeedUniquifier == 0) {
|
||||
// Try to generate a real random number from /dev/random.
|
||||
// Get from a different thread to avoid blocking indefinitely on a machine without much entrophy.
|
||||
final BlockingQueue<Long> queue = new LinkedBlockingQueue<Long>();
|
||||
Thread generatorThread = new Thread("initialSeedUniquifierGenerator") {
|
||||
@Override
|
||||
public void run() {
|
||||
SecureRandom random = new SecureRandom(); // Get the real random seed from /dev/random
|
||||
queue.add(random.nextLong());
|
||||
}
|
||||
};
|
||||
generatorThread.start();
|
||||
|
||||
// Get the random seed from the thread with timeout.
|
||||
final long timeoutSeconds = 3;
|
||||
final long deadLine = System.nanoTime() + TimeUnit.SECONDS.toNanos(timeoutSeconds);
|
||||
for (; ; ) {
|
||||
long waitTime = deadLine - System.nanoTime();
|
||||
if (waitTime <= 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
try {
|
||||
Long result = queue.poll(waitTime, TimeUnit.NANOSECONDS);
|
||||
if (result != null) {
|
||||
initialSeedUniquifier = result;
|
||||
break;
|
||||
}
|
||||
} catch (InterruptedException ignore) {
|
||||
// Ignore
|
||||
}
|
||||
}
|
||||
|
||||
// Just in case the initialSeedUniquifier is zero or some other constant
|
||||
initialSeedUniquifier ^= 0x3255ecdc33bae119L; // just a meaningless random number
|
||||
initialSeedUniquifier ^= Long.reverse(System.nanoTime());
|
||||
|
||||
ThreadLocalRandom.initialSeedUniquifier = initialSeedUniquifier;
|
||||
}
|
||||
|
||||
return initialSeedUniquifier;
|
||||
}
|
||||
|
||||
private static long newSeed() {
|
||||
for (; ; ) {
|
||||
final long current = seedUniquifier.get();
|
||||
final long actualCurrent = current != 0 ? current : getInitialSeedUniquifier();
|
||||
|
||||
// L'Ecuyer, "Tables of Linear Congruential Generators of Different Sizes and Good Lattice Structure", 1999
|
||||
final long next = actualCurrent * 181783497276652981L;
|
||||
|
||||
if (seedUniquifier.compareAndSet(current, next)) {
|
||||
return next ^ System.nanoTime();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// same constants as Random, but must be redeclared because private
|
||||
private static final long multiplier = 0x5DEECE66DL;
|
||||
|
||||
private static final long addend = 0xBL;
|
||||
|
||||
private static final long mask = (1L << 48) - 1;
|
||||
|
||||
/**
|
||||
* The random seed. We can't use super.seed.
|
||||
*/
|
||||
private long rnd;
|
||||
|
||||
/**
|
||||
* Initialization flag to permit calls to setSeed to succeed only while executing the Random constructor. We can't
|
||||
* allow others since it would cause setting seed in one part of a program to unintentionally impact other usages by
|
||||
* the thread.
|
||||
*/
|
||||
boolean initialized = false;
|
||||
|
||||
// Padding to help avoid memory contention among seed updates in
|
||||
// different TLRs in the common case that they are located near
|
||||
// each other.
|
||||
private long pad0, pad1, pad2, pad3, pad4, pad5, pad6, pad7;
|
||||
|
||||
/**
|
||||
* Constructor called only by localRandom.initialValue.
|
||||
*/
|
||||
ThreadLocalRandom() {
|
||||
super(newSeed());
|
||||
initialized = true;
|
||||
}
|
||||
|
||||
/**
|
||||
* The actual ThreadLocal
|
||||
*/
|
||||
private static final ThreadLocal<ThreadLocalRandom> localRandom = new ThreadLocal<ThreadLocalRandom>() {
|
||||
@Override
|
||||
protected ThreadLocalRandom initialValue() {
|
||||
return new ThreadLocalRandom();
|
||||
}
|
||||
};
|
||||
|
||||
/**
|
||||
* Returns the current thread's {@code ThreadLocalRandom}.
|
||||
*
|
||||
* @return the current thread's {@code ThreadLocalRandom}
|
||||
*/
|
||||
public static ThreadLocalRandom current() {
|
||||
return localRandom.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Throws {@code UnsupportedOperationException}. Setting seeds in this generator is not supported.
|
||||
*
|
||||
* @throws UnsupportedOperationException always
|
||||
*/
|
||||
@Override
|
||||
public void setSeed(long seed) {
|
||||
if (initialized) {
|
||||
throw new UnsupportedOperationException();
|
||||
}
|
||||
rnd = (seed ^ multiplier) & mask;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected int next(int bits) {
|
||||
rnd = (rnd * multiplier + addend) & mask;
|
||||
return (int) (rnd >>> (48 - bits));
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a pseudorandom, uniformly distributed value between the given least value (inclusive) and bound
|
||||
* (exclusive).
|
||||
*
|
||||
* @param least the least value returned
|
||||
* @param bound the upper bound (exclusive)
|
||||
* @return the next value
|
||||
* @throws IllegalArgumentException if least greater than or equal to bound
|
||||
*/
|
||||
public int nextInt(int least, int bound) {
|
||||
if (least >= bound) {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
return nextInt(bound - least) + least;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a pseudorandom, uniformly distributed value between 0 (inclusive) and the specified value (exclusive).
|
||||
*
|
||||
* @param n the bound on the random number to be returned. Must be positive.
|
||||
* @return the next value
|
||||
* @throws IllegalArgumentException if n is not positive
|
||||
*/
|
||||
public long nextLong(long n) {
|
||||
if (n <= 0) {
|
||||
throw new IllegalArgumentException("n must be positive");
|
||||
}
|
||||
|
||||
// Divide n by two until small enough for nextInt. On each
|
||||
// iteration (at most 31 of them but usually much less),
|
||||
// randomly choose both whether to include high bit in result
|
||||
// (offset) and whether to continue with the lower vs upper
|
||||
// half (which makes a difference only if odd).
|
||||
long offset = 0;
|
||||
while (n >= Integer.MAX_VALUE) {
|
||||
int bits = next(2);
|
||||
long half = n >>> 1;
|
||||
long nextn = ((bits & 2) == 0) ? half : n - half;
|
||||
if ((bits & 1) == 0) {
|
||||
offset += n - nextn;
|
||||
}
|
||||
n = nextn;
|
||||
}
|
||||
return offset + nextInt((int) n);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a pseudorandom, uniformly distributed value between the given least value (inclusive) and bound
|
||||
* (exclusive).
|
||||
*
|
||||
* @param least the least value returned
|
||||
* @param bound the upper bound (exclusive)
|
||||
* @return the next value
|
||||
* @throws IllegalArgumentException if least greater than or equal to bound
|
||||
*/
|
||||
public long nextLong(long least, long bound) {
|
||||
if (least >= bound) {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
return nextLong(bound - least) + least;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a pseudorandom, uniformly distributed {@code double} value between 0 (inclusive) and the specified value
|
||||
* (exclusive).
|
||||
*
|
||||
* @param n the bound on the random number to be returned. Must be positive.
|
||||
* @return the next value
|
||||
* @throws IllegalArgumentException if n is not positive
|
||||
*/
|
||||
public double nextDouble(double n) {
|
||||
if (n <= 0) {
|
||||
throw new IllegalArgumentException("n must be positive");
|
||||
}
|
||||
return nextDouble() * n;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a pseudorandom, uniformly distributed value between the given least value (inclusive) and bound
|
||||
* (exclusive).
|
||||
*
|
||||
* @param least the least value returned
|
||||
* @param bound the upper bound (exclusive)
|
||||
* @return the next value
|
||||
* @throws IllegalArgumentException if least greater than or equal to bound
|
||||
*/
|
||||
public double nextDouble(double least, double bound) {
|
||||
if (least >= bound) {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
return nextDouble() * (bound - least) + least;
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = -5851777807851030925L;
|
||||
}
|
@ -1,68 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client;
|
||||
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.client.naming.beat.BeatInfo;
|
||||
import com.alibaba.nacos.client.naming.beat.BeatReactor;
|
||||
import com.alibaba.nacos.client.naming.remote.http.NamingHttpClientProxy;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.HashMap;
|
||||
import java.util.concurrent.ScheduledThreadPoolExecutor;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class BeatReactorTest {
|
||||
|
||||
@Mock
|
||||
private NamingHttpClientProxy namingHttpClientProxy;
|
||||
|
||||
@Test
|
||||
public void test() throws NoSuchFieldException, IllegalAccessException, InterruptedException, NacosException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
beatInfo.setIp("11.11.11.11");
|
||||
beatInfo.setPort(1234);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(1000L);
|
||||
|
||||
BeatReactor beatReactor = new BeatReactor(namingHttpClientProxy);
|
||||
beatReactor.addBeatInfo("testService", beatInfo);
|
||||
|
||||
Assert.assertEquals(1, getActiveThread(beatReactor));
|
||||
Thread.sleep(1100L);
|
||||
beatReactor.removeBeatInfo("testService", beatInfo.getIp(), beatInfo.getPort());
|
||||
Thread.sleep(3100L);
|
||||
Assert.assertEquals(0, getActiveThread(beatReactor));
|
||||
}
|
||||
|
||||
private int getActiveThread(BeatReactor beatReactor) throws NoSuchFieldException, IllegalAccessException {
|
||||
Field field = BeatReactor.class.getDeclaredField("executorService");
|
||||
field.setAccessible(true);
|
||||
ScheduledThreadPoolExecutor scheduledExecutorService = (ScheduledThreadPoolExecutor) field.get(beatReactor);
|
||||
return scheduledExecutorService.getQueue().size();
|
||||
}
|
||||
|
||||
}
|
@ -0,0 +1,77 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.backups;
|
||||
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import com.alibaba.nacos.client.naming.cache.ServiceInfoHolder;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.Date;
|
||||
import java.util.HashMap;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
|
||||
public class FailoverReactorTest {
|
||||
|
||||
@Test
|
||||
public void testInit() throws NacosException, NoSuchFieldException, IllegalAccessException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
Mockito.when(holder.getServiceInfoMap()).thenReturn(new HashMap<>());
|
||||
FailoverReactor failoverReactor = new FailoverReactor(holder, "/tmp");
|
||||
Field executorService = FailoverReactor.class.getDeclaredField("executorService");
|
||||
executorService.setAccessible(true);
|
||||
ScheduledExecutorService o = (ScheduledExecutorService) executorService.get(failoverReactor);
|
||||
Assert.assertFalse(o.isShutdown());
|
||||
failoverReactor.shutdown();
|
||||
Assert.assertTrue(o.isShutdown());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAddDay() throws NacosException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
Mockito.when(holder.getServiceInfoMap()).thenReturn(new HashMap<>());
|
||||
FailoverReactor failoverReactor = new FailoverReactor(holder, "/tmp");
|
||||
Date date = new Date();
|
||||
Date actual = failoverReactor.addDay(date, 1);
|
||||
Assert.assertEquals(date.getTime() + 24 * 60 * 60 * 1000, actual.getTime());
|
||||
failoverReactor.shutdown();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsFailoverSwitch() throws NacosException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
Mockito.when(holder.getServiceInfoMap()).thenReturn(new HashMap<>());
|
||||
FailoverReactor failoverReactor = new FailoverReactor(holder, "/tmp");
|
||||
Assert.assertFalse(failoverReactor.isFailoverSwitch());
|
||||
failoverReactor.shutdown();
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetService() throws NacosException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
Mockito.when(holder.getServiceInfoMap()).thenReturn(new HashMap<>());
|
||||
FailoverReactor failoverReactor = new FailoverReactor(holder, "/tmp");
|
||||
ServiceInfo info = failoverReactor.getService("aa@@bb");
|
||||
Assert.assertEquals(new ServiceInfo("aa@@bb").toString(), info.toString());
|
||||
failoverReactor.shutdown();
|
||||
|
||||
}
|
||||
}
|
@ -0,0 +1,89 @@
|
||||
/*
|
||||
*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.beat;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
public class BeatInfoTest {
|
||||
|
||||
@Test
|
||||
public void testGetterAndSetter() {
|
||||
BeatInfo info = new BeatInfo();
|
||||
|
||||
String ip = "1.1.1.1";
|
||||
info.setIp(ip);
|
||||
int port = 10000;
|
||||
info.setPort(port);
|
||||
|
||||
double weight = 1.0;
|
||||
info.setWeight(weight);
|
||||
String serviceName = "serviceName";
|
||||
info.setServiceName(serviceName);
|
||||
String cluster = "cluster1";
|
||||
info.setCluster(cluster);
|
||||
Map<String, String> meta = new HashMap<>();
|
||||
meta.put("a", "b");
|
||||
info.setMetadata(meta);
|
||||
long period = 100;
|
||||
info.setPeriod(period);
|
||||
info.setScheduled(true);
|
||||
info.setStopped(true);
|
||||
|
||||
Assert.assertEquals(ip, info.getIp());
|
||||
Assert.assertEquals(port, info.getPort());
|
||||
Assert.assertEquals(weight, info.getWeight(), 0.1);
|
||||
Assert.assertEquals(serviceName, info.getServiceName());
|
||||
Assert.assertEquals(meta, info.getMetadata());
|
||||
Assert.assertEquals(period, info.getPeriod());
|
||||
Assert.assertTrue(info.isScheduled());
|
||||
Assert.assertTrue(info.isStopped());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testToString() {
|
||||
BeatInfo info = new BeatInfo();
|
||||
|
||||
String ip = "1.1.1.1";
|
||||
info.setIp(ip);
|
||||
int port = 10000;
|
||||
info.setPort(port);
|
||||
|
||||
double weight = 1.0;
|
||||
info.setWeight(weight);
|
||||
String serviceName = "serviceName";
|
||||
info.setServiceName(serviceName);
|
||||
String cluster = "cluster1";
|
||||
info.setCluster(cluster);
|
||||
Map<String, String> meta = new HashMap<>();
|
||||
meta.put("a", "b");
|
||||
info.setMetadata(meta);
|
||||
long period = 100;
|
||||
info.setPeriod(period);
|
||||
info.setScheduled(true);
|
||||
info.setStopped(true);
|
||||
String expect = "BeatInfo{port=10000, ip='1.1.1.1', " + "weight=1.0, serviceName='serviceName',"
|
||||
+ " cluster='cluster1', metadata={a=b}," + " scheduled=true, period=100, stopped=true}";
|
||||
Assert.assertEquals(expect, info.toString());
|
||||
}
|
||||
|
||||
}
|
@ -0,0 +1,313 @@
|
||||
/*
|
||||
*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.beat;
|
||||
|
||||
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||
import com.alibaba.nacos.api.common.Constants;
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.api.naming.utils.NamingUtils;
|
||||
import com.alibaba.nacos.client.naming.remote.http.NamingHttpClientProxy;
|
||||
import com.fasterxml.jackson.core.JsonProcessingException;
|
||||
import com.fasterxml.jackson.databind.JsonNode;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.Properties;
|
||||
import java.util.concurrent.ScheduledThreadPoolExecutor;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
public class BeatReactorTest {
|
||||
|
||||
@Test
|
||||
public void testConstruct() throws NoSuchFieldException, IllegalAccessException {
|
||||
Properties properties = new Properties();
|
||||
String threadSize = "10";
|
||||
properties.put(PropertyKeyConst.NAMING_CLIENT_BEAT_THREAD_COUNT, threadSize);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy, properties);
|
||||
Field field = BeatReactor.class.getDeclaredField("executorService");
|
||||
field.setAccessible(true);
|
||||
ScheduledThreadPoolExecutor scheduledExecutorService = (ScheduledThreadPoolExecutor) field.get(beatReactor);
|
||||
Assert.assertEquals(Integer.valueOf(threadSize).intValue(), scheduledExecutorService.getCorePoolSize());
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAddBeatInfo() throws NacosException, InterruptedException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
beatInfo.setIp("11.11.11.11");
|
||||
beatInfo.setPort(1234);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(10L);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
beatReactor.addBeatInfo(serviceName, beatInfo);
|
||||
TimeUnit.MILLISECONDS.sleep(15);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRemoveBeatInfo() throws InterruptedException, NacosException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
String ip = "11.11.11.11";
|
||||
beatInfo.setIp(ip);
|
||||
int port = 1234;
|
||||
beatInfo.setPort(port);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(10L);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
beatReactor.addBeatInfo(serviceName, beatInfo);
|
||||
TimeUnit.MILLISECONDS.sleep(15);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
beatReactor.removeBeatInfo(serviceName, ip, port);
|
||||
Assert.assertTrue(beatInfo.isStopped());
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
TimeUnit.MILLISECONDS.sleep(10);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBuildBeatInfo1() {
|
||||
String ip = "11.11.11.11";
|
||||
int port = 1234;
|
||||
double weight = 1.0;
|
||||
String serviceName = "service@@group1";
|
||||
String clusterName = "cluster1";
|
||||
|
||||
Map<String, String> meta = new HashMap<>();
|
||||
Instance instance = new Instance();
|
||||
instance.setServiceName(serviceName);
|
||||
instance.setIp(ip);
|
||||
instance.setPort(port);
|
||||
instance.setWeight(weight);
|
||||
instance.setMetadata(meta);
|
||||
instance.setClusterName(clusterName);
|
||||
|
||||
BeatInfo expectInfo = new BeatInfo();
|
||||
expectInfo.setServiceName(serviceName);
|
||||
expectInfo.setIp(ip);
|
||||
expectInfo.setPort(port);
|
||||
expectInfo.setCluster(clusterName);
|
||||
expectInfo.setWeight(weight);
|
||||
expectInfo.setMetadata(meta);
|
||||
expectInfo.setScheduled(false);
|
||||
expectInfo.setPeriod(Constants.DEFAULT_HEART_BEAT_INTERVAL);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
assertBeatInfoEquals(expectInfo, beatReactor.buildBeatInfo(instance));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBuildBeatInfo2() {
|
||||
String ip = "11.11.11.11";
|
||||
int port = 1234;
|
||||
double weight = 1.0;
|
||||
String serviceName = "service";
|
||||
String clusterName = "cluster1";
|
||||
|
||||
Map<String, String> meta = new HashMap<>();
|
||||
Instance instance = new Instance();
|
||||
instance.setServiceName(serviceName);
|
||||
instance.setIp(ip);
|
||||
instance.setPort(port);
|
||||
instance.setWeight(weight);
|
||||
instance.setMetadata(meta);
|
||||
instance.setClusterName(clusterName);
|
||||
|
||||
String groupedService = "group1@@service";
|
||||
|
||||
BeatInfo expectInfo = new BeatInfo();
|
||||
expectInfo.setServiceName(groupedService);
|
||||
expectInfo.setIp(ip);
|
||||
expectInfo.setPort(port);
|
||||
expectInfo.setCluster(clusterName);
|
||||
expectInfo.setWeight(weight);
|
||||
expectInfo.setMetadata(meta);
|
||||
expectInfo.setScheduled(false);
|
||||
expectInfo.setPeriod(Constants.DEFAULT_HEART_BEAT_INTERVAL);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
assertBeatInfoEquals(expectInfo, beatReactor.buildBeatInfo(groupedService, instance));
|
||||
}
|
||||
|
||||
void assertBeatInfoEquals(BeatInfo expect, BeatInfo actual) {
|
||||
Assert.assertEquals(expect.getCluster(), actual.getCluster());
|
||||
Assert.assertEquals(expect.getIp(), actual.getIp());
|
||||
Assert.assertEquals(expect.getMetadata(), actual.getMetadata());
|
||||
Assert.assertEquals(expect.getPeriod(), actual.getPeriod());
|
||||
Assert.assertEquals(expect.getPort(), actual.getPort());
|
||||
Assert.assertEquals(expect.getServiceName(), actual.getServiceName());
|
||||
Assert.assertEquals(expect.getWeight(), actual.getWeight(), 0.1);
|
||||
Assert.assertEquals(expect.isStopped(), actual.isStopped());
|
||||
Assert.assertEquals(expect.isScheduled(), actual.isScheduled());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBuildKey() {
|
||||
String ip = "11.11.11.11";
|
||||
int port = 1234;
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
Assert.assertEquals(
|
||||
serviceName + Constants.NAMING_INSTANCE_ID_SPLITTER + ip + Constants.NAMING_INSTANCE_ID_SPLITTER + port,
|
||||
beatReactor.buildKey(serviceName, ip, port));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShutdown() throws NacosException, NoSuchFieldException, IllegalAccessException {
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
Field field = BeatReactor.class.getDeclaredField("executorService");
|
||||
field.setAccessible(true);
|
||||
ScheduledThreadPoolExecutor scheduledExecutorService = (ScheduledThreadPoolExecutor) field.get(beatReactor);
|
||||
|
||||
Assert.assertFalse(scheduledExecutorService.isShutdown());
|
||||
beatReactor.shutdown();
|
||||
Assert.assertTrue(scheduledExecutorService.isShutdown());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLightBeatFromResponse() throws InterruptedException, NacosException, JsonProcessingException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
beatInfo.setIp("11.11.11.11");
|
||||
beatInfo.setPort(1234);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(10L);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
String jsonString = "{\"lightBeatEnabled\":true,\"clientBeatInterval\":10}";
|
||||
ObjectMapper mapper = new ObjectMapper();
|
||||
JsonNode actualObj = mapper.readTree(jsonString);
|
||||
|
||||
Mockito.when(proxy.sendBeat(beatInfo, false)).thenReturn(actualObj);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
beatReactor.addBeatInfo(serviceName, beatInfo);
|
||||
TimeUnit.MILLISECONDS.sleep(12);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
TimeUnit.MILLISECONDS.sleep(12);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, true);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIntervalFromResponse() throws JsonProcessingException, NacosException, InterruptedException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
beatInfo.setIp("11.11.11.11");
|
||||
beatInfo.setPort(1234);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(10L);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
String jsonString = "{\"clientBeatInterval\":20}";
|
||||
ObjectMapper mapper = new ObjectMapper();
|
||||
JsonNode actualObj = mapper.readTree(jsonString);
|
||||
|
||||
Mockito.when(proxy.sendBeat(beatInfo, false)).thenReturn(actualObj);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
beatReactor.addBeatInfo(serviceName, beatInfo);
|
||||
TimeUnit.MILLISECONDS.sleep(12);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
TimeUnit.MILLISECONDS.sleep(20);
|
||||
Mockito.verify(proxy, Mockito.times(2)).sendBeat(beatInfo, false);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNotFoundFromResponse() throws JsonProcessingException, NacosException, InterruptedException {
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName("test");
|
||||
beatInfo.setIp("11.11.11.11");
|
||||
beatInfo.setPort(1234);
|
||||
beatInfo.setCluster("clusterName");
|
||||
beatInfo.setWeight(1);
|
||||
beatInfo.setMetadata(new HashMap<String, String>());
|
||||
beatInfo.setScheduled(false);
|
||||
beatInfo.setPeriod(10L);
|
||||
|
||||
NamingHttpClientProxy proxy = Mockito.mock(NamingHttpClientProxy.class);
|
||||
String jsonString = "{\"clientBeatInterval\":10,\"code\":20404}";
|
||||
ObjectMapper mapper = new ObjectMapper();
|
||||
JsonNode actualObj = mapper.readTree(jsonString);
|
||||
|
||||
Mockito.when(proxy.sendBeat(beatInfo, false)).thenReturn(actualObj);
|
||||
|
||||
Mockito.when(proxy.sendBeat(beatInfo, false)).thenReturn(actualObj);
|
||||
BeatReactor beatReactor = new BeatReactor(proxy);
|
||||
String serviceName = "serviceName1";
|
||||
|
||||
beatReactor.addBeatInfo(serviceName, beatInfo);
|
||||
TimeUnit.MILLISECONDS.sleep(11);
|
||||
Mockito.verify(proxy, Mockito.times(1)).sendBeat(beatInfo, false);
|
||||
|
||||
Instance instance = new Instance();
|
||||
instance.setPort(beatInfo.getPort());
|
||||
instance.setIp(beatInfo.getIp());
|
||||
instance.setWeight(beatInfo.getWeight());
|
||||
instance.setMetadata(beatInfo.getMetadata());
|
||||
instance.setClusterName(beatInfo.getCluster());
|
||||
instance.setServiceName(beatInfo.getServiceName());
|
||||
instance.setInstanceId(null);
|
||||
instance.setEphemeral(true);
|
||||
|
||||
Mockito.verify(proxy, Mockito.times(1))
|
||||
.registerService(beatInfo.getServiceName(), NamingUtils.getGroupName(beatInfo.getServiceName()),
|
||||
instance);
|
||||
}
|
||||
|
||||
}
|
60
client/src/test/java/com/alibaba/nacos/client/naming/cache/ConcurrentDiskUtilTest.java
vendored
Normal file
60
client/src/test/java/com/alibaba/nacos/client/naming/cache/ConcurrentDiskUtilTest.java
vendored
Normal file
@ -0,0 +1,60 @@
|
||||
/*
|
||||
*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.cache;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.nio.ByteBuffer;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
public class ConcurrentDiskUtilTest {
|
||||
|
||||
@Test
|
||||
public void testReadAndWrite() throws IOException {
|
||||
File tempFile = File.createTempFile("aaa", "bbb");
|
||||
String fileName = tempFile.getAbsolutePath();
|
||||
String content = "hello";
|
||||
String charset = "UTF-8";
|
||||
ConcurrentDiskUtil.writeFileContent(fileName, content, charset);
|
||||
String actualContent = ConcurrentDiskUtil.getFileContent(fileName, charset);
|
||||
Assert.assertEquals(content, actualContent);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReadAndWrite2() throws IOException {
|
||||
File tempFile = File.createTempFile("aaa", "bbb");
|
||||
String content = "hello";
|
||||
String charset = "UTF-8";
|
||||
ConcurrentDiskUtil.writeFileContent(tempFile, content, charset);
|
||||
String actualContent = ConcurrentDiskUtil.getFileContent(tempFile, charset);
|
||||
Assert.assertEquals(content, actualContent);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testByteBufferToString() throws IOException {
|
||||
String msg = "test buff to string";
|
||||
ByteBuffer buff = ByteBuffer.wrap(msg.getBytes(StandardCharsets.UTF_8));
|
||||
String actual = ConcurrentDiskUtil.byteBufferToString(buff, "UTF-8");
|
||||
Assert.assertEquals(msg, actual);
|
||||
|
||||
}
|
||||
}
|
@ -19,6 +19,7 @@ package com.alibaba.nacos.client.naming.cache;
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import org.junit.After;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
@ -39,7 +40,7 @@ public class DiskCacheTest {
|
||||
private Instance instance;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
public void setUp() {
|
||||
System.out.println(CACHE_DIR);
|
||||
serviceInfo = new ServiceInfo("G@@testName", "testClusters");
|
||||
instance = new Instance();
|
||||
@ -94,4 +95,10 @@ public class DiskCacheTest {
|
||||
assertEquals(actual.getPort(), expected.getPort());
|
||||
assertEquals(actual.getMetadata(), expected.getMetadata());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetLineSeparator() {
|
||||
String lineSeparator = DiskCache.getLineSeparator();
|
||||
Assert.assertTrue(lineSeparator.length() > 0);
|
||||
}
|
||||
}
|
||||
|
151
client/src/test/java/com/alibaba/nacos/client/naming/cache/ServiceInfoHolderTest.java
vendored
Normal file
151
client/src/test/java/com/alibaba/nacos/client/naming/cache/ServiceInfoHolderTest.java
vendored
Normal file
@ -0,0 +1,151 @@
|
||||
/*
|
||||
*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.cache;
|
||||
|
||||
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import com.alibaba.nacos.client.naming.backups.FailoverReactor;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Properties;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
|
||||
public class ServiceInfoHolderTest {
|
||||
|
||||
@Test
|
||||
public void testGetServiceInfoMap() {
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
Assert.assertEquals(0, holder.getServiceInfoMap().size());
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcessServiceInfo() {
|
||||
ServiceInfo info = new ServiceInfo("a@@b@@c");
|
||||
Instance instance1 = createInstance("1.1.1.1", 1);
|
||||
Instance instance2 = createInstance("1.1.1.2", 2);
|
||||
List<Instance> hosts = new ArrayList<>();
|
||||
hosts.add(instance1);
|
||||
hosts.add(instance2);
|
||||
info.setHosts(hosts);
|
||||
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
|
||||
ServiceInfo actual1 = holder.processServiceInfo(info);
|
||||
Assert.assertEquals(info, actual1);
|
||||
|
||||
Instance newInstance1 = createInstance("1.1.1.1", 1);
|
||||
newInstance1.setWeight(2.0);
|
||||
Instance instance3 = createInstance("1.1.1.3", 3);
|
||||
List<Instance> hosts2 = new ArrayList<>();
|
||||
hosts2.add(newInstance1);
|
||||
hosts2.add(instance3);
|
||||
ServiceInfo info2 = new ServiceInfo("a@@b@@c");
|
||||
info2.setHosts(hosts2);
|
||||
|
||||
ServiceInfo actual2 = holder.processServiceInfo(info2);
|
||||
Assert.assertEquals(info2, actual2);
|
||||
}
|
||||
|
||||
private Instance createInstance(String ip, int port) {
|
||||
Instance instance = new Instance();
|
||||
instance.setIp(ip);
|
||||
instance.setPort(port);
|
||||
return instance;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcessServiceInfo2() {
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
String json = "{\"groupName\":\"a\",\"name\":\"b\",\"clusters\":\"c\"}";
|
||||
|
||||
ServiceInfo actual = holder.processServiceInfo(json);
|
||||
ServiceInfo expect = new ServiceInfo("a@@b@@c");
|
||||
expect.setJsonFromServer(json);
|
||||
Assert.assertEquals(expect.getKey(), actual.getKey());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcessServiceInfoWithPushEmpty() {
|
||||
ServiceInfo oldInfo = new ServiceInfo("a@@b@@c");
|
||||
Instance instance1 = createInstance("1.1.1.1", 1);
|
||||
Instance instance2 = createInstance("1.1.1.2", 2);
|
||||
List<Instance> hosts = new ArrayList<>();
|
||||
hosts.add(instance1);
|
||||
hosts.add(instance2);
|
||||
oldInfo.setHosts(hosts);
|
||||
|
||||
Properties prop = new Properties();
|
||||
prop.setProperty(PropertyKeyConst.NAMING_PUSH_EMPTY_PROTECTION, "true");
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
holder.processServiceInfo(oldInfo);
|
||||
|
||||
ServiceInfo newInfo = new ServiceInfo("a@@b@@c");
|
||||
|
||||
final ServiceInfo actual = holder.processServiceInfo(newInfo);
|
||||
|
||||
Assert.assertEquals(oldInfo.getKey(), actual.getKey());
|
||||
Assert.assertEquals(2, actual.getHosts().size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetServiceInfo() {
|
||||
ServiceInfo info = new ServiceInfo("a@@b@@c");
|
||||
Instance instance1 = createInstance("1.1.1.1", 1);
|
||||
List<Instance> hosts = new ArrayList<>();
|
||||
hosts.add(instance1);
|
||||
info.setHosts(hosts);
|
||||
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
|
||||
ServiceInfo expect = holder.processServiceInfo(info);
|
||||
String serviceName = "b";
|
||||
String groupName = "a";
|
||||
String clusters = "c";
|
||||
ServiceInfo actual = holder.getServiceInfo(serviceName, groupName, clusters);
|
||||
Assert.assertEquals(expect.getKey(), actual.getKey());
|
||||
Assert.assertEquals(expect.getHosts().size(), actual.getHosts().size());
|
||||
Assert.assertEquals(expect.getHosts().get(0), actual.getHosts().get(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShutdown() throws NacosException, NoSuchFieldException, IllegalAccessException {
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = new ServiceInfoHolder("aa", prop);
|
||||
Field field = ServiceInfoHolder.class.getDeclaredField("failoverReactor");
|
||||
field.setAccessible(true);
|
||||
FailoverReactor reactor = (FailoverReactor) field.get(holder);
|
||||
Field executorService = FailoverReactor.class.getDeclaredField("executorService");
|
||||
executorService.setAccessible(true);
|
||||
ScheduledExecutorService pool = (ScheduledExecutorService) executorService.get(reactor);
|
||||
Assert.assertFalse(pool.isShutdown());
|
||||
holder.shutdown();
|
||||
Assert.assertTrue(pool.isShutdown());
|
||||
}
|
||||
}
|
@ -0,0 +1,69 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.core;
|
||||
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Rule;
|
||||
import org.junit.Test;
|
||||
import org.junit.rules.ExpectedException;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
public class BalancerTest {
|
||||
|
||||
@Rule
|
||||
public final ExpectedException thrown = ExpectedException.none();
|
||||
|
||||
@Test
|
||||
public void testGetHostByRandomWeightNull() {
|
||||
Assert.assertNull(Balancer.getHostByRandomWeight(null));
|
||||
Assert.assertNull(Balancer.getHostByRandomWeight(new ArrayList<>()));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetHostByRandomWeight() {
|
||||
List<Instance> list = new ArrayList<>();
|
||||
Instance instance1 = new Instance();
|
||||
list.add(instance1);
|
||||
final Instance actual = Balancer.getHostByRandomWeight(list);
|
||||
Assert.assertEquals(instance1, actual);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSelectHost() {
|
||||
List<Instance> hosts = new ArrayList<>();
|
||||
Instance instance1 = new Instance();
|
||||
hosts.add(instance1);
|
||||
ServiceInfo serviceInfo = new ServiceInfo();
|
||||
serviceInfo.setHosts(hosts);
|
||||
|
||||
final Instance actual = Balancer.RandomByWeight.selectHost(serviceInfo);
|
||||
Assert.assertEquals(instance1, actual);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSelectHostEmpty() {
|
||||
thrown.expect(IllegalStateException.class);
|
||||
thrown.expectMessage("no host to srv for serviceInfo: null");
|
||||
ServiceInfo serviceInfo = new ServiceInfo();
|
||||
|
||||
Balancer.RandomByWeight.selectHost(serviceInfo);
|
||||
}
|
||||
}
|
@ -0,0 +1,37 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.core;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
public class ProtectModeTest {
|
||||
|
||||
@Test
|
||||
public void testProtectThresholdDefault() {
|
||||
final ProtectMode protectMode = new ProtectMode();
|
||||
Assert.assertEquals(0.8f, protectMode.getProtectThreshold(), 0.01f);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSetProtectThreshold() {
|
||||
final ProtectMode protectMode = new ProtectMode();
|
||||
float expect = 0.7f;
|
||||
protectMode.setProtectThreshold(expect);
|
||||
Assert.assertEquals(expect, protectMode.getProtectThreshold(), 0.01f);
|
||||
}
|
||||
}
|
@ -0,0 +1,153 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.core;
|
||||
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.client.naming.cache.ServiceInfoHolder;
|
||||
import com.alibaba.nacos.common.utils.IoUtils;
|
||||
import com.alibaba.nacos.common.utils.JacksonUtils;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Field;
|
||||
import java.net.DatagramPacket;
|
||||
import java.net.DatagramSocket;
|
||||
import java.net.InetAddress;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.concurrent.ExecutorService;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
public class PushReceiverTest {
|
||||
|
||||
@Test
|
||||
public void testTestRunDomAndService() throws InterruptedException, IOException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
final PushReceiver pushReceiver = new PushReceiver(holder);
|
||||
final ExecutorService executorService = Executors.newFixedThreadPool(1);
|
||||
executorService.submit(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
pushReceiver.run();
|
||||
}
|
||||
});
|
||||
TimeUnit.MILLISECONDS.sleep(10);
|
||||
|
||||
PushReceiver.PushPacket pack1 = new PushReceiver.PushPacket();
|
||||
pack1.type = "dom";
|
||||
pack1.data = "pack1";
|
||||
pack1.lastRefTime = 1;
|
||||
final String res1 = udpClientRun(pack1, pushReceiver);
|
||||
Assert.assertEquals("{\"type\": \"push-ack\", \"lastRefTime\":\"1\", \"data\":\"\"}", res1);
|
||||
verify(holder, times(1)).processServiceInfo(pack1.data);
|
||||
|
||||
PushReceiver.PushPacket pack2 = new PushReceiver.PushPacket();
|
||||
pack2.type = "service";
|
||||
pack2.data = "pack2";
|
||||
pack2.lastRefTime = 2;
|
||||
final String res2 = udpClientRun(pack2, pushReceiver);
|
||||
Assert.assertEquals("{\"type\": \"push-ack\", \"lastRefTime\":\"2\", \"data\":\"\"}", res2);
|
||||
verify(holder, times(1)).processServiceInfo(pack2.data);
|
||||
|
||||
}
|
||||
|
||||
private String udpClientRun(PushReceiver.PushPacket pack, PushReceiver pushReceiver) throws IOException {
|
||||
final int udpPort = pushReceiver.getUdpPort();
|
||||
String json = JacksonUtils.toJson(pack);
|
||||
final byte[] bytes = IoUtils.tryCompress(json, "UTF-8");
|
||||
final DatagramSocket datagramSocket = new DatagramSocket();
|
||||
datagramSocket.send(new DatagramPacket(bytes, bytes.length, InetAddress.getByName("localhost"), udpPort));
|
||||
byte[] buffer = new byte[20480];
|
||||
final DatagramPacket datagramPacket = new DatagramPacket(buffer, buffer.length);
|
||||
datagramSocket.receive(datagramPacket);
|
||||
final byte[] data = datagramPacket.getData();
|
||||
String res = new String(data, StandardCharsets.UTF_8);
|
||||
return res.trim();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testTestRunWithDump() throws InterruptedException, IOException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
final PushReceiver pushReceiver = new PushReceiver(holder);
|
||||
final ExecutorService executorService = Executors.newFixedThreadPool(1);
|
||||
executorService.submit(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
pushReceiver.run();
|
||||
}
|
||||
});
|
||||
TimeUnit.MILLISECONDS.sleep(10);
|
||||
|
||||
PushReceiver.PushPacket pack1 = new PushReceiver.PushPacket();
|
||||
pack1.type = "dump";
|
||||
pack1.data = "pack1";
|
||||
pack1.lastRefTime = 1;
|
||||
final String res1 = udpClientRun(pack1, pushReceiver);
|
||||
Assert.assertEquals("{\"type\": \"dump-ack\", \"lastRefTime\": \"1\", \"data\":\"{}\"}", res1);
|
||||
verify(holder, times(1)).getServiceInfoMap();
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testTestRunWithUnknown() throws InterruptedException, IOException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
final PushReceiver pushReceiver = new PushReceiver(holder);
|
||||
final ExecutorService executorService = Executors.newFixedThreadPool(1);
|
||||
executorService.submit(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
pushReceiver.run();
|
||||
}
|
||||
});
|
||||
TimeUnit.MILLISECONDS.sleep(10);
|
||||
|
||||
PushReceiver.PushPacket pack1 = new PushReceiver.PushPacket();
|
||||
pack1.type = "unknown";
|
||||
pack1.data = "pack1";
|
||||
pack1.lastRefTime = 1;
|
||||
final String res1 = udpClientRun(pack1, pushReceiver);
|
||||
Assert.assertEquals("{\"type\": \"unknown-ack\", \"lastRefTime\":\"1\", \"data\":\"\"}", res1);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShutdown() throws NacosException, NoSuchFieldException, IllegalAccessException {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
final PushReceiver pushReceiver = new PushReceiver(holder);
|
||||
|
||||
pushReceiver.shutdown();
|
||||
|
||||
final Field closed = PushReceiver.class.getDeclaredField("closed");
|
||||
closed.setAccessible(true);
|
||||
final boolean o = (boolean) closed.get(pushReceiver);
|
||||
Assert.assertTrue(o);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetUdpPort() {
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
final PushReceiver pushReceiver = new PushReceiver(holder);
|
||||
final int udpPort = pushReceiver.getUdpPort();
|
||||
System.out.println("udpPort = " + udpPort);
|
||||
Assert.assertTrue(udpPort > 0);
|
||||
}
|
||||
}
|
@ -0,0 +1,100 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.core;
|
||||
|
||||
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||
import com.alibaba.nacos.common.http.HttpRestResult;
|
||||
import com.alibaba.nacos.common.http.client.NacosRestTemplate;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Ignore;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.List;
|
||||
import java.util.Properties;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
|
||||
public class ServerListManagerTest {
|
||||
|
||||
@Test
|
||||
public void testConstructWithAddr() {
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.SERVER_ADDR, "127.0.0.1:8848,127.0.0.1:8849");
|
||||
final ServerListManager serverListManager = new ServerListManager(properties);
|
||||
final List<String> serverList = serverListManager.getServerList();
|
||||
Assert.assertEquals(2, serverList.size());
|
||||
Assert.assertEquals("127.0.0.1:8848", serverList.get(0));
|
||||
Assert.assertEquals("127.0.0.1:8849", serverList.get(1));
|
||||
}
|
||||
|
||||
@Ignore
|
||||
@Test
|
||||
public void testConstructWithEndpoint() throws Exception {
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.ENDPOINT, "127.0.0.1");
|
||||
final ServerListManager serverListManager = new ServerListManager(properties);
|
||||
NacosRestTemplate mock = Mockito.mock(NacosRestTemplate.class);
|
||||
|
||||
HttpRestResult<Object> a = new HttpRestResult<Object>();
|
||||
a.setData("127.0.0.1:8848");
|
||||
a.setCode(200);
|
||||
Mockito.when(mock.get(any(), any(), any(), any())).thenReturn(a);
|
||||
|
||||
final Field nacosRestTemplate = ServerListManager.class.getDeclaredField("nacosRestTemplate");
|
||||
nacosRestTemplate.setAccessible(true);
|
||||
nacosRestTemplate.set(serverListManager, mock);
|
||||
TimeUnit.SECONDS.sleep(31);
|
||||
final List<String> serverList = serverListManager.getServerList();
|
||||
Assert.assertEquals(1, serverList.size());
|
||||
Assert.assertEquals("127.0.0.1:8848", serverList.get(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsDomain() throws IOException {
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.SERVER_ADDR, "127.0.0.1:8848");
|
||||
final ServerListManager serverListManager = new ServerListManager(properties);
|
||||
Assert.assertTrue(serverListManager.isDomain());
|
||||
Assert.assertEquals("127.0.0.1:8848", serverListManager.getNacosDomain());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetCurrentServer() {
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.SERVER_ADDR, "127.0.0.1:8848");
|
||||
final ServerListManager serverListManager = new ServerListManager(properties);
|
||||
Assert.assertEquals("127.0.0.1:8848", serverListManager.getCurrentServer());
|
||||
Assert.assertEquals("127.0.0.1:8848", serverListManager.genNextServer());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShutdown() {
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.SERVER_ADDR, "127.0.0.1:8848");
|
||||
final ServerListManager serverListManager = new ServerListManager(properties);
|
||||
try {
|
||||
serverListManager.shutdown();
|
||||
} catch (Exception e) {
|
||||
Assert.fail();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
@ -0,0 +1,80 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.core;
|
||||
|
||||
import com.alibaba.nacos.api.exception.NacosException;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import com.alibaba.nacos.client.naming.cache.ServiceInfoHolder;
|
||||
import com.alibaba.nacos.client.naming.event.InstancesChangeNotifier;
|
||||
import com.alibaba.nacos.client.naming.remote.NamingClientProxy;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.util.Properties;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
public class ServiceInfoUpdateServiceTest {
|
||||
|
||||
@Test
|
||||
public void testScheduleUpdateIfAbsent() throws InterruptedException, NacosException {
|
||||
String serviceName = "aa";
|
||||
String group = "bb";
|
||||
String clusters = "cc";
|
||||
ServiceInfo info = new ServiceInfo();
|
||||
info.setName(serviceName);
|
||||
info.setGroupName(group);
|
||||
info.setClusters(clusters);
|
||||
info.setLastRefTime(System.currentTimeMillis());
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
NamingClientProxy proxy = Mockito.mock(NamingClientProxy.class);
|
||||
Mockito.when(proxy.queryInstancesOfService(serviceName, group, clusters, 0, false)).thenReturn(info);
|
||||
|
||||
InstancesChangeNotifier notifyer = Mockito.mock(InstancesChangeNotifier.class);
|
||||
Properties prop = new Properties();
|
||||
final ServiceInfoUpdateService serviceInfoUpdateService = new ServiceInfoUpdateService(prop, holder, proxy,
|
||||
notifyer);
|
||||
|
||||
serviceInfoUpdateService.scheduleUpdateIfAbsent("aa", "bb", "cc");
|
||||
TimeUnit.SECONDS.sleep(2);
|
||||
Mockito.verify(proxy).queryInstancesOfService(serviceName, group, clusters, 0, false);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testStopUpdateIfContain() throws NacosException {
|
||||
String serviceName = "aa";
|
||||
String group = "bb";
|
||||
String clusters = "cc";
|
||||
ServiceInfo info = new ServiceInfo();
|
||||
info.setName(serviceName);
|
||||
info.setGroupName(group);
|
||||
info.setClusters(clusters);
|
||||
info.setLastRefTime(System.currentTimeMillis());
|
||||
NamingClientProxy proxy = Mockito.mock(NamingClientProxy.class);
|
||||
Mockito.when(proxy.queryInstancesOfService(serviceName, group, clusters, 0, false)).thenReturn(info);
|
||||
|
||||
InstancesChangeNotifier notifyer = Mockito.mock(InstancesChangeNotifier.class);
|
||||
Properties prop = new Properties();
|
||||
ServiceInfoHolder holder = Mockito.mock(ServiceInfoHolder.class);
|
||||
|
||||
final ServiceInfoUpdateService serviceInfoUpdateService = new ServiceInfoUpdateService(prop, holder, proxy,
|
||||
notifyer);
|
||||
serviceInfoUpdateService.scheduleUpdateIfAbsent(serviceName, group, clusters);
|
||||
|
||||
serviceInfoUpdateService.stopUpdateIfContain(serviceName, group, clusters);
|
||||
serviceInfoUpdateService.shutdown();
|
||||
}
|
||||
}
|
@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.event;
|
||||
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
public class InstancesChangeEventTest {
|
||||
|
||||
@Test
|
||||
public void testGetServiceName() {
|
||||
String serviceName = "a";
|
||||
String groupName = "b";
|
||||
String clusters = "c";
|
||||
List<Instance> hosts = new ArrayList<>();
|
||||
Instance ins = new Instance();
|
||||
hosts.add(ins);
|
||||
InstancesChangeEvent event = new InstancesChangeEvent(serviceName, groupName, clusters, hosts);
|
||||
Assert.assertEquals(serviceName, event.getServiceName());
|
||||
Assert.assertEquals(clusters, event.getClusters());
|
||||
Assert.assertEquals(groupName, event.getGroupName());
|
||||
List<Instance> hosts1 = event.getHosts();
|
||||
Assert.assertEquals(hosts.size(), hosts1.size());
|
||||
Assert.assertEquals(hosts.get(0), hosts1.get(0));
|
||||
}
|
||||
}
|
@ -0,0 +1,101 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.event;
|
||||
|
||||
import com.alibaba.nacos.api.naming.listener.EventListener;
|
||||
import com.alibaba.nacos.api.naming.pojo.ServiceInfo;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.Mockito.times;
|
||||
|
||||
public class InstancesChangeNotifierTest {
|
||||
|
||||
@Test
|
||||
public void testRegisterListener() {
|
||||
String group = "a";
|
||||
String name = "b";
|
||||
String clusters = "c";
|
||||
InstancesChangeNotifier instancesChangeNotifier = new InstancesChangeNotifier();
|
||||
EventListener listener = Mockito.mock(EventListener.class);
|
||||
instancesChangeNotifier.registerListener(group, name, clusters, listener);
|
||||
List<ServiceInfo> subscribeServices = instancesChangeNotifier.getSubscribeServices();
|
||||
Assert.assertEquals(1, subscribeServices.size());
|
||||
Assert.assertEquals(group, subscribeServices.get(0).getGroupName());
|
||||
Assert.assertEquals(name, subscribeServices.get(0).getName());
|
||||
Assert.assertEquals(clusters, subscribeServices.get(0).getClusters());
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDeregisterListener() {
|
||||
String group = "a";
|
||||
String name = "b";
|
||||
String clusters = "c";
|
||||
InstancesChangeNotifier instancesChangeNotifier = new InstancesChangeNotifier();
|
||||
EventListener listener = Mockito.mock(EventListener.class);
|
||||
instancesChangeNotifier.registerListener(group, name, clusters, listener);
|
||||
List<ServiceInfo> subscribeServices = instancesChangeNotifier.getSubscribeServices();
|
||||
Assert.assertEquals(1, subscribeServices.size());
|
||||
|
||||
instancesChangeNotifier.deregisterListener(group, name, clusters, listener);
|
||||
|
||||
List<ServiceInfo> subscribeServices2 = instancesChangeNotifier.getSubscribeServices();
|
||||
Assert.assertEquals(1, subscribeServices2.size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsSubscribed() {
|
||||
String group = "a";
|
||||
String name = "b";
|
||||
String clusters = "c";
|
||||
InstancesChangeNotifier instancesChangeNotifier = new InstancesChangeNotifier();
|
||||
EventListener listener = Mockito.mock(EventListener.class);
|
||||
Assert.assertFalse(instancesChangeNotifier.isSubscribed(group, name, clusters));
|
||||
|
||||
instancesChangeNotifier.registerListener(group, name, clusters, listener);
|
||||
Assert.assertTrue(instancesChangeNotifier.isSubscribed(group, name, clusters));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOnEvent() {
|
||||
String group = "a";
|
||||
String name = "b";
|
||||
String clusters = "c";
|
||||
InstancesChangeNotifier instancesChangeNotifier = new InstancesChangeNotifier();
|
||||
EventListener listener = Mockito.mock(EventListener.class);
|
||||
|
||||
instancesChangeNotifier.registerListener(group, name, clusters, listener);
|
||||
InstancesChangeEvent event1 = Mockito.mock(InstancesChangeEvent.class);
|
||||
Mockito.when(event1.getClusters()).thenReturn(clusters);
|
||||
Mockito.when(event1.getGroupName()).thenReturn(group);
|
||||
Mockito.when(event1.getServiceName()).thenReturn(name);
|
||||
|
||||
instancesChangeNotifier.onEvent(event1);
|
||||
Mockito.verify(listener, times(1)).onEvent(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSubscribeType() {
|
||||
InstancesChangeNotifier instancesChangeNotifier = new InstancesChangeNotifier();
|
||||
Assert.assertEquals(InstancesChangeEvent.class, instancesChangeNotifier.subscribeType());
|
||||
}
|
||||
}
|
@ -0,0 +1,63 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
public class CollectionUtilsTest {
|
||||
|
||||
@Test
|
||||
public void testSubtract() {
|
||||
List<String> subtract = (List<String>) CollectionUtils
|
||||
.subtract(Arrays.asList("a", "b"), Arrays.asList("b", "c"));
|
||||
Assert.assertEquals(1, subtract.size());
|
||||
Assert.assertEquals("a", subtract.get(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetCardinalityMap() {
|
||||
List<String> list1 = Arrays.asList("2", "2", "3");
|
||||
Map<String, Integer> map1 = CollectionUtils.getCardinalityMap(list1);
|
||||
Assert.assertEquals(2, map1.size());
|
||||
Assert.assertEquals(2, map1.get("2").intValue());
|
||||
Assert.assertEquals(1, map1.get("3").intValue());
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsEqualCollection() {
|
||||
List<String> list1 = Arrays.asList("2", "2", "3");
|
||||
List<String> list2 = Arrays.asList("3", "2", "2");
|
||||
List<String> list3 = Arrays.asList("3", "2", "3");
|
||||
Assert.assertTrue(CollectionUtils.isEqualCollection(list1, list2));
|
||||
Assert.assertFalse(CollectionUtils.isEqualCollection(list1, list3));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsEmpty() {
|
||||
Assert.assertTrue(CollectionUtils.isEmpty(null));
|
||||
Assert.assertTrue(CollectionUtils.isEmpty(new ArrayList<String>()));
|
||||
Assert.assertFalse(CollectionUtils.isEmpty(Arrays.asList("aa")));
|
||||
}
|
||||
}
|
@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.Arrays;
|
||||
|
||||
public class GenericPollerTest {
|
||||
|
||||
@Test
|
||||
public void testNext() {
|
||||
String item1 = "item1";
|
||||
String item2 = "item2";
|
||||
GenericPoller<String> poller = new GenericPoller<>(Arrays.asList(item1, item2));
|
||||
Assert.assertEquals(item1, poller.next());
|
||||
Assert.assertEquals(item2, poller.next());
|
||||
Assert.assertEquals(item1, poller.next());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRefresh() {
|
||||
String item1 = "item1";
|
||||
String item2 = "item2";
|
||||
GenericPoller<String> poller = new GenericPoller<>(Arrays.asList(item1, item2));
|
||||
Poller<String> poller1 = poller.refresh(Arrays.asList(item2));
|
||||
Assert.assertEquals(item2, poller1.next());
|
||||
}
|
||||
}
|
@ -0,0 +1,119 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import com.alibaba.nacos.api.PropertyKeyConst;
|
||||
import com.alibaba.nacos.api.SystemPropertyKeyConst;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.Properties;
|
||||
|
||||
public class InitUtilsTest {
|
||||
|
||||
@Test
|
||||
public void testInitWebRootContext() {
|
||||
String ctx = "/aaa";
|
||||
Properties properties = new Properties();
|
||||
properties.put(PropertyKeyConst.CONTEXT_PATH, ctx);
|
||||
InitUtils.initWebRootContext(properties);
|
||||
Assert.assertEquals(ctx, UtilAndComs.webContext);
|
||||
Assert.assertEquals(ctx + "/v1/ns", UtilAndComs.nacosUrlBase);
|
||||
Assert.assertEquals(ctx + "/v1/ns/instance", UtilAndComs.nacosUrlInstance);
|
||||
}
|
||||
|
||||
/**
|
||||
* current namespace priority 1. system.Properties 2. user.Properties 3. default value
|
||||
*/
|
||||
@Test
|
||||
public void testInitNamespaceForNamingDefault() {
|
||||
//DEFAULT
|
||||
Properties prop = new Properties();
|
||||
String ns = InitUtils.initNamespaceForNaming(prop);
|
||||
Assert.assertEquals("public", ns);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitNamespaceForNamingFromProp() {
|
||||
Properties prop = new Properties();
|
||||
String expect = "ns1";
|
||||
prop.put(PropertyKeyConst.NAMESPACE, expect);
|
||||
String ns = InitUtils.initNamespaceForNaming(prop);
|
||||
Assert.assertEquals(expect, ns);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitNamespaceForNamingFromSystem() {
|
||||
try {
|
||||
String expect1 = "ns1";
|
||||
System.setProperty(PropertyKeyConst.NAMESPACE, expect1);
|
||||
Properties prop = new Properties();
|
||||
prop.put(PropertyKeyConst.NAMESPACE, "cccccc");
|
||||
String ns = InitUtils.initNamespaceForNaming(prop);
|
||||
Assert.assertEquals(expect1, ns);
|
||||
} finally {
|
||||
System.clearProperty(PropertyKeyConst.NAMESPACE);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 1. System.property tenant.id 2. System.property ans.namespace 2. System.env ALIBABA_ALIWARE_NAMESPACE
|
||||
*/
|
||||
@Test
|
||||
public void testInitNamespaceForNamingFromCloud() {
|
||||
try {
|
||||
String expect1 = "ns1";
|
||||
System.setProperty(PropertyKeyConst.IS_USE_CLOUD_NAMESPACE_PARSING, " true");
|
||||
System.setProperty(SystemPropertyKeyConst.ANS_NAMESPACE, expect1);
|
||||
Properties prop = new Properties();
|
||||
prop.put(PropertyKeyConst.NAMESPACE, "cccccc");
|
||||
String ns = InitUtils.initNamespaceForNaming(prop);
|
||||
Assert.assertEquals(expect1, ns);
|
||||
} finally {
|
||||
System.clearProperty(PropertyKeyConst.IS_USE_CLOUD_NAMESPACE_PARSING);
|
||||
System.clearProperty(SystemPropertyKeyConst.ANS_NAMESPACE);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitEndpoint() {
|
||||
Properties prop = new Properties();
|
||||
String endpoint = "1.1.1.1";
|
||||
String endpointPort = "1234";
|
||||
prop.put(PropertyKeyConst.ENDPOINT, endpoint);
|
||||
prop.put(PropertyKeyConst.ENDPOINT_PORT, endpointPort);
|
||||
String actual = InitUtils.initEndpoint(prop);
|
||||
Assert.assertEquals(endpoint + ":" + endpointPort, actual);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitEndpointAns() {
|
||||
try {
|
||||
System.setProperty(PropertyKeyConst.IS_USE_ENDPOINT_PARSING_RULE, "true");
|
||||
Properties prop = new Properties();
|
||||
String endpoint = "${key:test.com}";
|
||||
prop.put(PropertyKeyConst.ENDPOINT, endpoint);
|
||||
String actual = InitUtils.initEndpoint(prop);
|
||||
//defaultEndpointPort is "8080";
|
||||
Assert.assertEquals("test.com:8080", actual);
|
||||
} finally {
|
||||
System.clearProperty(PropertyKeyConst.IS_USE_ENDPOINT_PARSING_RULE);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
@ -0,0 +1,38 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import com.alibaba.nacos.common.constant.HttpHeaderConsts;
|
||||
import com.alibaba.nacos.common.http.param.Header;
|
||||
import com.alibaba.nacos.common.utils.VersionUtils;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
public class NamingHttpUtilTest {
|
||||
|
||||
@Test
|
||||
public void testBuilderHeader() {
|
||||
Header header = NamingHttpUtil.builderHeader();
|
||||
Assert.assertNotNull(header);
|
||||
Assert.assertEquals(header.getValue(HttpHeaderConsts.CLIENT_VERSION_HEADER), VersionUtils.version);
|
||||
Assert.assertEquals(header.getValue(HttpHeaderConsts.USER_AGENT_HEADER), VersionUtils.getFullClientVersion());
|
||||
Assert.assertEquals(header.getValue(HttpHeaderConsts.ACCEPT_ENCODING), "gzip,deflate,sdch");
|
||||
Assert.assertEquals(header.getValue(HttpHeaderConsts.CONNECTION), "Keep-Alive");
|
||||
Assert.assertNotNull(header.getValue(HttpHeaderConsts.REQUEST_ID));
|
||||
Assert.assertEquals(header.getValue(HttpHeaderConsts.REQUEST_MODULE), "Naming");
|
||||
}
|
||||
}
|
@ -0,0 +1,32 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
public class PairTest {
|
||||
|
||||
@Test
|
||||
public void testItem() {
|
||||
String item = "aa";
|
||||
double weight = 1.0;
|
||||
Pair<String> pair = new Pair<>(item, weight);
|
||||
Assert.assertEquals(weight, pair.weight(), 0.01);
|
||||
Assert.assertEquals(item, pair.item());
|
||||
}
|
||||
}
|
@ -0,0 +1,29 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.client.naming.utils;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
public class SignUtilTest {
|
||||
|
||||
@Test
|
||||
public void testSign() throws Exception {
|
||||
String actual = SignUtil.sign("aaa", "b");
|
||||
Assert.assertEquals("DxyaKScrqL26yXYOuHXE3OwfQ0Y=", actual);
|
||||
}
|
||||
}
|
@ -561,7 +561,7 @@ public abstract class RpcClient implements Closeable {
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
LoggerUtils.printIfWarnEnabled(LOGGER, "[{}] Fail to re connect to server ,error is ", name, e);
|
||||
LoggerUtils.printIfWarnEnabled(LOGGER, "[{}] Fail to re connect to server ,error is {}", name, e);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -42,7 +42,7 @@ import io.grpc.stub.StreamObserver;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import java.util.concurrent.SynchronousQueue;
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
import java.util.concurrent.ThreadPoolExecutor;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@ -245,10 +245,14 @@ public abstract class GrpcClient extends RpcClient {
|
||||
public Connection connectToServer(ServerInfo serverInfo) {
|
||||
try {
|
||||
if (grpcExecutor == null) {
|
||||
grpcExecutor = new ThreadPoolExecutor(0, Runtime.getRuntime().availableProcessors() * 8, 10L,
|
||||
TimeUnit.SECONDS, new SynchronousQueue(),
|
||||
|
||||
grpcExecutor = new ThreadPoolExecutor(Runtime.getRuntime().availableProcessors() * 8,
|
||||
Runtime.getRuntime().availableProcessors() * 8, 10L, TimeUnit.SECONDS,
|
||||
new LinkedBlockingQueue<>(10000),
|
||||
new ThreadFactoryBuilder().setDaemon(true).setNameFormat("nacos-grpc-client-executor-%d")
|
||||
.build());
|
||||
grpcExecutor.allowCoreThreadTimeOut(true);
|
||||
|
||||
}
|
||||
RequestGrpc.RequestFutureStub newChannelStubTemp = createNewChannelStub(serverInfo.getServerIp(),
|
||||
serverInfo.getServerPort() + rpcPortOffset());
|
||||
|
@ -157,7 +157,7 @@ public class GrpcConnection extends Connection {
|
||||
requestCallBack.onException(throwable);
|
||||
}
|
||||
}
|
||||
}, this.executor);
|
||||
}, requestCallBack.getExecutor() != null ? requestCallBack.getExecutor() : this.executor);
|
||||
// set timeout future.
|
||||
ListenableFuture<Payload> payloadListenableFuture = Futures
|
||||
.withTimeout(requestFuture, requestCallBack.getTimeout(), TimeUnit.MILLISECONDS,
|
||||
|
@ -48,11 +48,8 @@ import com.alibaba.nacos.config.server.utils.RequestUtil;
|
||||
import com.alibaba.nacos.config.server.utils.TimeUtils;
|
||||
import com.alibaba.nacos.config.server.utils.ZipUtils;
|
||||
import com.alibaba.nacos.sys.utils.InetUtils;
|
||||
import org.apache.catalina.connector.Request;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.apache.commons.lang3.time.DateFormatUtils;
|
||||
import org.apache.tomcat.util.buf.ByteChunk;
|
||||
import org.apache.tomcat.util.http.Parameters;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
@ -73,14 +70,12 @@ import javax.servlet.ServletException;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Field;
|
||||
import java.net.URLDecoder;
|
||||
import java.sql.Timestamp;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.Date;
|
||||
import java.util.HashMap;
|
||||
import java.util.LinkedHashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
@ -300,46 +295,6 @@ public class ConfigController {
|
||||
return rr;
|
||||
}
|
||||
|
||||
private void removeRequestContext(HttpServletRequest request) {
|
||||
try {
|
||||
|
||||
request.removeAttribute("body");
|
||||
|
||||
Map<String, String[]> parameterMap = request.getParameterMap();
|
||||
Field locked = parameterMap.getClass().getDeclaredField("locked");
|
||||
locked.setAccessible(true);
|
||||
locked.set(parameterMap, false);
|
||||
parameterMap.remove(Constants.PROBE_MODIFY_REQUEST);
|
||||
|
||||
Field inneRequestFiled = request.getClass().getDeclaredField("request");
|
||||
inneRequestFiled.setAccessible(true);
|
||||
Request innerRequest = (Request) inneRequestFiled.get(request);
|
||||
|
||||
Field coyoteRequest = innerRequest.getClass().getDeclaredField("coyoteRequest");
|
||||
coyoteRequest.setAccessible(true);
|
||||
org.apache.coyote.Request coyotoRequest = (org.apache.coyote.Request) coyoteRequest.get(innerRequest);
|
||||
Parameters parameters = coyotoRequest.getParameters();
|
||||
Field hashMapField = parameters.getClass().getDeclaredField("paramHashValues");
|
||||
hashMapField.setAccessible(true);
|
||||
LinkedHashMap hashMaps = (LinkedHashMap) hashMapField.get(parameters);
|
||||
hashMaps.remove(Constants.PROBE_MODIFY_REQUEST);
|
||||
|
||||
Field tmpNameField = parameters.getClass().getDeclaredField("tmpName");
|
||||
tmpNameField.setAccessible(true);
|
||||
ByteChunk tmpName = (ByteChunk) tmpNameField.get(parameters);
|
||||
byte[] bytemp = new byte[0];
|
||||
tmpName.setBytes(bytemp, 0, 0);
|
||||
|
||||
Field tmpValueField = parameters.getClass().getDeclaredField("tmpValue");
|
||||
tmpValueField.setAccessible(true);
|
||||
ByteChunk tmpValue = (ByteChunk) tmpValueField.get(parameters);
|
||||
tmpValue.setBytes(bytemp, 0, 0);
|
||||
|
||||
} catch (Exception e) {
|
||||
LOGGER.warn("remove listen request param error", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* The client listens for configuration changes.
|
||||
*/
|
||||
@ -349,8 +304,6 @@ public class ConfigController {
|
||||
throws ServletException, IOException {
|
||||
|
||||
request.setAttribute("org.apache.catalina.ASYNC_SUPPORTED", true);
|
||||
//remove large listen context , reduce request content to optimize cms gc.
|
||||
removeRequestContext(request);
|
||||
String probeModify = request.getParameter("Listening-Configs");
|
||||
if (StringUtils.isBlank(probeModify)) {
|
||||
LOGGER.warn("invalid probeModify is blank");
|
||||
|
@ -360,8 +360,6 @@ public class ServerMemberManager implements ApplicationListener<WebServerInitial
|
||||
tmpMap.put(address, existMember);
|
||||
}
|
||||
|
||||
// Ensure that the node is created only once
|
||||
tmpMap.put(address, member);
|
||||
if (NodeState.UP.equals(member.getState())) {
|
||||
tmpAddressInfo.add(address);
|
||||
}
|
||||
|
@ -169,7 +169,11 @@ public class GrpcBiStreamRequestAcceptor extends BiRequestStreamGrpc.BiRequestSt
|
||||
//client close the stream.
|
||||
return;
|
||||
} else {
|
||||
serverCallStreamObserver.onCompleted();
|
||||
try {
|
||||
serverCallStreamObserver.onCompleted();
|
||||
} catch (Throwable throwable) {
|
||||
//ignore
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -186,7 +190,12 @@ public class GrpcBiStreamRequestAcceptor extends BiRequestStreamGrpc.BiRequestSt
|
||||
//client close the stream.
|
||||
return;
|
||||
} else {
|
||||
serverCallStreamObserver.onCompleted();
|
||||
try {
|
||||
serverCallStreamObserver.onCompleted();
|
||||
} catch (Throwable throwable) {
|
||||
//ignore
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -39,6 +39,9 @@ public class GrpcClusterServer extends BaseGrpcServer {
|
||||
|
||||
@Override
|
||||
public ThreadPoolExecutor getRpcExecutor() {
|
||||
if (!GlobalExecutor.clusterRpcExecutor.allowsCoreThreadTimeOut()) {
|
||||
GlobalExecutor.clusterRpcExecutor.allowCoreThreadTimeOut(true);
|
||||
}
|
||||
return GlobalExecutor.clusterRpcExecutor;
|
||||
}
|
||||
}
|
||||
|
@ -20,8 +20,8 @@ import com.alibaba.nacos.common.executor.ExecutorFactory;
|
||||
import com.alibaba.nacos.common.executor.NameThreadFactory;
|
||||
import com.google.common.util.concurrent.ThreadFactoryBuilder;
|
||||
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
import java.util.concurrent.SynchronousQueue;
|
||||
import java.util.concurrent.ThreadPoolExecutor;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
@ -43,15 +43,15 @@ public class GlobalExecutor {
|
||||
new NameThreadFactory("com.alibaba.nacos.core.protocal.distro"));
|
||||
|
||||
public static final ThreadPoolExecutor sdkRpcExecutor = new ThreadPoolExecutor(
|
||||
Runtime.getRuntime().availableProcessors(),
|
||||
Runtime.getRuntime().availableProcessors() * RemoteUtils.getRemoteExecutorTimesOfProcessors(),
|
||||
Runtime.getRuntime().availableProcessors() * RemoteUtils.getRemoteExecutorTimesOfProcessors(), 60L,
|
||||
TimeUnit.SECONDS, new SynchronousQueue(),
|
||||
TimeUnit.SECONDS, new LinkedBlockingQueue<>(RemoteUtils.getRemoteExecutorQueueSize()),
|
||||
new ThreadFactoryBuilder().setDaemon(true).setNameFormat("nacos-grpc-executor-%d").build());
|
||||
|
||||
public static final ThreadPoolExecutor clusterRpcExecutor = new ThreadPoolExecutor(
|
||||
Runtime.getRuntime().availableProcessors(),
|
||||
Runtime.getRuntime().availableProcessors() * RemoteUtils.getRemoteExecutorTimesOfProcessors(),
|
||||
Runtime.getRuntime().availableProcessors() * RemoteUtils.getRemoteExecutorTimesOfProcessors(), 60L,
|
||||
TimeUnit.SECONDS, new SynchronousQueue(),
|
||||
TimeUnit.SECONDS, new LinkedBlockingQueue<>(RemoteUtils.getRemoteExecutorQueueSize()),
|
||||
new ThreadFactoryBuilder().setDaemon(true).setNameFormat("nacos-cluster-grpc-executor-%d").build());
|
||||
|
||||
public static void runWithoutThread(Runnable runnable) {
|
||||
|
@ -28,7 +28,15 @@ public class RemoteUtils {
|
||||
|
||||
public static final float LOADER_FACTOR = 0.1f;
|
||||
|
||||
private static final int REMOTE_EXECUTOR_TIMES_OF_PROCESSORS = 64;
|
||||
/**
|
||||
* Default remote execute times for CPU count of task processors.
|
||||
*/
|
||||
private static final int REMOTE_EXECUTOR_TIMES_OF_PROCESSORS = 1 << 4;
|
||||
|
||||
/**
|
||||
* Default remote execute queue size: 16384.
|
||||
*/
|
||||
private static final int REMOTE_EXECUTOR_QUEUE_SIZE = 1 << 14;
|
||||
|
||||
/**
|
||||
* get remote executors thread times of processors,default is 64. see the usage of this method for detail.
|
||||
@ -44,4 +52,14 @@ public class RemoteUtils {
|
||||
return REMOTE_EXECUTOR_TIMES_OF_PROCESSORS;
|
||||
}
|
||||
}
|
||||
|
||||
public static int getRemoteExecutorQueueSize() {
|
||||
String queueSizeString = System.getProperty("remote.executor.queue.size");
|
||||
if (NumberUtil.isDigits(queueSizeString)) {
|
||||
Integer size = Integer.valueOf(queueSizeString);
|
||||
return size > 0 ? size : REMOTE_EXECUTOR_QUEUE_SIZE;
|
||||
} else {
|
||||
return REMOTE_EXECUTOR_QUEUE_SIZE;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -36,6 +36,7 @@ import com.alibaba.nacos.naming.cluster.remote.response.DistroDataResponse;
|
||||
import com.alibaba.nacos.naming.core.v2.event.client.ClientEvent;
|
||||
import com.alibaba.nacos.naming.misc.GlobalExecutor;
|
||||
import com.alibaba.nacos.naming.misc.Loggers;
|
||||
import com.alibaba.nacos.naming.monitor.NamingTpsMonitor;
|
||||
|
||||
import java.util.concurrent.Executor;
|
||||
|
||||
@ -89,7 +90,7 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
DistroDataRequest request = new DistroDataRequest(data, data.getType());
|
||||
Member member = memberManager.find(targetServer);
|
||||
try {
|
||||
clusterRpcClientProxy.asyncRequest(member, request, new DistroRpcCallbackWrapper(callback));
|
||||
clusterRpcClientProxy.asyncRequest(member, request, new DistroRpcCallbackWrapper(callback, member));
|
||||
} catch (NacosException nacosException) {
|
||||
callback.onFailed(nacosException);
|
||||
}
|
||||
@ -126,7 +127,7 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
Member member = memberManager.find(targetServer);
|
||||
try {
|
||||
DistroVerifyCallbackWrapper wrapper = new DistroVerifyCallbackWrapper(targetServer,
|
||||
verifyData.getDistroKey().getResourceKey(), callback);
|
||||
verifyData.getDistroKey().getResourceKey(), callback, member);
|
||||
clusterRpcClientProxy.asyncRequest(member, request, wrapper);
|
||||
} catch (NacosException nacosException) {
|
||||
callback.onFailed(nacosException);
|
||||
@ -199,8 +200,11 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
|
||||
private final DistroCallback distroCallback;
|
||||
|
||||
public DistroRpcCallbackWrapper(DistroCallback distroCallback) {
|
||||
private final Member member;
|
||||
|
||||
public DistroRpcCallbackWrapper(DistroCallback distroCallback, Member member) {
|
||||
this.distroCallback = distroCallback;
|
||||
this.member = member;
|
||||
}
|
||||
|
||||
@Override
|
||||
@ -217,8 +221,10 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
@Override
|
||||
public void onResponse(Response response) {
|
||||
if (checkResponse(response)) {
|
||||
NamingTpsMonitor.distroSyncSuccess(member.getAddress(), member.getIp());
|
||||
distroCallback.onSuccess();
|
||||
} else {
|
||||
NamingTpsMonitor.distroSyncFail(member.getAddress(), member.getIp());
|
||||
distroCallback.onFailed(null);
|
||||
}
|
||||
}
|
||||
@ -236,11 +242,15 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
private final String clientId;
|
||||
|
||||
private final DistroCallback distroCallback;
|
||||
|
||||
private final Member member;
|
||||
|
||||
private DistroVerifyCallbackWrapper(String targetServer, String clientId, DistroCallback distroCallback) {
|
||||
private DistroVerifyCallbackWrapper(String targetServer, String clientId, DistroCallback distroCallback,
|
||||
Member member) {
|
||||
this.targetServer = targetServer;
|
||||
this.clientId = clientId;
|
||||
this.distroCallback = distroCallback;
|
||||
this.member = member;
|
||||
}
|
||||
|
||||
@Override
|
||||
@ -257,10 +267,12 @@ public class DistroClientTransportAgent implements DistroTransportAgent {
|
||||
@Override
|
||||
public void onResponse(Response response) {
|
||||
if (checkResponse(response)) {
|
||||
NamingTpsMonitor.distroVerifySuccess(member.getAddress(), member.getIp());
|
||||
distroCallback.onSuccess();
|
||||
} else {
|
||||
Loggers.DISTRO.info("Target {} verify client {} failed, sync new client", targetServer, clientId);
|
||||
NotifyCenter.publishEvent(new ClientEvent.ClientVerifyFailedEvent(clientId, targetServer));
|
||||
NamingTpsMonitor.distroVerifyFail(member.getAddress(), member.getIp());
|
||||
distroCallback.onFailed(null);
|
||||
}
|
||||
}
|
||||
|
@ -0,0 +1,40 @@
|
||||
/*
|
||||
* Copyright (c) 1999-2021 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.heartbeat;
|
||||
|
||||
import com.alibaba.nacos.naming.core.DistroMapper;
|
||||
import com.alibaba.nacos.sys.utils.ApplicationUtils;
|
||||
|
||||
/**
|
||||
* Instance responsibility check interceptor.
|
||||
*
|
||||
* @author gengtuo.ygt
|
||||
* on 2021/3/24
|
||||
*/
|
||||
public class InstanceBeatCheckResponsibleInterceptor extends AbstractBeatCheckInterceptor {
|
||||
|
||||
@Override
|
||||
public boolean intercept(InstanceBeatCheckTask object) {
|
||||
return !ApplicationUtils.getBean(DistroMapper.class).responsible(object.getClient().getResponsibleId());
|
||||
}
|
||||
|
||||
@Override
|
||||
public int order() {
|
||||
return Integer.MIN_VALUE + 2;
|
||||
}
|
||||
|
||||
}
|
@ -17,7 +17,7 @@
|
||||
package com.alibaba.nacos.naming.healthcheck.heartbeat;
|
||||
|
||||
import com.alibaba.nacos.common.spi.NacosServiceLoader;
|
||||
import com.alibaba.nacos.naming.core.v2.client.Client;
|
||||
import com.alibaba.nacos.naming.core.v2.client.impl.IpPortBasedClient;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.HealthCheckInstancePublishInfo;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.interceptor.Interceptable;
|
||||
@ -34,7 +34,7 @@ public class InstanceBeatCheckTask implements Interceptable {
|
||||
|
||||
private static final List<InstanceBeatChecker> CHECKERS = new LinkedList<>();
|
||||
|
||||
private final Client client;
|
||||
private final IpPortBasedClient client;
|
||||
|
||||
private final Service service;
|
||||
|
||||
@ -46,7 +46,7 @@ public class InstanceBeatCheckTask implements Interceptable {
|
||||
CHECKERS.addAll(NacosServiceLoader.load(InstanceBeatChecker.class));
|
||||
}
|
||||
|
||||
public InstanceBeatCheckTask(Client client, Service service, HealthCheckInstancePublishInfo instancePublishInfo) {
|
||||
public InstanceBeatCheckTask(IpPortBasedClient client, Service service, HealthCheckInstancePublishInfo instancePublishInfo) {
|
||||
this.client = client;
|
||||
this.service = service;
|
||||
this.instancePublishInfo = instancePublishInfo;
|
||||
@ -63,7 +63,7 @@ public class InstanceBeatCheckTask implements Interceptable {
|
||||
public void afterIntercept() {
|
||||
}
|
||||
|
||||
public Client getClient() {
|
||||
public IpPortBasedClient getClient() {
|
||||
return client;
|
||||
}
|
||||
|
||||
|
@ -34,6 +34,7 @@ public class NamingTpsMonitor {
|
||||
private NamingTpsMonitor() {
|
||||
this.tpsMonitorManager = ApplicationUtils.getBean(TpsMonitorManager.class);
|
||||
registerPushMonitorPoint();
|
||||
registerDistroMonitorPoint();
|
||||
}
|
||||
|
||||
private void registerPushMonitorPoint() {
|
||||
@ -45,6 +46,15 @@ public class NamingTpsMonitor {
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_UDP_PUSH_FAIL.name()));
|
||||
}
|
||||
|
||||
private void registerDistroMonitorPoint() {
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_SYNC.name()));
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_SYNC_SUCCESS.name()));
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_SYNC_FAIL.name()));
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_VERIFY.name()));
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_VERIFY_SUCCESS.name()));
|
||||
tpsMonitorManager.registerTpsControlPoint(new TpsMonitorPoint(TpsMonitorItem.NAMING_DISTRO_VERIFY_FAIL.name()));
|
||||
}
|
||||
|
||||
public static NamingTpsMonitor getInstance() {
|
||||
return INSTANCE;
|
||||
}
|
||||
@ -92,4 +102,48 @@ public class NamingTpsMonitor {
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_UDP_PUSH.name(), clientId, clientIp);
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_UDP_PUSH_FAIL.name(), clientId, clientIp);
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply distro sync success.
|
||||
*
|
||||
* @param clientId client id
|
||||
* @param clientIp client ip
|
||||
*/
|
||||
public static void distroSyncSuccess(String clientId, String clientIp) {
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_SYNC.name(), clientId, clientIp);
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_SYNC_SUCCESS.name(), clientId, clientIp);
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply distro sync fail.
|
||||
*
|
||||
* @param clientId client id
|
||||
* @param clientIp client ip
|
||||
*/
|
||||
public static void distroSyncFail(String clientId, String clientIp) {
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_SYNC.name(), clientId, clientIp);
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_SYNC_FAIL.name(), clientId, clientIp);
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply distro verify success.
|
||||
*
|
||||
* @param clientId client id
|
||||
* @param clientIp client ip
|
||||
*/
|
||||
public static void distroVerifySuccess(String clientId, String clientIp) {
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_VERIFY.name(), clientId, clientIp);
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_VERIFY_SUCCESS.name(), clientId, clientIp);
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply distro verify fail.
|
||||
*
|
||||
* @param clientId client id
|
||||
* @param clientIp client ip
|
||||
*/
|
||||
public static void distroVerifyFail(String clientId, String clientIp) {
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_VERIFY.name(), clientId, clientIp);
|
||||
INSTANCE.tpsMonitorManager.applyTpsForClientIp(TpsMonitorItem.NAMING_DISTRO_VERIFY_FAIL.name(), clientId, clientIp);
|
||||
}
|
||||
}
|
||||
|
@ -51,5 +51,35 @@ public enum TpsMonitorItem {
|
||||
/**
|
||||
* Naming udp fail push.
|
||||
*/
|
||||
NAMING_UDP_PUSH_FAIL;
|
||||
NAMING_UDP_PUSH_FAIL,
|
||||
|
||||
/**
|
||||
* Naming rpc distro sync total count.
|
||||
*/
|
||||
NAMING_DISTRO_SYNC,
|
||||
|
||||
/**
|
||||
* Naming rpc distro sync success count.
|
||||
*/
|
||||
NAMING_DISTRO_SYNC_SUCCESS,
|
||||
|
||||
/**
|
||||
* Naming rpc distro sync fail count.
|
||||
*/
|
||||
NAMING_DISTRO_SYNC_FAIL,
|
||||
|
||||
/**
|
||||
* Naming rpc distro verify fail count.
|
||||
*/
|
||||
NAMING_DISTRO_VERIFY,
|
||||
|
||||
/**
|
||||
* Naming rpc distro verify fail count.
|
||||
*/
|
||||
NAMING_DISTRO_VERIFY_SUCCESS,
|
||||
|
||||
/**
|
||||
* Naming rpc distro verify fail count.
|
||||
*/
|
||||
NAMING_DISTRO_VERIFY_FAIL,
|
||||
}
|
||||
|
@ -106,7 +106,7 @@ public class PushExecuteTask extends AbstractExecuteTask {
|
||||
@Override
|
||||
public long getTimeout() {
|
||||
// TODO timeout should can be config
|
||||
return 3000L;
|
||||
return 5000L;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
@ -30,6 +30,7 @@ import org.apache.commons.lang3.StringUtils;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.LinkedList;
|
||||
@ -158,7 +159,7 @@ public class ServiceUtil {
|
||||
start = 0;
|
||||
}
|
||||
if (start >= result.size()) {
|
||||
return result;
|
||||
return Collections.emptyList();
|
||||
}
|
||||
int end = start + pageSize;
|
||||
if (end > result.size()) {
|
||||
|
@ -16,9 +16,12 @@
|
||||
|
||||
package com.alibaba.nacos.naming.web;
|
||||
|
||||
import com.alibaba.nacos.api.exception.runtime.NacosDeserializationException;
|
||||
import com.alibaba.nacos.common.utils.IPUtil;
|
||||
import com.alibaba.nacos.common.utils.JacksonUtils;
|
||||
import com.alibaba.nacos.common.utils.StringUtils;
|
||||
import com.alibaba.nacos.core.utils.ReuseHttpServletRequest;
|
||||
import com.alibaba.nacos.naming.healthcheck.RsInfo;
|
||||
|
||||
/**
|
||||
* Distro IP and port tag generator.
|
||||
@ -27,6 +30,8 @@ import com.alibaba.nacos.core.utils.ReuseHttpServletRequest;
|
||||
*/
|
||||
public class DistroIpPortTagGenerator implements DistroTagGenerator {
|
||||
|
||||
private static final String PARAMETER_BEAT = "beat";
|
||||
|
||||
private static final String PARAMETER_IP = "ip";
|
||||
|
||||
private static final String PARAMETER_PORT = "port";
|
||||
@ -35,6 +40,18 @@ public class DistroIpPortTagGenerator implements DistroTagGenerator {
|
||||
public String getResponsibleTag(ReuseHttpServletRequest request) {
|
||||
String ip = request.getParameter(PARAMETER_IP);
|
||||
String port = request.getParameter(PARAMETER_PORT);
|
||||
if (StringUtils.isBlank(ip)) {
|
||||
// some old version clients using beat parameter
|
||||
String beatStr = request.getParameter(PARAMETER_BEAT);
|
||||
if (StringUtils.isNotBlank(beatStr)) {
|
||||
try {
|
||||
RsInfo rsInfo = JacksonUtils.toObj(beatStr, RsInfo.class);
|
||||
ip = rsInfo.getIp();
|
||||
port = String.valueOf(rsInfo.getPort());
|
||||
} catch (NacosDeserializationException ignored) {
|
||||
}
|
||||
}
|
||||
}
|
||||
if (StringUtils.isNotBlank(ip)) {
|
||||
ip = ip.trim();
|
||||
}
|
||||
|
@ -16,3 +16,4 @@
|
||||
|
||||
com.alibaba.nacos.naming.healthcheck.heartbeat.ServiceEnableBeatCheckInterceptor
|
||||
com.alibaba.nacos.naming.healthcheck.heartbeat.InstanceEnableBeatCheckInterceptor
|
||||
com.alibaba.nacos.naming.healthcheck.heartbeat.InstanceBeatCheckResponsibleInterceptor
|
||||
|
@ -0,0 +1,110 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.v2;
|
||||
|
||||
import com.alibaba.nacos.naming.core.v2.client.impl.IpPortBasedClient;
|
||||
import com.alibaba.nacos.naming.core.v2.metadata.NamingMetadataManager;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.misc.SwitchDomain;
|
||||
import com.alibaba.nacos.sys.utils.ApplicationUtils;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class HealthCheckTaskV2Test {
|
||||
|
||||
private HealthCheckTaskV2 healthCheckTaskV2;
|
||||
|
||||
@Mock
|
||||
private IpPortBasedClient ipPortBasedClient;
|
||||
|
||||
@Mock
|
||||
private ConfigurableApplicationContext context;
|
||||
|
||||
@Mock
|
||||
private SwitchDomain switchDomain;
|
||||
|
||||
@Mock
|
||||
private Service service;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
ApplicationUtils.injectContext(context);
|
||||
when(ApplicationUtils.getBean(SwitchDomain.class)).thenReturn(switchDomain);
|
||||
when(switchDomain.getTcpHealthParams()).thenReturn(new SwitchDomain.TcpHealthParams());
|
||||
when(ApplicationUtils.getBean(NamingMetadataManager.class)).thenReturn(new NamingMetadataManager());
|
||||
healthCheckTaskV2 = new HealthCheckTaskV2(ipPortBasedClient);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDoHealthCheck() {
|
||||
when(ipPortBasedClient.getAllPublishedService()).thenReturn(returnService());
|
||||
|
||||
healthCheckTaskV2.setCheckRtWorst(1);
|
||||
healthCheckTaskV2.setCheckRtLastLast(1);
|
||||
Assert.assertEquals(1, healthCheckTaskV2.getCheckRtWorst());
|
||||
Assert.assertEquals(1, healthCheckTaskV2.getCheckRtLastLast());
|
||||
|
||||
healthCheckTaskV2.run();
|
||||
healthCheckTaskV2.passIntercept();
|
||||
healthCheckTaskV2.doHealthCheck();
|
||||
|
||||
verify(ipPortBasedClient, times(3)).getAllPublishedService();
|
||||
verify(switchDomain, times(3)).isHealthCheckEnabled(service.getGroupedServiceName());
|
||||
}
|
||||
|
||||
private List<Service> returnService() {
|
||||
return Collections.singletonList(service);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetClient() {
|
||||
Assert.assertNotNull(healthCheckTaskV2.getClient());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetAndSet() {
|
||||
healthCheckTaskV2.setCheckRtBest(1);
|
||||
healthCheckTaskV2.setCheckRtNormalized(1);
|
||||
healthCheckTaskV2.setCheckRtLast(1);
|
||||
healthCheckTaskV2.setCancelled(true);
|
||||
healthCheckTaskV2.setStartTime(1615796485783L);
|
||||
|
||||
Assert.assertEquals(1, healthCheckTaskV2.getCheckRtBest());
|
||||
Assert.assertEquals(1, healthCheckTaskV2.getCheckRtNormalized());
|
||||
Assert.assertEquals(1, healthCheckTaskV2.getCheckRtLast());
|
||||
Assert.assertTrue(healthCheckTaskV2.isCancelled());
|
||||
Assert.assertEquals(1615796485783L, healthCheckTaskV2.getStartTime());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAfterIntercept() {
|
||||
healthCheckTaskV2.afterIntercept();
|
||||
}
|
||||
}
|
@ -0,0 +1,55 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.v2;
|
||||
|
||||
import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.naming.core.v2.client.Client;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.InstancePublishInfo;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.core.v2.service.impl.PersistentClientOperationServiceImpl;
|
||||
import com.alibaba.nacos.naming.utils.InstanceUtil;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class PersistentHealthStatusSynchronizerTest {
|
||||
|
||||
@Mock
|
||||
private PersistentClientOperationServiceImpl persistentClientOperationService;
|
||||
|
||||
@Mock
|
||||
private Client client;
|
||||
|
||||
@Test
|
||||
public void testInstanceHealthStatusChange() {
|
||||
Service service = Service.newService("public", "DEFAULT", "nacos", true);
|
||||
InstancePublishInfo instancePublishInfo = new InstancePublishInfo("127.0.0.1", 8080);
|
||||
PersistentHealthStatusSynchronizer persistentHealthStatusSynchronizer = new PersistentHealthStatusSynchronizer(
|
||||
persistentClientOperationService);
|
||||
persistentHealthStatusSynchronizer.instanceHealthStatusChange(true, client, service, instancePublishInfo);
|
||||
|
||||
Instance updateInstance = InstanceUtil.parseToApiInstance(service, instancePublishInfo);
|
||||
updateInstance.setHealthy(true);
|
||||
|
||||
verify(client).getClientId();
|
||||
verify(persistentClientOperationService).registerInstance(service, updateInstance, client.getClientId());
|
||||
}
|
||||
}
|
@ -0,0 +1,118 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.v2.processor;
|
||||
|
||||
import com.alibaba.nacos.naming.core.v2.client.impl.IpPortBasedClient;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.HealthCheckInstancePublishInfo;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.healthcheck.v2.HealthCheckTaskV2;
|
||||
import com.alibaba.nacos.naming.misc.SwitchDomain;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class HealthCheckCommonV2Test {
|
||||
|
||||
@Mock
|
||||
private SwitchDomain.HealthParams healthParams;
|
||||
|
||||
@Mock
|
||||
private HealthCheckTaskV2 healthCheckTaskV2;
|
||||
|
||||
@Mock
|
||||
private Service service;
|
||||
|
||||
@Mock
|
||||
private IpPortBasedClient ipPortBasedClient;
|
||||
|
||||
@Mock
|
||||
private HealthCheckInstancePublishInfo healthCheckInstancePublishInfo;
|
||||
|
||||
private HealthCheckCommonV2 healthCheckCommonV2;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
healthCheckCommonV2 = new HealthCheckCommonV2();
|
||||
when(healthCheckTaskV2.getClient()).thenReturn(ipPortBasedClient);
|
||||
when(ipPortBasedClient.getInstancePublishInfo(service)).thenReturn(healthCheckInstancePublishInfo);
|
||||
when(healthCheckInstancePublishInfo.getFailCount()).thenReturn(new AtomicInteger());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReEvaluateCheckRT() {
|
||||
healthCheckCommonV2.reEvaluateCheckRT(1, healthCheckTaskV2, healthParams);
|
||||
|
||||
verify(healthParams, times(2)).getMax();
|
||||
verify(healthParams, times(1)).getMin();
|
||||
verify(healthParams, times(2)).getFactor();
|
||||
|
||||
verify(healthCheckTaskV2).getCheckRtWorst();
|
||||
verify(healthCheckTaskV2).getCheckRtBest();
|
||||
verify(healthCheckTaskV2).getCheckRtNormalized();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckOk() {
|
||||
healthCheckCommonV2.checkOk(healthCheckTaskV2, service, "test checkOk");
|
||||
|
||||
verify(healthCheckTaskV2).getClient();
|
||||
verify(service).getGroupedServiceName();
|
||||
verify(ipPortBasedClient).getInstancePublishInfo(service);
|
||||
verify(healthCheckInstancePublishInfo).isHealthy();
|
||||
verify(healthCheckInstancePublishInfo).getCluster();
|
||||
verify(healthCheckInstancePublishInfo).resetFailCount();
|
||||
verify(healthCheckInstancePublishInfo).finishCheck();
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckFail() {
|
||||
when(healthCheckInstancePublishInfo.isHealthy()).thenReturn(true);
|
||||
healthCheckCommonV2.checkFail(healthCheckTaskV2, service, "test checkFail");
|
||||
|
||||
verify(healthCheckTaskV2).getClient();
|
||||
verify(service).getGroupedServiceName();
|
||||
verify(ipPortBasedClient).getInstancePublishInfo(service);
|
||||
verify(healthCheckInstancePublishInfo).isHealthy();
|
||||
verify(healthCheckInstancePublishInfo).getCluster();
|
||||
verify(healthCheckInstancePublishInfo).resetOkCount();
|
||||
verify(healthCheckInstancePublishInfo).finishCheck();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCheckFailNow() {
|
||||
when(healthCheckInstancePublishInfo.isHealthy()).thenReturn(true);
|
||||
healthCheckCommonV2.checkFailNow(healthCheckTaskV2, service, "test checkFailNow");
|
||||
|
||||
verify(healthCheckTaskV2).getClient();
|
||||
verify(service).getGroupedServiceName();
|
||||
verify(ipPortBasedClient).getInstancePublishInfo(service);
|
||||
verify(healthCheckInstancePublishInfo).isHealthy();
|
||||
verify(healthCheckInstancePublishInfo).getCluster();
|
||||
verify(healthCheckInstancePublishInfo).resetOkCount();
|
||||
verify(healthCheckInstancePublishInfo).finishCheck();
|
||||
}
|
||||
}
|
@ -0,0 +1,94 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.v2.processor;
|
||||
|
||||
import com.alibaba.nacos.api.naming.pojo.healthcheck.HealthCheckType;
|
||||
import com.alibaba.nacos.naming.core.v2.client.impl.IpPortBasedClient;
|
||||
import com.alibaba.nacos.naming.core.v2.metadata.ClusterMetadata;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.healthcheck.extend.HealthCheckExtendProvider;
|
||||
import com.alibaba.nacos.naming.healthcheck.v2.HealthCheckTaskV2;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class HealthCheckProcessorV2DelegateTest {
|
||||
|
||||
@Mock
|
||||
private HealthCheckExtendProvider healthCheckExtendProvider;
|
||||
|
||||
@Mock
|
||||
private HealthCheckTaskV2 healthCheckTaskV2;
|
||||
|
||||
@Mock
|
||||
private Service service;
|
||||
|
||||
@Mock
|
||||
private ClusterMetadata clusterMetadata;
|
||||
|
||||
private HealthCheckProcessorV2Delegate healthCheckProcessorV2Delegate;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
healthCheckProcessorV2Delegate = new HealthCheckProcessorV2Delegate(healthCheckExtendProvider);
|
||||
verify(healthCheckExtendProvider).init();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAddProcessor() throws NoSuchFieldException, IllegalAccessException {
|
||||
List<HealthCheckProcessorV2> list = new ArrayList<>();
|
||||
list.add(new TcpHealthCheckProcessor(null, null));
|
||||
healthCheckProcessorV2Delegate.addProcessor(list);
|
||||
|
||||
Class<HealthCheckProcessorV2Delegate> healthCheckProcessorV2DelegateClass = HealthCheckProcessorV2Delegate.class;
|
||||
Field field = healthCheckProcessorV2DelegateClass.getDeclaredField("healthCheckProcessorMap");
|
||||
field.setAccessible(true);
|
||||
Map<String, HealthCheckProcessorV2> map = (Map<String, HealthCheckProcessorV2>) field
|
||||
.get(healthCheckProcessorV2Delegate);
|
||||
HealthCheckProcessorV2 healthCheckProcessorV2 = map.get(HealthCheckType.TCP.name());
|
||||
Assert.assertNotNull(healthCheckProcessorV2);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcess() throws NoSuchFieldException, IllegalAccessException {
|
||||
testAddProcessor();
|
||||
when(clusterMetadata.getHealthyCheckType()).thenReturn(HealthCheckType.TCP.name());
|
||||
when(healthCheckTaskV2.getClient()).thenReturn(new IpPortBasedClient("127.0.0.1:80#true", true));
|
||||
|
||||
healthCheckProcessorV2Delegate.process(healthCheckTaskV2, service, clusterMetadata);
|
||||
|
||||
verify(clusterMetadata).getHealthyCheckType();
|
||||
verify(healthCheckTaskV2).getClient();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetType() {
|
||||
Assert.assertNull(healthCheckProcessorV2Delegate.getType());
|
||||
}
|
||||
}
|
@ -0,0 +1,179 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.naming.healthcheck.v2.processor;
|
||||
|
||||
import com.alibaba.nacos.api.naming.pojo.healthcheck.HealthCheckType;
|
||||
import com.alibaba.nacos.common.model.RestResult;
|
||||
import com.alibaba.nacos.naming.core.v2.client.impl.IpPortBasedClient;
|
||||
import com.alibaba.nacos.naming.core.v2.metadata.ClusterMetadata;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.HealthCheckInstancePublishInfo;
|
||||
import com.alibaba.nacos.naming.core.v2.pojo.Service;
|
||||
import com.alibaba.nacos.naming.healthcheck.v2.HealthCheckTaskV2;
|
||||
import com.alibaba.nacos.naming.misc.SwitchDomain;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
|
||||
import java.lang.reflect.Constructor;
|
||||
import java.lang.reflect.InvocationTargetException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.net.ConnectException;
|
||||
import java.net.HttpURLConnection;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class HttpHealthCheckProcessorTest {
|
||||
|
||||
@Mock
|
||||
private HealthCheckCommonV2 healthCheckCommon;
|
||||
|
||||
@Mock
|
||||
private SwitchDomain switchDomain;
|
||||
|
||||
@Mock
|
||||
private HealthCheckTaskV2 healthCheckTaskV2;
|
||||
|
||||
@Mock
|
||||
private Service service;
|
||||
|
||||
@Mock
|
||||
private ClusterMetadata clusterMetadata;
|
||||
|
||||
@Mock
|
||||
private IpPortBasedClient ipPortBasedClient;
|
||||
|
||||
@Mock
|
||||
private HealthCheckInstancePublishInfo healthCheckInstancePublishInfo;
|
||||
|
||||
@Mock
|
||||
private RestResult restResult;
|
||||
|
||||
@Mock
|
||||
private ConnectException connectException;
|
||||
|
||||
private HttpHealthCheckProcessor httpHealthCheckProcessor;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
when(switchDomain.getHttpHealthParams()).thenReturn(new SwitchDomain.HttpHealthParams());
|
||||
when(healthCheckTaskV2.getClient()).thenReturn(ipPortBasedClient);
|
||||
when(ipPortBasedClient.getInstancePublishInfo(service)).thenReturn(healthCheckInstancePublishInfo);
|
||||
httpHealthCheckProcessor = new HttpHealthCheckProcessor(healthCheckCommon, switchDomain);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcess() {
|
||||
httpHealthCheckProcessor.process(healthCheckTaskV2, service, clusterMetadata);
|
||||
|
||||
verify(healthCheckTaskV2).getClient();
|
||||
verify(healthCheckInstancePublishInfo).tryStartCheck();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetType() {
|
||||
Assert.assertEquals(httpHealthCheckProcessor.getType(), HealthCheckType.HTTP.name());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testConstructor()
|
||||
throws NoSuchMethodException, IllegalAccessException, InvocationTargetException, InstantiationException {
|
||||
Class<HttpHealthCheckProcessor> healthCheckProcessorClass = HttpHealthCheckProcessor.class;
|
||||
Class<?>[] classes = healthCheckProcessorClass.getDeclaredClasses();
|
||||
Class<?> aClass = Arrays.stream(classes).findFirst().get();
|
||||
Constructor<?> constructor = aClass
|
||||
.getConstructor(HttpHealthCheckProcessor.class, HealthCheckInstancePublishInfo.class,
|
||||
HealthCheckTaskV2.class, Service.class);
|
||||
Object objects = constructor
|
||||
.newInstance(httpHealthCheckProcessor, healthCheckInstancePublishInfo, healthCheckTaskV2, service);
|
||||
|
||||
Assert.assertNotNull(objects);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOnReceive()
|
||||
throws NoSuchMethodException, IllegalAccessException, InvocationTargetException, InstantiationException {
|
||||
Class<HttpHealthCheckProcessor> healthCheckProcessorClass = HttpHealthCheckProcessor.class;
|
||||
Class<?>[] classes = healthCheckProcessorClass.getDeclaredClasses();
|
||||
Class<?> aClass = Arrays.stream(classes).findFirst().get();
|
||||
Constructor<?> constructor = aClass
|
||||
.getConstructor(HttpHealthCheckProcessor.class, HealthCheckInstancePublishInfo.class,
|
||||
HealthCheckTaskV2.class, Service.class);
|
||||
Object objects = constructor
|
||||
.newInstance(httpHealthCheckProcessor, healthCheckInstancePublishInfo, healthCheckTaskV2, service);
|
||||
List<Integer> codeList = Stream
|
||||
.of(HttpURLConnection.HTTP_OK, HttpURLConnection.HTTP_UNAVAILABLE, HttpURLConnection.HTTP_NOT_FOUND)
|
||||
.collect(Collectors.toList());
|
||||
for (Integer code : codeList) {
|
||||
when(restResult.getCode()).thenReturn(code);
|
||||
Method onReceive = aClass.getMethod("onReceive", RestResult.class);
|
||||
onReceive.invoke(objects, restResult);
|
||||
|
||||
//verify
|
||||
this.verifyCall(code);
|
||||
}
|
||||
}
|
||||
|
||||
private void verifyCall(int code) {
|
||||
switch (code) {
|
||||
case HttpURLConnection.HTTP_OK:
|
||||
verify(healthCheckCommon).checkOk(healthCheckTaskV2, service, "http:" + restResult.getCode());
|
||||
break;
|
||||
case HttpURLConnection.HTTP_UNAVAILABLE:
|
||||
verify(healthCheckCommon).checkFail(healthCheckTaskV2, service, "http:" + restResult.getCode());
|
||||
verify(healthCheckCommon)
|
||||
.reEvaluateCheckRT(healthCheckTaskV2.getCheckRtNormalized() * 2, healthCheckTaskV2,
|
||||
switchDomain.getHttpHealthParams());
|
||||
break;
|
||||
case HttpURLConnection.HTTP_NOT_FOUND:
|
||||
verify(healthCheckCommon).checkFailNow(healthCheckTaskV2, service, "http:" + restResult.getCode());
|
||||
verify(healthCheckCommon)
|
||||
.reEvaluateCheckRT(switchDomain.getHttpHealthParams().getMax(), healthCheckTaskV2,
|
||||
switchDomain.getHttpHealthParams());
|
||||
break;
|
||||
default:
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testOnError()
|
||||
throws NoSuchMethodException, IllegalAccessException, InvocationTargetException, InstantiationException {
|
||||
Class<HttpHealthCheckProcessor> healthCheckProcessorClass = HttpHealthCheckProcessor.class;
|
||||
Class<?>[] classes = healthCheckProcessorClass.getDeclaredClasses();
|
||||
Class<?> aClass = Arrays.stream(classes).findFirst().get();
|
||||
Constructor<?> constructor = aClass
|
||||
.getConstructor(HttpHealthCheckProcessor.class, HealthCheckInstancePublishInfo.class,
|
||||
HealthCheckTaskV2.class, Service.class);
|
||||
Object objects = constructor
|
||||
.newInstance(httpHealthCheckProcessor, healthCheckInstancePublishInfo, healthCheckTaskV2, service);
|
||||
Method onReceive = aClass.getMethod("onError", Throwable.class);
|
||||
onReceive.invoke(objects, connectException);
|
||||
|
||||
verify(healthCheckCommon)
|
||||
.checkFailNow(healthCheckTaskV2, service, "http:unable2connect:" + connectException.getMessage());
|
||||
verify(healthCheckCommon).reEvaluateCheckRT(switchDomain.getHttpHealthParams().getMax(), healthCheckTaskV2,
|
||||
switchDomain.getHttpHealthParams());
|
||||
}
|
||||
}
|
9
pom.xml
9
pom.xml
@ -162,6 +162,7 @@
|
||||
<hessian.version>4.0.63</hessian.version>
|
||||
<reflections.version>0.9.11</reflections.version>
|
||||
<mockito-all.version>1.10.19</mockito-all.version>
|
||||
<mockito-core.version>3.8.0</mockito-core.version>
|
||||
<hamcrest-all.version>1.3</hamcrest-all.version>
|
||||
<prometheus-simpleclient.version>0.5.0</prometheus-simpleclient.version>
|
||||
<tomcat-embed-jasper.version>9.0.40</tomcat-embed-jasper.version>
|
||||
@ -1029,6 +1030,12 @@
|
||||
<version>${mockito-all.version}</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.mockito</groupId>
|
||||
<artifactId>mockito-core</artifactId>
|
||||
<version>${mockito-core.version}</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.hamcrest</groupId>
|
||||
<artifactId>hamcrest-all</artifactId>
|
||||
@ -1066,6 +1073,6 @@
|
||||
<url>https://oss.sonatype.org/service/local/staging/deploy/maven2/</url>
|
||||
</repository>
|
||||
</distributionManagement>
|
||||
|
||||
|
||||
</project>
|
||||
|
||||
|
@ -1,49 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.test;
|
||||
|
||||
import junit.framework.Test;
|
||||
import junit.framework.TestCase;
|
||||
import junit.framework.TestSuite;
|
||||
|
||||
/**
|
||||
* Unit test for simple App.
|
||||
*/
|
||||
public class AppTest
|
||||
extends TestCase {
|
||||
/**
|
||||
* Create the test case
|
||||
*
|
||||
* @param testName name of the test case
|
||||
*/
|
||||
public AppTest(String testName) {
|
||||
super(testName);
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the suite of tests being tested
|
||||
*/
|
||||
public static Test suite() {
|
||||
return new TestSuite(AppTest.class);
|
||||
}
|
||||
|
||||
/**
|
||||
* Rigourous Test :-)
|
||||
*/
|
||||
public void testApp() {
|
||||
assertTrue(true);
|
||||
}
|
||||
}
|
@ -1,39 +0,0 @@
|
||||
/*
|
||||
* Copyright 1999-2018 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.nacos.test;
|
||||
|
||||
import com.alibaba.nacos.common.utils.JacksonUtils;
|
||||
import com.alibaba.nacos.naming.healthcheck.RsInfo;
|
||||
import org.junit.Test;
|
||||
|
||||
/**
|
||||
* @author <a href="mailto:liaochuntao@live.com">liaochuntao</a>
|
||||
* @author yangyi
|
||||
*
|
||||
* @deprecated It seems no necessary for super test class, will be removed.
|
||||
*/
|
||||
@Deprecated
|
||||
public class BaseTest {
|
||||
|
||||
@Test
|
||||
public void test_rs_json() {
|
||||
String json = "{\"cluster\":\"DEFAULT\",\"ip\":\"127.0.0.1\",\"metadata\":{},\"port\":60000,\"scheduled\":true,\"serviceName\":\"DEFAULT_GROUP@@jinhan9J7ye.Vj6hx.net\",\"weight\":1.0}";
|
||||
RsInfo client = JacksonUtils.toObj(json, RsInfo.class);
|
||||
System.out.println(client);
|
||||
}
|
||||
|
||||
}
|
@ -48,7 +48,7 @@ import java.util.Map;
|
||||
@FixMethodOrder(MethodSorters.JVM)
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class NacosAsyncRestTemplate_ITCase {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -47,7 +47,7 @@ import java.util.Map;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
@FixMethodOrder(MethodSorters.JVM)
|
||||
public class NacosRestTemplate_ITCase {
|
||||
|
||||
|
@ -51,7 +51,7 @@ import java.util.Map;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
@FixMethodOrder(MethodSorters.JVM)
|
||||
public class NacosRestTemplate_Interceptors_ITCase {
|
||||
|
||||
|
@ -45,7 +45,7 @@ import static org.junit.Assert.fail;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class ConfigAuth_ITCase extends AuthBase {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -43,7 +43,7 @@ import static org.junit.Assert.fail;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class NamingAuth_ITCase extends AuthBase {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -44,7 +44,7 @@ import java.util.concurrent.TimeUnit;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class Permission_ITCase extends HttpClient4Test {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -44,7 +44,7 @@ import java.util.concurrent.TimeUnit;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class Role_ITCase extends HttpClient4Test {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -46,7 +46,7 @@ import java.util.concurrent.TimeUnit;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class User_ITCase extends HttpClient4Test {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -25,10 +25,8 @@ import com.alibaba.nacos.core.cluster.lookup.LookupFactory;
|
||||
import com.alibaba.nacos.core.cluster.MemberLookup;
|
||||
import com.alibaba.nacos.core.cluster.lookup.StandaloneMemberLookup;
|
||||
import com.alibaba.nacos.sys.env.EnvUtil;
|
||||
import com.alibaba.nacos.sys.utils.ApplicationUtils;
|
||||
import com.alibaba.nacos.sys.utils.DiskUtils;
|
||||
import com.alibaba.nacos.sys.utils.InetUtils;
|
||||
import com.alibaba.nacos.test.BaseTest;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.junit.After;
|
||||
import org.junit.Assert;
|
||||
@ -49,7 +47,7 @@ import java.util.Map;
|
||||
* @author <a href="mailto:liaochuntao@live.com">liaochuntao</a>
|
||||
*/
|
||||
@FixMethodOrder(value = MethodSorters.NAME_ASCENDING)
|
||||
public class MemberLookup_ITCase extends BaseTest {
|
||||
public class MemberLookup_ITCase {
|
||||
|
||||
static final String path = Paths.get(System.getProperty("user.home"), "/member_look")
|
||||
.toString();
|
||||
|
@ -29,6 +29,7 @@ import com.alibaba.nacos.sys.env.EnvUtil;
|
||||
import org.junit.After;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Ignore;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
@ -50,7 +51,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.randomDomainName;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class AutoDeregisterInstance_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
@ -89,6 +90,7 @@ public class AutoDeregisterInstance_ITCase {
|
||||
* @throws Exception
|
||||
*/
|
||||
@Test
|
||||
@Ignore("Nacos 2.0 does not support HTTP beat check any more, it uses gRPC instead.")
|
||||
public void autoDregDomClustersTest() throws Exception {
|
||||
String serviceName = randomDomainName();
|
||||
|
||||
@ -100,7 +102,7 @@ public class AutoDeregisterInstance_ITCase {
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
NacosNamingService namingServiceImpl = (NacosNamingService) naming;
|
||||
|
||||
@ -126,6 +128,7 @@ public class AutoDeregisterInstance_ITCase {
|
||||
* @throws Exception
|
||||
*/
|
||||
@Test
|
||||
@Ignore("Nacos 2.0 does not support HTTP beat check any more, it uses gRPC instead.")
|
||||
public void autoDregDomTest() throws Exception {
|
||||
String serviceName = randomDomainName();
|
||||
|
||||
@ -137,7 +140,7 @@ public class AutoDeregisterInstance_ITCase {
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
NacosNamingService namingServiceImpl = (NacosNamingService) naming;
|
||||
|
||||
@ -158,19 +161,19 @@ public class AutoDeregisterInstance_ITCase {
|
||||
* @throws Exception
|
||||
*/
|
||||
@Test
|
||||
@Ignore("Nacos 2.0 does not support HTTP beat check any more, it uses gRPC instead.")
|
||||
public void autoRegDomTest() throws Exception {
|
||||
|
||||
String serviceName = randomDomainName();
|
||||
|
||||
naming.registerInstance(serviceName, "127.0.0.1", TEST_PORT);
|
||||
naming.registerInstance(serviceName, "127.0.0.2", TEST_PORT);
|
||||
|
||||
TimeUnit.SECONDS.sleep(5);
|
||||
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
NacosNamingService namingServiceImpl = (NacosNamingService) naming;
|
||||
|
||||
@ -178,11 +181,11 @@ public class AutoDeregisterInstance_ITCase {
|
||||
.removeBeatInfo(Constants.DEFAULT_GROUP + Constants.SERVICE_INFO_SPLITER + serviceName, "127.0.0.1",
|
||||
TEST_PORT);
|
||||
|
||||
verifyInstanceList(instances, 1, serviceName);
|
||||
verifyInstanceList(instances, 0, serviceName);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(1, instances.size());
|
||||
Assert.assertEquals(0, instances.size());
|
||||
BeatInfo beatInfo = new BeatInfo();
|
||||
beatInfo.setServiceName(Constants.DEFAULT_GROUP + Constants.SERVICE_INFO_SPLITER + serviceName);
|
||||
beatInfo.setIp("127.0.0.1");
|
||||
@ -191,11 +194,11 @@ public class AutoDeregisterInstance_ITCase {
|
||||
NamingTestUtils.getBeatReactorByReflection(namingServiceImpl)
|
||||
.addBeatInfo(Constants.DEFAULT_GROUP + Constants.SERVICE_INFO_SPLITER + serviceName, beatInfo);
|
||||
|
||||
verifyInstanceList(instances, 2, serviceName);
|
||||
verifyInstanceList(instances, 1, serviceName);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
}
|
||||
|
||||
/**
|
||||
@ -204,19 +207,19 @@ public class AutoDeregisterInstance_ITCase {
|
||||
* @throws Exception
|
||||
*/
|
||||
@Test
|
||||
@Ignore("Nacos 2.0 does not support HTTP beat check any more, it uses gRPC instead.")
|
||||
public void autoRegDomClustersTest() throws Exception {
|
||||
|
||||
String serviceName = randomDomainName();
|
||||
|
||||
naming.registerInstance(serviceName, "127.0.0.1", TEST_PORT, "c1");
|
||||
naming.registerInstance(serviceName, "127.0.0.2", TEST_PORT, "c2");
|
||||
|
||||
TimeUnit.SECONDS.sleep(5);
|
||||
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
NacosNamingService namingServiceImpl = (NacosNamingService) naming;
|
||||
|
||||
@ -242,7 +245,7 @@ public class AutoDeregisterInstance_ITCase {
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
instances = naming.getAllInstances(serviceName, Arrays.asList("c2"));
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
@ -55,7 +55,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class CPInstancesAPI_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
@ -130,7 +130,8 @@ public class CPInstancesAPI_ITCase {
|
||||
instance.setIp("11.11.11.11");
|
||||
instance.setPort(80);
|
||||
naming1.registerInstance(serviceName, TEST_GROUP_1, instance);
|
||||
|
||||
TimeUnit.SECONDS.sleep(3L);
|
||||
naming1.deregisterInstance(serviceName, TEST_GROUP_1, instance);
|
||||
namingServiceDelete(serviceName, TEST_NAMESPACE_1, TEST_GROUP_1);
|
||||
}
|
||||
|
||||
@ -175,11 +176,11 @@ public class CPInstancesAPI_ITCase {
|
||||
* @ExpectResult :
|
||||
*/
|
||||
@Test
|
||||
public void deleteService_hasInstace() throws Exception {
|
||||
public void deleteService_hasInstace() {
|
||||
String serviceName = NamingBase.randomDomainName();
|
||||
namingServiceCreate(serviceName, TEST_NAMESPACE_1);
|
||||
|
||||
ResponseEntity<String> response = request(NamingBase.NAMING_CONTROLLER_PATH + "/instance",
|
||||
ResponseEntity<String> registerResponse = request(NamingBase.NAMING_CONTROLLER_PATH + "/instance",
|
||||
Params.newParams()
|
||||
.appendParam("serviceName", serviceName)
|
||||
.appendParam("ip", "11.11.11.11")
|
||||
@ -188,8 +189,16 @@ public class CPInstancesAPI_ITCase {
|
||||
.done(),
|
||||
String.class,
|
||||
HttpMethod.POST);
|
||||
Assert.assertTrue(response.getStatusCode().is2xxSuccessful());
|
||||
namingServiceDelete(serviceName, TEST_NAMESPACE_1);
|
||||
Assert.assertTrue(registerResponse.getStatusCode().is2xxSuccessful());
|
||||
|
||||
ResponseEntity<String> deleteServiceResponse = request(NamingBase.NAMING_CONTROLLER_PATH + "/service",
|
||||
Params.newParams()
|
||||
.appendParam("serviceName", serviceName)
|
||||
.appendParam("namespaceId", TEST_NAMESPACE_1)
|
||||
.done(),
|
||||
String.class,
|
||||
HttpMethod.DELETE);
|
||||
Assert.assertTrue(deleteServiceResponse.getStatusCode().is4xxClientError());
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -40,7 +40,7 @@ import java.util.concurrent.TimeUnit;
|
||||
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class ClientBeat_ITCase extends NamingBase {
|
||||
|
||||
private NamingService naming;
|
||||
@ -81,7 +81,7 @@ public class ClientBeat_ITCase extends NamingBase {
|
||||
TimeUnit.SECONDS.sleep(2L);
|
||||
|
||||
List<Instance> list = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(2, list.size());
|
||||
Assert.assertEquals(1, list.size());
|
||||
for (Instance instance1 : list) {
|
||||
Assert.assertEquals("1.2.3.4", instance1.getIp());
|
||||
Assert.assertTrue(instance1.getPort() == 80 || instance1.getPort() == 81);
|
||||
@ -95,7 +95,7 @@ public class ClientBeat_ITCase extends NamingBase {
|
||||
TimeUnit.SECONDS.sleep(35L);
|
||||
|
||||
list = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(2, list.size());
|
||||
Assert.assertEquals(1, list.size());
|
||||
for (Instance instance1 : list) {
|
||||
Assert.assertEquals("1.2.3.4", instance1.getIp());
|
||||
Assert.assertTrue(instance1.getPort() == 80 || instance1.getPort() == 81);
|
||||
@ -120,7 +120,7 @@ public class ClientBeat_ITCase extends NamingBase {
|
||||
TimeUnit.SECONDS.sleep(35L);
|
||||
|
||||
list = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(2, list.size());
|
||||
Assert.assertEquals(1, list.size());
|
||||
for (Instance instance1 : list) {
|
||||
Assert.assertEquals("1.2.3.4", instance1.getIp());
|
||||
Assert.assertTrue(instance1.getPort() == 80 || instance1.getPort() == 81);
|
||||
|
@ -49,7 +49,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
@Ignore
|
||||
public class Cmdb_ITCase {
|
||||
|
||||
|
@ -45,7 +45,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.randomDomainName;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class DeregisterInstance_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
@ -87,13 +87,12 @@ public class DeregisterInstance_ITCase {
|
||||
String serviceName = randomDomainName();
|
||||
System.out.println(serviceName);
|
||||
naming.registerInstance(serviceName, "127.0.0.1", TEST_PORT);
|
||||
naming.registerInstance(serviceName, "127.0.0.2", TEST_PORT);
|
||||
|
||||
List<Instance> instances = naming.getAllInstances(serviceName);
|
||||
verifyInstanceList(instances, 2, serviceName);
|
||||
verifyInstanceList(instances, 1, serviceName);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
naming.deregisterInstance(serviceName, "127.0.0.1", TEST_PORT);
|
||||
|
||||
@ -101,15 +100,7 @@ public class DeregisterInstance_ITCase {
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(instances.size(), 1);
|
||||
Assert.assertEquals(instances.get(0).getIp(), "127.0.0.2");
|
||||
|
||||
naming.deregisterInstance(serviceName, "127.0.0.2", TEST_PORT);
|
||||
|
||||
TimeUnit.SECONDS.sleep(3);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(0, instances.size());
|
||||
Assert.assertEquals(instances.size(), 0);
|
||||
}
|
||||
|
||||
/**
|
||||
@ -124,14 +115,13 @@ public class DeregisterInstance_ITCase {
|
||||
System.out.println(serviceName);
|
||||
|
||||
naming.registerInstance(serviceName, "127.0.0.1", TEST_PORT, "c1");
|
||||
naming.registerInstance(serviceName, "127.0.0.2", TEST_PORT, "c2");
|
||||
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
verifyInstanceList(instances, 2, serviceName);
|
||||
verifyInstanceList(instances, 1, serviceName);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(instances.size(), 2);
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
naming.deregisterInstance(serviceName, "127.0.0.1", TEST_PORT, "c1");
|
||||
|
||||
@ -139,12 +129,7 @@ public class DeregisterInstance_ITCase {
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
instances = naming.getAllInstances(serviceName, Arrays.asList("c2"));
|
||||
Assert.assertEquals(instances.size(), 1);
|
||||
|
||||
instances = naming.getAllInstances(serviceName, Arrays.asList("c1"));
|
||||
Assert.assertEquals(0, instances.size());
|
||||
}
|
||||
|
||||
|
||||
@ -161,14 +146,12 @@ public class DeregisterInstance_ITCase {
|
||||
String serviceName = randomDomainName();
|
||||
|
||||
naming.registerInstance(serviceName, "127.0.0.1", TEST_PORT, "c1");
|
||||
naming.registerInstance(serviceName, "127.0.0.2", TEST_PORT, "c2");
|
||||
|
||||
List<Instance> instances;
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
verifyInstanceList(instances, 2, serviceName);
|
||||
verifyInstanceList(instances, 1, serviceName);
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
naming.deregisterInstance(serviceName, "127.0.0.1", TEST_PORT, "c1");
|
||||
|
||||
@ -176,14 +159,6 @@ public class DeregisterInstance_ITCase {
|
||||
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
instances = naming.getAllInstances(serviceName, Arrays.asList("c2"));
|
||||
Assert.assertEquals(1, instances.size());
|
||||
|
||||
naming.deregisterInstance(serviceName,"127.0.0.2", TEST_PORT, "c2");
|
||||
TimeUnit.SECONDS.sleep(5);
|
||||
instances = naming.getAllInstances(serviceName);
|
||||
Assert.assertEquals(0, instances.size());
|
||||
}
|
||||
|
||||
|
@ -48,7 +48,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class MultiTenant_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -60,7 +60,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.randomDomainName;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class MultiTenant_InstanceAPI_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -45,7 +45,7 @@ import java.util.concurrent.TimeUnit;
|
||||
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = NamingApp.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class NamingHttpClientProxy_ITCase {
|
||||
@LocalServerPort
|
||||
private int port;
|
||||
|
@ -27,8 +27,6 @@ import com.alibaba.nacos.api.naming.pojo.Instance;
|
||||
import com.alibaba.nacos.api.naming.pojo.Service;
|
||||
import com.alibaba.nacos.api.selector.ExpressionSelector;
|
||||
import com.alibaba.nacos.api.selector.NoneSelector;
|
||||
import com.alibaba.nacos.sys.utils.ApplicationUtils;
|
||||
import com.alibaba.nacos.test.BaseTest;
|
||||
import org.junit.After;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
@ -51,8 +49,8 @@ import static com.alibaba.nacos.test.naming.NamingBase.randomDomainName;
|
||||
**/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
public class NamingMaintainService_ITCase extends BaseTest {
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class NamingMaintainService_ITCase {
|
||||
|
||||
private NamingMaintainService namingMaintainService;
|
||||
private NamingService namingService;
|
||||
|
@ -46,7 +46,7 @@ import java.util.concurrent.TimeUnit;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
"server.servlet.context-path=/nacos"}, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class RegisterInstance_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
@ -95,7 +95,7 @@ public class RegisterInstance_ITCase {
|
||||
naming.registerInstance(serviceName, "127.0.0.2", 80, "c2");
|
||||
List<Instance> instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(2, instances.size());
|
||||
Assert.assertEquals(1, instances.size());
|
||||
}
|
||||
|
||||
/**
|
||||
@ -114,8 +114,6 @@ public class RegisterInstance_ITCase {
|
||||
List<Instance> instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(1, instances.size());
|
||||
Assert.assertTrue(instances.get(0).getInstanceId().contains(serviceName));
|
||||
//Assert.assertEquals(instances.get(0).getService().getName(), serviceName);
|
||||
Assert.assertEquals(instances.get(0).getIp(), NamingBase.TEST_IP_4_DOM_1);
|
||||
Assert.assertEquals(instances.get(0).getPort(), NamingBase.TEST_PORT);
|
||||
}
|
||||
@ -139,20 +137,14 @@ public class RegisterInstance_ITCase {
|
||||
List<Instance> instances = naming.getAllInstances(serviceName);
|
||||
|
||||
Assert.assertEquals(1, instances.size());
|
||||
Assert.assertTrue(instances.get(0).getInstanceId().contains(serviceName));
|
||||
//Assert.assertEquals(instances2.get(0).getService().getName(), serviceName);
|
||||
Assert.assertEquals(instances.get(0).getIp(), NamingBase.TEST_IP_4_DOM_1);
|
||||
Assert.assertEquals(instances.get(0).getPort(), NamingBase.TEST_PORT);
|
||||
//Assert.assertEquals(instances.get(0).getCluster().getName(), TEST_NEW_CLUSTER_4_DOM_1);
|
||||
|
||||
List<Instance> instances2 = naming.getAllInstances(serviceName, Arrays.asList(NamingBase.TEST_NEW_CLUSTER_4_DOM_1));
|
||||
|
||||
Assert.assertEquals(instances2.size(), 1);
|
||||
Assert.assertTrue(instances2.get(0).getInstanceId().contains(serviceName));
|
||||
//Assert.assertEquals(instances2.get(0).getService().getName(), serviceName);
|
||||
Assert.assertEquals(instances2.get(0).getIp(), NamingBase.TEST_IP_4_DOM_1);
|
||||
Assert.assertEquals(instances2.get(0).getPort(), NamingBase.TEST_PORT);
|
||||
//Assert.assertEquals(instances2.get(0).getCluster().getName(), TEST_NEW_CLUSTER_4_DOM_1);
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -35,7 +35,7 @@ import java.net.URL;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class RestAPI_ITCase extends NamingBase {
|
||||
|
||||
@LocalServerPort
|
||||
|
@ -47,7 +47,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class SelectInstances_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -42,7 +42,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class SelectOneHealthyInstance_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -45,7 +45,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.randomDomainName;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class ServiceListTest_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -46,7 +46,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class SubscribeCluster_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -48,7 +48,7 @@ import java.util.concurrent.TimeUnit;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class Subscribe_ITCase extends RestAPI_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -45,7 +45,7 @@ import static com.alibaba.nacos.test.naming.NamingBase.*;
|
||||
*/
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = Nacos.class, properties = {"server.servlet.context-path=/nacos"},
|
||||
webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
|
||||
public class Unsubscribe_ITCase {
|
||||
|
||||
private NamingService naming;
|
||||
|
@ -17,8 +17,6 @@
|
||||
|
||||
<configuration>
|
||||
|
||||
<springProperty scope="context" name="logPath" source="nacos.logging.path" defaultValue="${nacos.home}/logs"/>
|
||||
|
||||
<!--<appender name="DefaultAppender" class="ch.qos.logback.core.ConsoleAppender">-->
|
||||
<!--<append>true</append>-->
|
||||
<!--<encoder>-->
|
||||
|
Loading…
Reference in New Issue
Block a user