expensive_operation_rpc_handler.h

Go to the documentation of this file.

// Copyright 2016 Google Inc.

#ifndef PAGESPEED_CONTROLLER_EXPENSIVE_OPERATION_RPC_HANDLER_H_
#define PAGESPEED_CONTROLLER_EXPENSIVE_OPERATION_RPC_HANDLER_H_

#include "pagespeed/controller/controller.pb.h"
#include "pagespeed/controller/controller.grpc.pb.h"
#include "pagespeed/controller/rpc_handler.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/base/function.h"
#include "pagespeed/kernel/util/grpc.h"
#include "pagespeed/controller/expensive_operation_controller.h"
#include "pagespeed/controller/request_result_rpc_handler.h"

namespace net_instaweb {


class ExpensiveOperationRpcHandler
    : public RequestResultRpcHandler<
          ExpensiveOperationRpcHandler, ExpensiveOperationController,
          grpc::CentralControllerRpcService::AsyncService,
          ScheduleExpensiveOperationRequest,
          ScheduleExpensiveOperationResponse> {
 protected:
  ExpensiveOperationRpcHandler(
      grpc::CentralControllerRpcService::AsyncService* service,
      ::grpc::ServerCompletionQueue* cq,
      ExpensiveOperationController* controller);

  void HandleClientRequest(
      const ScheduleExpensiveOperationRequest& req, Function* cb) override;
  void HandleClientResult(
      const ScheduleExpensiveOperationRequest& req) override;
  void HandleOperationFailed() override;

  void InitResponder(grpc::CentralControllerRpcService::AsyncService* service,
                     ::grpc::ServerContext* ctx, ReaderWriterT* responder,
                     ::grpc::ServerCompletionQueue* cq,
                     void* callback) override;

 private:
  friend class RequestResultRpcHandler;
  friend class ExpensiveOperationRpcHandlerTest;

  
};

}  

#endif