/**
* This software is licensed to you under the Apache License, Version 2.0 (the
* "Apache License").
*
* LinkedIn's contributions are made under the Apache License. If you contribute
* to the Software, the contributions will be deemed to have been made under the
* Apache License, unless you expressly indicate otherwise. Please do not make any
* contributions that would be inconsistent with the Apache License.
*
* You may obtain a copy of the Apache License at http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, this software
* distributed under the Apache License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the Apache
* License for the specific language governing permissions and limitations for the
* software governed under the Apache License.
*
* © 2012 LinkedIn Corp. All Rights Reserved.
*/
package com.sensei.search.req.protobuf;
import com.google.protobuf.ByteString;
import com.google.protobuf.InvalidProtocolBufferException;
import com.google.protobuf.TextFormat;
import com.linkedin.norbert.network.Serializer;
import com.senseidb.search.req.SenseiRequest;
import com.senseidb.search.req.SenseiResult;
public class SenseiReqProtoSerializer implements Serializer<SenseiRequest, SenseiResult> {
public String requestName() {
return SenseiRequestBPO.Request.getDescriptor().getName();
}
public String responseName() {
return SenseiResultBPO.Result.getDescriptor().getName();
}
public byte[] requestToBytes(SenseiRequest request) {
return SenseiRequestBPO.Request.newBuilder().setVal(serialize(request)).build().toByteArray();
}
public byte[] responseToBytes(SenseiResult response) {
return SenseiResultBPO.Result.newBuilder().setVal(serialize(response)).build().toByteArray();
}
private <T> ByteString serialize(T obj) {
try {
return ProtoConvertUtil.serializeOut(obj);
} catch (TextFormat.ParseException e) {
throw new RuntimeException(e);
}
}
public SenseiRequest requestFromBytes(byte[] request) {
try {
return (SenseiRequest) ProtoConvertUtil.serializeIn(SenseiRequestBPO.Request.newBuilder().mergeFrom(request).build().getVal());
} catch (TextFormat.ParseException e) {
throw new RuntimeException(e);
} catch (InvalidProtocolBufferException e) {
throw new RuntimeException(e);
}
}
public SenseiResult responseFromBytes(byte[] result) {
try {
return (SenseiResult) ProtoConvertUtil.serializeIn(SenseiResultBPO.Result.newBuilder().mergeFrom(result).build().getVal());
} catch (TextFormat.ParseException e) {
throw new RuntimeException(e);
} catch (InvalidProtocolBufferException e) {
throw new RuntimeException(e);
}
}
}