{
+
+ private final int index;
+
+ private final String name;
+
+ private final AttributeInitFunction function;
+
+ AttributeKey(int index, String name, AttributeInitFunction function) {
+ this.name = name;
+ this.index = index;
+ this.function = function;
+ }
+
+ public int getIndex() {
+ return index;
+ }
+
+ public String getName() {
+ return name;
+ }
+
+ public AttributeInitFunction getFunction() {
+ return function;
+ }
+
+ @Override
+ public String toString() {
+ return "name: " + name + ", index: " + index;
+ }
+
+}
diff --git a/firenio-core/src/main/java/com/firenio/collection/AttributeMap.java b/firenio-core/src/main/java/com/firenio/collection/AttributeMap.java
new file mode 100644
index 000000000..d60515a57
--- /dev/null
+++ b/firenio-core/src/main/java/com/firenio/collection/AttributeMap.java
@@ -0,0 +1,150 @@
+/*
+ * Copyright 2015 The FireNio Project
+ *
+ * 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.firenio.collection;
+
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * use for static variable, eg:
+ *
+ * static final AttributeKey KEY_NAME = AttributeMap.valueOfKey(Channel.class, "KEY_NAME");
+ *
+ */
+//TODO 考虑多个ChannelContext时，每个Context里Channel存储的数据key不一样，会造成内存浪费
+public abstract class AttributeMap {
+
+ private static final Map, AttributeKeys> INDEX_MAPPING = new ConcurrentHashMap<>();
+
+ private final Object[] attributes;
+
+ private volatile int memoryBarrier = 0;
+
+ public AttributeMap() {
+ AttributeKeys keys = getKeys();
+ if (keys == null) {
+ attributes = null;
+ return;
+ }
+ keys.initialized = true;
+ this.attributes = new Object[keys.getCounter()];
+ for (AttributeKey key : keys.keys.values()) {
+ AttributeInitFunction function = key.getFunction();
+ function.setValue(this, key.getIndex());
+ }
+ }
+
+ public T getAttribute(AttributeKey key) {
+ return (T) getAttribute(key.getIndex());
+ }
+
+ public T getAttributeUnsafe(AttributeKey key) {
+ return (T) getAttributeUnsafe(key.getIndex());
+ }
+
+ public Object getAttribute(int key) {
+ memoryBarrier = 1;
+ return attributes[key];
+ }
+
+ public Object getAttributeUnsafe(int key) {
+ return attributes[key];
+ }
+
+ public void setAttribute(AttributeKey key, Object value) {
+ attributes[key.getIndex()] = value;
+ }
+
+ public void setAttribute(int key, Object value) {
+ memoryBarrier = 1;
+ attributes[key] = value;
+ }
+
+ public void setAttributeUnsafe(int key, Object value) {
+ attributes[key] = value;
+ }
+
+ public static AttributeKeys getKeys(Class clazz) {
+ return INDEX_MAPPING.get(clazz);
+ }
+
+ public static AttributeKey valueOfKey(Class clazz, String name) {
+ return valueOfKey(clazz, name, null);
+ }
+
+ public static AttributeKey valueOfKey(Class clazz, String name, AttributeInitFunction function) {
+ AttributeKeys attributeKeys = INDEX_MAPPING.get(clazz);
+ if (attributeKeys == null) {
+ synchronized (INDEX_MAPPING) {
+ attributeKeys = INDEX_MAPPING.get(clazz);
+ if (attributeKeys == null) {
+ attributeKeys = new AttributeKeys();
+ INDEX_MAPPING.put(clazz, attributeKeys);
+ }
+ }
+ }
+ return attributeKeys.valueOf(name, function);
+ }
+
+ protected int getMemoryBarrier() {
+ return memoryBarrier;
+ }
+
+ protected abstract AttributeKeys getKeys();
+
+ public static class AttributeKeys {
+
+ final Map keys = new HashMap<>();
+ final Map ro_keys = Collections.unmodifiableMap(keys);
+
+ int counter = 0;
+ volatile boolean initialized = false;
+
+ synchronized AttributeKey valueOf(String name, AttributeInitFunction function) {
+ if (initialized) {
+ throw new RuntimeException("incorrect usage, AttributeKey can only be static constant");
+ }
+ if (function == null) {
+ function = NULL_VALUE_ATTRIBUTE_INIT_FUNCTION;
+ }
+ keys.put(name, new AttributeKey(counter++, name, function));
+ return keys.get(name);
+ }
+
+ public Map getKeys() {
+ return ro_keys;
+ }
+
+ public int getCounter() {
+ return counter;
+ }
+
+ }
+
+ public interface AttributeInitFunction {
+
+ void setValue(AttributeMap map, int key);
+
+ }
+
+ static final AttributeInitFunction NULL_VALUE_ATTRIBUTE_INIT_FUNCTION = new AttributeInitFunction() {
+ @Override
+ public void setValue(AttributeMap map, int key) { }
+ };
+
+}
diff --git a/firenio-core/src/main/java/com/firenio/collection/AttributesImpl.java b/firenio-core/src/main/java/com/firenio/collection/AttributesImpl.java
deleted file mode 100644
index d5d6f5f04..000000000
--- a/firenio-core/src/main/java/com/firenio/collection/AttributesImpl.java
+++ /dev/null
@@ -1,56 +0,0 @@
-/*
- * Copyright 2015 The FireNio Project
- *
- * 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.firenio.collection;
-
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Set;
-
-public class AttributesImpl implements Attributes {
-
- private final Map attributes = new HashMap<>();
-
- @Override
- public Map attributes() {
- return attributes;
- }
-
- @Override
- public void clearAttributes() {
- this.attributes.clear();
- }
-
- @Override
- public Object getAttribute(Object key) {
- return this.attributes.get(key);
- }
-
- @Override
- public Set