类Metadata
提供对读取和写入元数据数值的访问,元数据数值在调用期间交换。
key 容许关联到多个值。
这个类不是线程安全,实现应该保证 header 的读取和写入不在多个线程中并发发生。
类定义
package io.grpc;
@NotThreadSafe
public final class Metadata{}
类属性
// 所有二进制 header 在他们的名字中应该有这个后缀。相反也是。
// 它的值是"-bin"。ASCII header 的名字一定不能以这个结尾。
public static final String BINARY_HEADER_SUFFIX = "-bin";
// 保存的数据
private byte[][] namesAndValues;
// 当前未缩放的 header 数量
private int size;
构造函数
// 构造函数,当应用层想发送元数据时调用
public Metadata() {}
// 构造函数, 当传输层接收到二进制元数据时调用。
Metadata(byte[]... binaryValues) {
// 注意这里长度除以 2 了
this(binaryValues.length / 2, binaryValues);
}
Metadata(int usedNames, byte[]... binaryValues) {
size = usedNames;
namesAndValues = binaryValues;
}
和size相关的几个私有方法:
len()
private int len() { return size * 2; }
headerCount(): 返回在这个元数据中的键值 header 的总数量,包含重复
public int headerCount() { return size; }
存取数据的方法
containsKey(): 如果给定的key有值则返回true,注意这里没有检查value是否可能为空列表
public boolean containsKey(Key<?> key) { for (int i = 0; i < size; i++) { if (bytesEqual(key.asciiName(), name(i))) { return true; } } return false; }
注意这里是线性搜索,因此如果后面跟有 get() 或者 getAll() 方法,最好直接调用他们然后检查返回值是否为null。
get(): 返回最后一个用给定name添加的元数据项,作为T解析,如果没有则返回null
public < T > T get(Key < T > key) { for (int i = size - 1; i >= 0; i--) { if (bytesEqual(key.asciiName(), name(i))) { return key.parseBytes(value(i)); } } return null; }
getAll(): 返回所有用给定name添加的元数据项,作为T解析,如果没有则返回null。顺序和接收到的一致。
public < T> Iterable< T> getAll(final Key< T> key) { for (int i = 0; i < size; i++) { if (bytesEqual(key.asciiName(), name(i))) { return new IterableAt<T>(key, i); } } return null; }
这里一旦返现有 key 匹配,就直接new 一个 IterableAt 对象。后面对所有 同样 name 的游离是通过 IterableAt 来实现。
keys(): 返回存储中的所有key的不可变集合
@SuppressWarnings("deprecation") // The String ctor is deprecated, but fast. public Set<String> keys() { if (isEmpty()) { return Collections.emptySet(); } Set<String> ks = new HashSet<String>(size); for (int i = 0; i < size; i++) { ks.add(new String(name(i), 0 /* hibyte */)); } // immutable in case we decide to change the implementation later. return Collections.unmodifiableSet(ks); }
实现逻辑很简单,但是有意思的是对 String 构造方法的使用。
put(): 添加键值对。如果 key 已经有值,值被添加到最后。同一个key的重复的值是容许的。
public < T > void put(Key< T> key, T value) { Preconditions.checkNotNull(key, "key"); Preconditions.checkNotNull(value, "value"); maybeExpand(); name(size, key.asciiName()); value(size, key.toBytes(value)); size++; }
remove(): 删除key的第一个出现的value,如果删除成功返回true,如果没有找到返回false
public < T> boolean remove(Key< T> key, T value) { Preconditions.checkNotNull(key, "key"); Preconditions.checkNotNull(value, "value"); for (int i = 0; i < size; i++) { // 从头开始游历 if (!bytesEqual(key.asciiName(), name(i))) { // 找 匹配的name continue; } @SuppressWarnings("unchecked") T stored = key.parseBytes(value(i)); if (!value.equals(stored)) { // 如果 value 不匹配,跳过 continue; } // name 和 value 都匹配了,准备做删除 int writeIdx = i * 2; int readIdx = (i + 1) * 2; int readLen = len() - readIdx; // 将后面的数据向前复制 System.arraycopy(namesAndValues, readIdx, namesAndValues, writeIdx, readLen); size -= 1; // 将最后一个位置的数据设置为null name(size, null); value(size, null); return true; } return false; }
removeAll(): 删除给定key的所有值并返回这些被删除的值。如果没有找到值,则返回null
public <T> Iterable<T> removeAll(final Key<T> key) { if (isEmpty()) { return null; } int writeIdx = 0; int readIdx = 0; List<T> ret = null; for (; readIdx < size; readIdx++) { if (bytesEqual(key.asciiName(), name(readIdx))) { ret = ret != null ? ret : new LinkedList<T>(); ret.add(key.parseBytes(value(readIdx))); continue; } name(writeIdx, name(readIdx)); value(writeIdx, value(readIdx)); writeIdx++; } int newSize = writeIdx; // Multiply by two since namesAndValues is interleaved. Arrays.fill(namesAndValues, writeIdx * 2, len(), null); size = newSize; return ret; }
discardAll(): 删除给定key的所有值但是不返回这些被删除的值。相比removeAll()方法有细微的性能提升,如果不需要使用之前的值。
内部定义的类
Marshaller
定义了两个 Marshaller 的interface,分别用于处理二进制和ASCII字符的值:
- BinaryMarshaller: 用于序列化为原始二进制的元数据值的装配器
AsciiMarshaller: 用于序列化为 ASCII 字符的元数据值的装配器,值只包含下列字符:
- 空格:0x20,但是不能在值的开头或者末尾。开始或者末尾的空白字符可能被去除。
- ASCII 可见字符 (0x21-0x7E)
然后实现了这两个接口的内部匿名类:
public static final BinaryMarshaller<byte[]> BINARY_BYTE_MARSHALLER = new BinaryMarshaller<byte[]>() {};
public static final AsciiMarshaller<String> ASCII_STRING_MARSHALLER = new AsciiMarshaller<String>() {};
static final AsciiMarshaller<Integer> INTEGER_MARSHALLER = new AsciiMarshaller<Integer>() {};
内部类 Metadata.Key
元数据项的key。考虑到元数据的解析和序列化。
key名称中的有效字符
在key的名字中仅容许下列ASCII字符:
- 数字: 0-9
- 大写字符: A-Z(标准化到小写)
- 小写字符: a-z
- 特殊字符: -_.
这是 HTTP 字段名规则的一个严格子集。应用不应该发送或者接收带有非法key名称的元数据。当然,gRPC 类库可能加工任何接收到的元数据,即使它不遵守上面的限制。此外,如果元数据包含不遵守的字段名,他们也将被发送。在这种方式下,未知元数据字段被解析,序列化和加工,但是从不中断。他们类似protobuf的未知字段。
注意,有一个有效的 HTTP/2 token字符的子集,被定义在 RFC7230 3.2.6 节和 RFC5234 B.1节。
// 字段名的有效字符被定义在 RFC7230 和 RFC5234
private static final BitSet VALID_T_CHARS = generateValidTChars();
//在这里看到,仅有三个特殊字符,数字和小写字母是有效的
private static BitSet generateValidTChars() {
BitSet valid = new BitSet(0x7f);
valid.set('-');
valid.set('_');
valid.set('.');
for (char c = '0'; c <= '9'; c++) {
valid.set(c);
}
// 仅在标准化之后验证,因此排除大写字母
for (char c = 'a'; c <= 'z'; c++) {
valid.set(c);
}
return valid;
}
validateName()方法用于验证输入的字符串(会先转为小写)是否有效:
private static String validateName(String n) {
checkNotNull(n);
checkArgument(n.length() != 0, "token must have at least 1 tchar");
for (int i = 0; i < n.length(); i++) {
char tChar = n.charAt(i);
// TODO(notcarl): remove this hack once pseudo headers are properly handled
if (tChar == ':' && i == 0) {
continue;
}
checkArgument(VALID_T_CHARS.get(tChar),
"Invalid character '%s' in key name '%s'", tChar, n);
}
return n;
}
Key的类定义
public abstract static class Key<T> {}
这是一个abstract类,后面将看到它的几个子类实现。
Key的属性和构造函数
private final String originalName;
private final String name;
private final byte[] nameBytes;
private Key(String name) {
// 原始名字是构造函数中传递过来的原始输入
this.originalName = checkNotNull(name);
// Intern the result for faster string identity checking.
// 有效名字是经过处理并验证有效的名字:这里的处理就是简单的改为小写
this.name = validateName(this.originalName.toLowerCase(Locale.ROOT)).intern();
// nameBytes是有效名字的字节数组
this.nameBytes = this.name.getBytes(US_ASCII);
}
Key的抽象方法
定义了两个方法用来做元数据和字节数组之间的转换和解析:
// 将元数据序列化到byte数组
abstract byte[] toBytes(T value);
// 从byte数组解析被序列化的元数据值
abstract T parseBytes(byte[] serialized);
Key的子类实现
BinaryKey
private static class BinaryKey<T> extends Key<T> { private final BinaryMarshaller<T> marshaller; // 构造函数传入一个BinaryMarshaller private BinaryKey(String name, BinaryMarshaller<T> marshaller) { ...... this.marshaller = checkNotNull(marshaller, "marshaller is null"); } @Override byte[] toBytes(T value) { // 使用这个BinaryMarshaller来转换 Key return marshaller.toBytes(value); } @Override T parseBytes(byte[] serialized) { // 使用这个BinaryMarshaller来解析 Key return marshaller.parseBytes(serialized); } }
AsciiKey: 类似,这次使用的是 AsciiMarshaller