# 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
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# 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.

#
# This script instruments hadoop rpc layer to print out all the request/response messages to the standard output.
#

RULE mock request vote
CLASS org.apache.ratis.grpc.server.GrpcServerProtocolClient
METHOD requestVote
AT ENTRY
BIND service:org.apache.ratis.grpc.server.GrpcServerProtocolClient = $0;
     raftPeerId:org.apache.ratis.protocol.RaftPeerId = service.raftPeerId;
     result:org.apache.ratis.proto.RaftProtos$RequestVoteReplyProto = org.apache.hadoop.hdds.freon.FakeRatisFollower.requestVote(raftPeerId, $1);
IF true
DO return result;
ENDRULE


RULE mock append log
CLASS org.apache.ratis.grpc.server.GrpcServerProtocolClient
METHOD appendEntries
AT ENTRY
BIND service:org.apache.ratis.grpc.server.GrpcServerProtocolClient = $0;
     raftPeerId:org.apache.ratis.protocol.RaftPeerId = service.raftPeerId;
     result:org.apache.ratis.thirdparty.io.grpc.stub.StreamObserver = org.apache.hadoop.hdds.freon.FakeRatisFollower.appendEntries(raftPeerId, $1);
IF true
DO return result;
ENDRULE