change function name, bind database name at the begining

Author: Tianyi Chen

src/include/network/postgres_protocol_handler.h

@@ -78,7 +78,7 @@ class PostgresProtocolHandler : public ProtocolHandler {
   static bool ReadPacket(Buffer &rbuf, InputPacket &rpkt);
 
   /* Routine to deal with the first packet from the client */
-  bool ProcessInitialPacket(InputPacket* pkt, Client client, bool ssl_able, bool& ssl_sent, bool& finish_startup_packet);
+  bool ProcessInitialPackets(InputPacket* pkt, Client client, bool ssl_able, bool& ssl_sent, bool& finish_startup_packet);
 
   /* Routine to deal with SSL request message */
   void ProcessSSLRequestPacket(bool ssl_able, bool& ssl_handshake);

src/include/network/protocol_handler.h

@@ -36,7 +36,7 @@ class ProtocolHandler {
   //  bool ManageStartupPacket();
   virtual void SendInitialResponse();
 
-  virtual bool ProcessInitialPacket(InputPacket* pkt, Client client, bool ssl_able, bool& ssl_sent, bool& finish_startup_packet) = 0;
+  virtual bool ProcessInitialPackets(InputPacket* pkt, Client client, bool ssl_able, bool& ssl_sent, bool& finish_startup_packet) = 0;
 
   virtual ProcessResult Process(Buffer &rbuf, const size_t thread_id);
 

src/network/connection_handle.cpp

@@ -204,7 +204,7 @@ Transition ConnectionHandle::FillReadBuffer() {
   Transition result = Transition::NEED_DATA;
   ssize_t bytes_read = 0;
   bool done = false;
-  
+
   // reset buffer if all the contents have been read
   if (rbuf_->buf_ptr == rbuf_->buf_size) rbuf_->Reset();
 
@@ -454,7 +454,7 @@ ProcessResult ConnectionHandle::ProcessInitial() {
   }
   // We need to handle startup packet first
   //TODO(Tianyi): If other protocols are added, this need to be changed
-  bool result = protocol_handler_->ProcessInitialPacket(
+  bool result = protocol_handler_->ProcessInitialPackets(
     &initial_packet_, client_, ssl_able_, ssl_handshake_,
     finish_startup_packet_);
   // Clean up the initial_packet after finishing processing.

src/network/postgres_protocol_handler.cpp

@@ -1053,7 +1053,7 @@ bool PostgresProtocolHandler::ReadPacket(Buffer &rbuf, InputPacket &rpkt) {
  * process_startup_packet - Processes the startup packet
  *  (after the size field of the header).
  */
-bool PostgresProtocolHandler::ProcessInitialPacket(InputPacket *pkt, Client client, bool ssl_able, bool& ssl_handshake, bool& finish_startup_packet) {
+bool PostgresProtocolHandler::ProcessInitialPackets(InputPacket *pkt, Client client, bool ssl_able, bool& ssl_handshake, bool& finish_startup_packet) {
   std::string token, value;
   std::unique_ptr<OutputPacket> response(new OutputPacket);
 
@@ -1098,6 +1098,7 @@ bool PostgresProtocolHandler::ProcessStartupPacket(InputPacket* pkt, int32_t pro
       // loop end?
       if (pkt->ptr >= pkt->len) break;
       GetStringToken(pkt, client.dbname);
+      traffic_cop_->SetDefaultDatabaseName(client.dbname);
     } else if (token.compare(("user")) == 0) {
       // loop end?
       if (pkt->ptr >= pkt->len) break;