tools/fault-injection-service/Service/cpp/failure_injector_svc_server.cc

/**                                                                             
 * Licensed to the Apache Software Foundation (ASF) under one or more           
 * contributor license agreements.  See the NOTICE file distributed with this   
 * work for additional information regarding copyright ownership.  The ASF      
 * licenses this file to you under the Apache License, Version 2.0 (the         
 * "License"); you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at                                      
 * <p>                                                                          
 * http://www.apache.org/licenses/LICENSE-2.0                                   
 * <p>                                                                          
 * Unless required by applicable law or agreed to in writing, software          
 * distributed under the License is distributed on an "AS IS" BASIS,WITHOUT     
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the     
 * License for the specific language governing permissions and limitations under
 * the License.                                                                 
 */                  

#include <iostream>
#include <memory>
#include <string>
#include <iostream>

#include <grpcpp/grpcpp.h>

#include "failure_injection_service.grpc.pb.h"
#include "failure_injector.h"
#include "failure_injector_svc_server.h"
#include "run_grpc_service.h"

using grpc::Server;
using grpc::ServerBuilder;
using grpc::ServerContext;
using grpc::Status;

using namespace NoiseInjector;
using namespace std;

Status FailureInjectorSvcServiceImpl::GetStatus(
    ServerContext* context,
    const GetStatusRequest* request,
    GetStatusReply* reply)
{

    string prefix("GetStatus ");
    cout << "Received request " << request->name() << "\n";
    reply->set_message(prefix + request->name());
    return Status::OK;
}

Status FailureInjectorSvcServiceImpl::InjectFailure(
    ServerContext* context,
    const InjectFailureRequest* request,
    InjectFailureReply* reply)
{
  cout << "Received request " << request->path() << ":" 
            << request->op_name() << "\n";
  mFailureInjector->InjectFailure(request->path(), request->op_name(),
                                InjectedAction((int)request->action_code(),
                                               (int)request->action_param(),
                                               (int)request->action_param()));
  return Status::OK;
}

Status FailureInjectorSvcServiceImpl::ResetFailure(
    ServerContext* context,
    const ResetFailureRequest* request,
    ResetFailureReply* reply)
{
    cout << "Received request ";
    if (request->has_path()) { 
            cout << request->path() << "\n";
        if (request->has_op_name()) {
            cout << request->op_name() << "\n";
        }
    }

    if (request->has_path() && !request->path().empty()) {
        if (request->has_op_name() && !request->op_name().empty()) {
            mFailureInjector->ResetFailure(request->path(), request->op_name());
        } else {
            mFailureInjector->ResetFailure(request->path());
        }
    } else {
        mFailureInjector->ResetFailure();
    }
    return Status::OK;
}

FailureInjectorSvcServiceImpl::FailureInjectorSvcServiceImpl(
    FailureInjector* injector)
{
    mFailureInjector = injector;
}