com.generallycloud.baseio.codec.protobuf.ProtobufUtil Maven / Gradle / Ivy
/*
* Copyright 2015-2017 GenerallyCloud.com
*
* 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.generallycloud.baseio.codec.protobuf;
import java.util.HashMap;
import java.util.Map;
import com.generallycloud.baseio.codec.protobase.future.ProtobaseFuture;
import com.google.protobuf.InvalidProtocolBufferException;
import com.google.protobuf.MessageLite;
import com.google.protobuf.Parser;
public class ProtobufUtil {
private Map> parses = new HashMap>();
public void regist(String name, Parser parser) {
parses.put(name, parser);
}
public void regist(MessageLite messageLite) {
parses.put(messageLite.getClass().getName(), messageLite.getParserForType());
}
public Parser getParser(String name)
throws InvalidProtocolBufferException {
Parser parser = parses.get(name);
if (parser == null) {
throw new InvalidProtocolBufferException("did not found parse by name " + name);
}
return parser;
}
public MessageLite getMessage(ProtobaseFuture future)
throws InvalidProtocolBufferException {
Parser parser = getParser(future.getFutureName());
return parser.parseFrom(future.getBinary());
}
public void writeProtobuf(MessageLite messageLite, ProtobaseFuture future)
throws InvalidProtocolBufferException {
writeProtobuf(messageLite.getClass().getName(), messageLite, future);
}
public void writeProtobuf(String parserName, MessageLite messageLite,
ProtobaseFuture future) throws InvalidProtocolBufferException {
future.setFutureName(parserName);
// FIXME 判断array是否过大
byte[] array = messageLite.toByteArray();
future.writeBinary(array, 0, array.length);
}
}