diff --git a/data_store_service_client.cpp b/data_store_service_client.cpp index 21170b8..afc6e47 100644 --- a/data_store_service_client.cpp +++ b/data_store_service_client.cpp @@ -403,7 +403,7 @@ bool DataStoreServiceClient::PersistKV( << callback_data->Result().error_msg(); return false; } - DLOG(INFO) << "DataStoreHandler::PersistKV success."; + LOG(INFO) << "DataStoreHandler::PersistKV success."; return true; } @@ -3213,7 +3213,7 @@ void DataStoreServiceClient::ReadInternal(ReadClosure *read_closure) EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *read_closure->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = read_closure->ReadRequest(); auto *resp = read_closure->ReadResponse(); stub.Read(&cntl, req, resp, read_closure); @@ -3266,7 +3266,7 @@ void DataStoreServiceClient::DeleteRangeInternal( EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *delete_range_clouse->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = delete_range_clouse->DeleteRangeRequest(); auto *resp = delete_range_clouse->DeleteRangeResponse(); stub.DeleteRange(&cntl, req, resp, delete_range_clouse); @@ -3315,7 +3315,7 @@ void DataStoreServiceClient::FlushDataInternal( EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *flush_data_closure->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = flush_data_closure->FlushDataRequest(); auto *resp = flush_data_closure->FlushDataResponse(); stub.FlushData(&cntl, req, resp, flush_data_closure); @@ -3366,7 +3366,7 @@ void DataStoreServiceClient::DropTableInternal( EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *drop_table_closure->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = drop_table_closure->DropTableRequest(); auto *resp = drop_table_closure->DropTableResponse(); stub.DropTable(&cntl, req, resp, drop_table_closure); @@ -3435,7 +3435,7 @@ void DataStoreServiceClient::ScanNextInternal( EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *scan_next_closure->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = scan_next_closure->ScanNextRequest(); auto *resp = scan_next_closure->ScanNextResponse(); stub.ScanNext(&cntl, req, resp, scan_next_closure); @@ -3487,7 +3487,7 @@ void DataStoreServiceClient::ScanCloseInternal( EloqDS::remote::DataStoreRpcService_Stub stub(channel); brpc::Controller &cntl = *scan_next_closure->Controller(); - cntl.set_timeout_ms(5000); + cntl.set_timeout_ms(60000); auto *req = scan_next_closure->ScanNextRequest(); auto *resp = scan_next_closure->ScanNextResponse(); stub.ScanClose(&cntl, req, resp, scan_next_closure); @@ -3806,6 +3806,8 @@ void DataStoreServiceClient::BatchWriteRecordsInternal( // send request remote::DataStoreRpcService_Stub stub(channel); + brpc::Controller &cntl = *closure->Controller(); + cntl.set_timeout_ms(60000); stub.BatchWriteRecords(closure->Controller(), closure->RemoteRequest(), closure->RemoteResponse(), diff --git a/data_store_service_client_closure.h b/data_store_service_client_closure.h index d7ee3b2..a38037c 100644 --- a/data_store_service_client_closure.h +++ b/data_store_service_client_closure.h @@ -978,7 +978,7 @@ class FlushDataClosure : public ::google::protobuf::Closure, public Poolable if (cntl_.Failed()) { // RPC failed. - LOG(ERROR) << "Failed for DeleteRange RPC request " + LOG(ERROR) << "Failed for FlushData RPC request " << ", with Error code: " << cntl_.ErrorCode() << ". Error Msg: " << cntl_.ErrorText(); if (cntl_.ErrorCode() != brpc::EOVERCROWDED && @@ -1725,7 +1725,7 @@ class BatchWriteRecordsClosure : public ::google::protobuf::Closure, } } - if (need_retry && retry_count_ < 2) + if (need_retry && retry_count_ < ds_service_client_->retry_limit_) { self_guard.Release(); retry_count_++;