Used oneof in blimp_message.proto

blimp/net/blimp_message_checkpointer.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "blimp/net/blimp_message_checkpointer.h"
 
 #include "base/logging.h"
 #include "blimp/common/create_blimp_message.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/common/proto/protocol_control.pb.h"
@@ skipping 17 matching lines @@
   DCHECK(incoming_processor_);
   DCHECK(outgoing_processor_);
   DCHECK(checkpoint_observer_);
 }
 
 BlimpMessageCheckpointer::~BlimpMessageCheckpointer() {}
 
 void BlimpMessageCheckpointer::ProcessMessage(
     std::unique_ptr<BlimpMessage> message,
     const net::CompletionCallback& callback) {
-  if (message->type() == BlimpMessage::PROTOCOL_CONTROL &&
+  if (message->has_protocol_control() &&

[Kevin M, 2016/05/02 17:47:48]

Check the case

[shaktisahu, 2016/05/16 20:19:02]

I am checking the case it only in DCHECK

       message->protocol_control().type() ==
           ProtocolControlMessage::CHECKPOINT_ACK) {
     if (message->protocol_control().has_checkpoint_ack() &&
         message->protocol_control().checkpoint_ack().has_checkpoint_id()) {
       checkpoint_observer_->OnMessageCheckpoint(
           message->protocol_control().checkpoint_ack().checkpoint_id());
       callback.Run(net::OK);
     } else {
       DLOG(WARNING) << "Invalid checkpoint ACK. Dropping connection.";
       callback.Run(net::ERR_FAILED);
@@ skipping 27 matching lines @@
     int result) {
   callback.Run(result);
 }
 
 void BlimpMessageCheckpointer::SendCheckpointAck() {
   outgoing_processor_->ProcessMessage(
       CreateCheckpointAckMessage(checkpoint_id_), net::CompletionCallback());
 }
 
 }  // namespace blimp