Add try-with-resources support for Entry class (#550)
- `Entry` class now implements `AutoCloseable` so it supports try-with-resources (`close` method adapts to `exit`) Signed-off-by: Eric Zhao <sczyh16@gmail.com>
This commit is contained in:
parent
e857b5cef7
commit
775484ba61
|
|
@ -48,7 +48,7 @@ import com.alibaba.csp.sentinel.context.Context;
|
||||||
* @see Context
|
* @see Context
|
||||||
* @see ContextUtil
|
* @see ContextUtil
|
||||||
*/
|
*/
|
||||||
public abstract class Entry {
|
public abstract class Entry implements AutoCloseable {
|
||||||
|
|
||||||
private static final Object[] OBJECTS0 = new Object[0];
|
private static final Object[] OBJECTS0 = new Object[0];
|
||||||
|
|
||||||
|
|
@ -70,6 +70,11 @@ public abstract class Entry {
|
||||||
return resourceWrapper;
|
return resourceWrapper;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Complete the current resource entry and restore the entry stack in context.
|
||||||
|
*
|
||||||
|
* @throws ErrorEntryFreeException if entry in current context does not match current entry
|
||||||
|
*/
|
||||||
public void exit() throws ErrorEntryFreeException {
|
public void exit() throws ErrorEntryFreeException {
|
||||||
exit(1, OBJECTS0);
|
exit(1, OBJECTS0);
|
||||||
}
|
}
|
||||||
|
|
@ -78,11 +83,21 @@ public abstract class Entry {
|
||||||
exit(count, OBJECTS0);
|
exit(count, OBJECTS0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Equivalent to {@link #exit()}. Support try-with-resources since JDK 1.7.
|
||||||
|
*
|
||||||
|
* @since 1.5.0
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void close() {
|
||||||
|
exit();
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Exit this entry. This method should invoke if and only if once at the end of the resource protection.
|
* Exit this entry. This method should invoke if and only if once at the end of the resource protection.
|
||||||
*
|
*
|
||||||
* @param count tokens to release.
|
* @param count tokens to release.
|
||||||
* @param args
|
* @param args extra parameters
|
||||||
* @throws ErrorEntryFreeException, if {@link Context#getCurEntry()} is not this entry.
|
* @throws ErrorEntryFreeException, if {@link Context#getCurEntry()} is not this entry.
|
||||||
*/
|
*/
|
||||||
public abstract void exit(int count, Object... args) throws ErrorEntryFreeException;
|
public abstract void exit(int count, Object... args) throws ErrorEntryFreeException;
|
||||||
|
|
@ -91,7 +106,7 @@ public abstract class Entry {
|
||||||
* Exit this entry.
|
* Exit this entry.
|
||||||
*
|
*
|
||||||
* @param count tokens to release.
|
* @param count tokens to release.
|
||||||
* @param args
|
* @param args extra parameters
|
||||||
* @return next available entry after exit, that is the parent entry.
|
* @return next available entry after exit, that is the parent entry.
|
||||||
* @throws ErrorEntryFreeException, if {@link Context#getCurEntry()} is not this entry.
|
* @throws ErrorEntryFreeException, if {@link Context#getCurEntry()} is not this entry.
|
||||||
*/
|
*/
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue