protoc-gen-grpc
Version:
Protocol Buffers Compiler (protoc) plugin for generating grpc interfaces in TypeScript..
78 lines (77 loc) • 2.83 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.EntryMap = void 0;
class EntryMap {
constructor() {
this.messageEntryMap = {};
this.enumEntryMap = {};
}
/**
* 获取 MessageEntry 结构
*
* @param {string} scopeName
* @return {MessageEntry | undefined}
*/
getMessageEntry(scopeName) {
return this.messageEntryMap[scopeName];
}
/**
* 获取 EnumEntry
*
* @param {string} scopeName
* @return {EnumEntry | undefined}
*/
getEnumEntry(scopeName) {
return this.enumEntryMap[scopeName];
}
/**
* 将 fileDescriptor 中的内容进行解析，并填充到 messageEntry 和 enumEntry 中
*
* @param {FileDescriptorProto} fileDescriptor
*/
parseFileDescriptor(fileDescriptor) {
const scope = fileDescriptor.getPackage();
fileDescriptor.getMessageTypeList().forEach(messageType => {
this.parseMessageNested(scope, fileDescriptor, messageType);
});
fileDescriptor.getEnumTypeList().forEach(enumType => {
this.enumEntryMap[scope + '.' + enumType.getName()] = {
pkg: fileDescriptor.getPackage(),
fileName: fileDescriptor.getName(),
enumOptions: enumType.getOptions(),
};
});
}
/**
* 解析 Message NestedType
*
* @param {string} scope
* @param {FileDescriptorProto} fileDescriptor
* @param {DescriptorProto} message
*/
parseMessageNested(scope, fileDescriptor, message) {
const messageEntry = {
pkg: fileDescriptor.getPackage(),
fileName: fileDescriptor.getName(),
messageOptions: message.getOptions(),
mapFieldOptions: message.getOptions() && message.getOptions().getMapEntry() ? {
key: [message.getFieldList()[0].getType(), message.getFieldList()[0].getTypeName().slice(1)],
value: [message.getFieldList()[1].getType(), message.getFieldList()[1].getTypeName().slice(1)],
} : undefined,
};
const entryName = `${scope ? scope + '.' : ''}${message.getName()}`;
this.messageEntryMap[entryName] = messageEntry;
message.getNestedTypeList().forEach(nested => {
this.parseMessageNested(scope + '.' + message.getName(), fileDescriptor, nested);
});
message.getEnumTypeList().forEach(enumType => {
const identifier = scope + '.' + message.getName() + '.' + enumType.getName();
this.enumEntryMap[identifier] = {
pkg: fileDescriptor.getPackage(),
fileName: fileDescriptor.getName(),
enumOptions: enumType.getOptions(),
};
});
}
}
exports.EntryMap = EntryMap;