java/src/main/com/google/appengine/api/conversion/DocumentProtoConverter.java

   1 // Copyright 2011 Google Inc. All Rights Reserved.
   2
   3 package com.google.appengine.api.conversion;
   4
   5 import com.google.appengine.api.conversion.ConversionServicePb.AssetInfo;
   6 import com.google.appengine.api.conversion.ConversionServicePb.DocumentInfo;
   7 import com.google.common.base.Converter;
   8 import com.google.common.collect.Lists;
   9
  10 import java.util.List;
  11
  12 /**
  13  * A reversible converter between {@link Document} and its corresponding
  14  * protocol buffer.
  15  *
  16  */
  17 class DocumentProtoConverter extends Converter<Document, DocumentInfo> {
  18
  19   private final AssetProtoConverter assetProtoConverter = new AssetProtoConverter();
  20
  21   /**
  22    * Converts a {@link Document} to its corresponding protocol buffer.
  23    *
  24    * @throws IllegalArgumentException if the size of document is over limit
  25    */
  26   @Override
  27   public DocumentInfo doForward(Document document) {
  28     ConversionHelper.validateDocumentSize(document);
  29
  30     DocumentInfo.Builder documentInfo = DocumentInfo.newBuilder();
  31     for (Asset asset : document.getAssets()) {
  32       documentInfo.addAsset(assetProtoConverter.convert(asset));
  33     }
  34     return documentInfo.build();
  35   }
  36
  37   @Override
  38   public Document doBackward(DocumentInfo documentPb) {
  39     List<Asset> assets = Lists.newArrayList();
  40     for (AssetInfo asset : documentPb.getAssetList()) {
  41       assets.add(assetProtoConverter.reverse(asset));
  42     }
  43     return new Document(assets);
  44   }
  45 }