java/src/main/com/google/appengine/tools/appstats/MailRpcCostCalculator.java

   1 // Copyright 2012 Google Inc. All Rights Reserved.
   2 package com.google.appengine.tools.appstats;
   3
   4 import com.google.appengine.api.mail.MailServicePb;
   5
   6 import static com.google.appengine.tools.appstats.StatsProtos.BilledOpProto.BilledOp.MAIL_RECIPIENT;
   7
   8 import java.util.Arrays;
   9 import java.util.Map;
  10
  11 /**
  12  * {@link RpcCostCalculator} implementation for the Mail API.
  13  *
  14  */
  15 class MailRpcCostCalculator implements RpcCostCalculator {
  16
  17   private static final String PKG = "mail";
  18
  19   private final long recipientCostMicropennies;
  20
  21   MailRpcCostCalculator(long recipientCostMicropennies) {
  22     this.recipientCostMicropennies = recipientCostMicropennies;
  23   }
  24
  25   @Override
  26   public RpcCost determineCost(String methodName, byte[] request, byte[] response) {
  27     if (methodName.equals("Send") || methodName.equals("SendToAdmin")) {
  28       MailServicePb.MailMessage proto = new MailServicePb.MailMessage();
  29       proto.parseFrom(request);
  30       int numRecipients = proto.toSize() + proto.ccSize() + proto.bccSize();
  31       StatsProtos.BilledOpProto billedOpProto = StatsProtos.BilledOpProto.newBuilder()
  32           .setNumOps(numRecipients).setOp(MAIL_RECIPIENT).build();
  33       return new RpcCost(recipientCostMicropennies * numRecipients, Arrays.asList(billedOpProto));
  34     }
  35     return FREE;
  36   }
  37
  38   static void register(
  39       Map<String, RpcCostCalculator> costCalculatorMap, RpcOperationCostManager opCostMgr) {
  40     costCalculatorMap.put(PKG, new MailRpcCostCalculator(opCostMgr.costOf(MAIL_RECIPIENT)));
  41   }
  42 }