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

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