好的...我看到了您的解决方案的两个问题：
- 对
tags 的访问不安全，任何其他类都可以操纵它。更多内容如下。
- 就速度而言，检索不是最佳的。然而，这在 Java 社区中被广泛接受，因为像 ThreadLocal 这样的可憎之物也存在。所以让我们接受这一点。
因此，您需要做的是保护地图免受恶意访问。
我已经构建了三个增加安全考虑的示例。
示例包结构：
stackoverflow.trait0r/
AnalysisTag.java
Trait0r.java
/bad/
BadResource.java
/better/
BetterResource.java
BetterResourceContainer.java
/best/
BestResource.java
BestResourceContainer.java
SamePackageIllegalAccessTest.java
示例 1：您的，只是改进了一点
package stackoverflow.trait0r.bad;
import java.util.WeakHashMap;
import stackoverflow.trait0r.AnalysisTag;
public interface BadResource {
/* best to always use full type (WeakHashMap instead of Map) when in control of code.
* so downcast is always possible and upcast never necessary.
* Plus access to the references object is usually faster
*/
WeakHashMap<BadResource, AnalysisTag<?>> tags = new WeakHashMap<>();
default AnalysisTag<?> getTag() {
return tags.get(this);
}
default BadResource tag(final AnalysisTag<?> tag) {
tags.put(this, tag);
return this;
}
}
示例 2：更好一点，使用受保护的包
package stackoverflow.trait0r.better;
import stackoverflow.trait0r.AnalysisTag;
public interface BetterResource {
default AnalysisTag<?> getTag() {
return BetterResourceConainer.tags.get(this);
}
default BetterResource tag(final AnalysisTag<?> tag) {
BetterResourceConainer.tags.put(this, tag);
return this;
}
}
和
package stackoverflow.trait0r.better;
import java.util.WeakHashMap;
import stackoverflow.trait0r.AnalysisTag;
class BetterResourceConainer {
static /* package */ WeakHashMap<BetterResource, AnalysisTag<?>> tags = new WeakHashMap<>();
}
示例 3：最佳保护：包保护和线程访问检查
package stackoverflow.trait0r.best;
import stackoverflow.trait0r.AnalysisTag;
public interface BestResource {
default AnalysisTag<?> getTag() {
return BestResourceConainer.getTag(this);
}
default BestResource tag(final AnalysisTag<?> tag) {
BestResourceConainer.setTag(this, tag);
return this;
}
}
和
package stackoverflow.trait0r.best;
import java.util.WeakHashMap;
import stackoverflow.trait0r.AnalysisTag;
class BestResourceConainer {
static private WeakHashMap<BestResource, AnalysisTag<?>> tags = new WeakHashMap<>();
static private void ensureAccess() {
final int CALLER_INDEX = 3;
final StackTraceElement[] st = Thread.currentThread().getStackTrace();
if (st.length < CALLER_INDEX + 1) throw new IllegalAccessError("Cannot access BestResourceConainer via external caller!");
final StackTraceElement ste = st[CALLER_INDEX];
final boolean grantAccess = BestResource.class.getName().equals(ste.getClassName());
if (!grantAccess) throw new IllegalAccessError("Cannot access BestResourceConainer via external caller!");
}
static /* package */ void setTag(final BestResource pRes, final AnalysisTag<?> pTag) {
ensureAccess();
tags.put(pRes, pTag);
}
static AnalysisTag<?> getTag(final BestResource pRes) {
ensureAccess();
return tags.get(pRes);
}
}
以及一次肮脏的小黑客尝试：
package stackoverflow.trait0r.best;
import stackoverflow.trait0r.AnalysisTag;
public class SamePackageIllegalAccessTest {
public static void main(final String[] args) {
try {
BestResourceConainer.setTag(new BestResource() {/**/}, new AnalysisTag<>("[ malicious hack ]"));
System.out.println("Write successful!");
} catch (final Error e) {
System.out.println("Write access failed as expected");
}
try {
final AnalysisTag<?> tag = BestResourceConainer.getTag(new BestResource() {/**/});
System.out.println("TAG: " + tag);
// ThreadLocal<String> x;
} catch (final Error e) {
System.out.println("Read access failed as expected");
}
}
}
如您所见，如果您真的想要对tags 映射进行某种安全管理，有一些技术可以实现。
但是请注意，您可能必须调整 CALLER_INDEX，如果从一开始就将 BestResource 类替换为相同类名的另一个实现，这仍然可能被欺骗。