From issues-return-99364-archive-asf-public=cust-asf.ponee.io@nifi.apache.org Fri Jun 26 13:50:56 2020 Return-Path: X-Original-To: archive-asf-public@cust-asf.ponee.io Delivered-To: archive-asf-public@cust-asf.ponee.io Received: from mail.apache.org (hermes.apache.org [207.244.88.153]) by mx-eu-01.ponee.io (Postfix) with SMTP id EED38180656 for ; Fri, 26 Jun 2020 15:50:55 +0200 (CEST) Received: (qmail 83712 invoked by uid 500); 26 Jun 2020 13:50:55 -0000 Mailing-List: contact issues-help@nifi.apache.org; run by ezmlm Precedence: bulk List-Help: List-Unsubscribe: List-Post: List-Id: Reply-To: dev@nifi.apache.org Delivered-To: mailing list issues@nifi.apache.org Received: (qmail 83703 invoked by uid 99); 26 Jun 2020 13:50:55 -0000 Received: from ec2-52-202-80-70.compute-1.amazonaws.com (HELO gitbox.apache.org) (52.202.80.70) by apache.org (qpsmtpd/0.29) with ESMTP; Fri, 26 Jun 2020 13:50:55 +0000 From: =?utf-8?q?GitBox?= To: issues@nifi.apache.org Subject: =?utf-8?q?=5BGitHub=5D_=5Bnifi-minifi-cpp=5D_hunyadi-dev_commented_on_a_chan?= =?utf-8?q?ge_in_pull_request_=23821=3A_MINIFICPP-1251_-_Implement_and_test_?= =?utf-8?q?RetryFlowFile_processor?= Message-ID: <159317945521.8807.17598322665293108323.asfpy@gitbox.apache.org> Date: Fri, 26 Jun 2020 13:50:55 -0000 Content-Type: text/plain; charset=utf-8 Content-Transfer-Encoding: 8bit In-Reply-To: References: hunyadi-dev commented on a change in pull request #821: URL: https://github.com/apache/nifi-minifi-cpp/pull/821#discussion_r446196483 ########## File path: extensions/standard-processors/processors/RetryFlowFile.cpp ########## @@ -0,0 +1,212 @@ +/** + * + * 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. + */ + +#include "RetryFlowFile.h" + +#include "core/PropertyValidation.h" + +namespace org { +namespace apache { +namespace nifi { +namespace minifi { +namespace processors { + +core::Property RetryFlowFile::RetryAttribute(core::PropertyBuilder::createProperty("Retry Attribute") + ->withDescription( + "The name of the attribute that contains the current retry count for the FlowFile." + "WARNING: If the name matches an attribute already on the FlowFile that does not contain a numerical value, " + "the processor will either overwrite that attribute with '1' or fail based on configuration.") + ->withDefaultValue("flowfile.retries") + ->supportsExpressionLanguage(true) + ->build()); + +core::Property RetryFlowFile::MaximumRetries(core::PropertyBuilder::createProperty("Maximum Retries") + ->withDescription("The maximum number of times a FlowFile can be retried before being passed to the 'retries_exceeded' relationship.") + ->withDefaultValue(3) + ->supportsExpressionLanguage(true) + ->build()); + +core::Property RetryFlowFile::PenalizeRetries(core::PropertyBuilder::createProperty("Penalize Retries") + ->withDescription("If set to 'true', this Processor will penalize input FlowFiles before passing them to the 'retry' relationship. This does not apply to the 'retries_exceeded' relationship.") + ->withDefaultValue(true) + ->build()); + +core::Property RetryFlowFile::FailOnNonNumericalOverwrite(core::PropertyBuilder::createProperty("Fail on Non-numerical Overwrite") + ->withDescription("If the FlowFile already has the attribute defined in 'Retry Attribute' that is *not* a number, fail the FlowFile instead of resetting that value to '1'") + ->withDefaultValue(false) + ->build()); + +core::Property RetryFlowFile::ReuseMode(core::PropertyBuilder::createProperty("Reuse Mode") + ->withDescription( + "Defines how the Processor behaves if the retry FlowFile has a different retry UUID than " + "the instance that received the FlowFile. This generally means that the attribute was " + "not reset after being successfully retried by a previous instance of this processor.") + ->withAllowableValues({FAIL_ON_REUSE, WARN_ON_REUSE, RESET_REUSE}) + ->withDefaultValue(FAIL_ON_REUSE) + ->build()); + +core::Relationship RetryFlowFile::Retry("retry", + "Input FlowFile has not exceeded the configured maximum retry count, pass this relationship back to the input Processor to create a limited feedback loop."); +core::Relationship RetryFlowFile::RetriesExceeded("retries_exceeded", + "Input FlowFile has exceeded the configured maximum retry count, do not pass this relationship back to the input Processor to terminate the limited feedback loop."); +core::Relationship RetryFlowFile::Failure("failure", + "The processor is configured such that a non-numerical value on 'Retry Attribute' results in a failure instead of resetting " + "that value to '1'. This will immediately terminate the limited feedback loop. Might also include when 'Maximum Retries' contains " + " attribute expression language that does not resolve to an Integer."); + +void RetryFlowFile::initialize() { + setSupportedProperties({ + RetryAttribute, + MaximumRetries, + PenalizeRetries, + FailOnNonNumericalOverwrite, + ReuseMode, + }); + setSupportedRelationships({ + Retry, + RetriesExceeded, + Failure, + }); +} + +void RetryFlowFile::onSchedule(core::ProcessContext* context, core::ProcessSessionFactory* /* sessionFactory */) { + context->getProperty(RetryAttribute.getName(), retry_attribute_); + context->getProperty(MaximumRetries.getName(), maximum_retries_); + context->getProperty(PenalizeRetries.getName(), penalize_retries_); + context->getProperty(FailOnNonNumericalOverwrite.getName(), fail_on_non_numerical_overwrite_); + context->getProperty(ReuseMode.getName(), reuse_mode_); + readDynamicPropertyKeys(context); +} + +void RetryFlowFile::onTrigger(core::ProcessContext* context, core::ProcessSession* session) { + std::shared_ptr flow_file = std::static_pointer_cast (session->get()); + if (!flow_file) { + return; + } + + bool failure_due_to_non_numerical_retry; + uint64_t retry_property_value; + std::tie(retry_property_value, failure_due_to_non_numerical_retry) = getRetryPropertyValue(flow_file); + if (failure_due_to_non_numerical_retry) { + session->transfer(flow_file, Failure); + return; + } + if (updateUUIDMarkerAndCheckFailOnReuse(flow_file)) { + session->transfer(flow_file, Failure); + return; + } + + if (retry_property_value < maximum_retries_) { + try { + flow_file->setAttribute(retry_attribute_, std::to_string(gsl::narrow_cast(retry_property_value + 1))); + } + catch(const gsl::narrowing_error& e) { + logger_->log_error("Narrowing Exception: %s", e.what()); + session->transfer(flow_file, Failure); + return; + } + if (penalize_retries_) { + session->penalize(flow_file); + } + session->transfer(flow_file, Retry); + return; + } + if (!setRetriesExceededAttributesOnFlowFile(context, flow_file)) { + session->transfer(flow_file, Failure); + yield(); + return; + } + session->transfer(flow_file, RetriesExceeded); +} + +void RetryFlowFile::readDynamicPropertyKeys(core::ProcessContext* context) { + exceeded_flowfile_attribute_keys.clear(); + const std::vector dynamic_prop_keys = context->getDynamicPropertyKeys(); + logger_->log_info("RetryFlowFile registering %d keys", dynamic_prop_keys.size()); + for (const auto& key : dynamic_prop_keys) { + exceeded_flowfile_attribute_keys.emplace_back(core::PropertyBuilder::createProperty(key)->withDescription("auto generated")->supportsExpressionLanguage(true)->build()); + logger_->log_info("RetryFlowFile registered attribute '%s'", key); + } +} + +// Returns (1, true) on non-numerical or out-of-bounds retry value +std::pair RetryFlowFile::getRetryPropertyValue(const std::shared_ptr& flow_file) { + std::string value_as_string; + try { + if (flow_file->getAttribute(retry_attribute_, value_as_string)) { + return std::make_pair(std::stoul(value_as_string), false); + } + } + catch(const std::invalid_argument&) { + if (fail_on_non_numerical_overwrite_) { + logger_->log_info("Non-numerical retry property in RetryFlowFile. Sending flowfile to failure...", value_as_string); + return std::make_pair(1, true); + } + logger_->log_info("Non-numerical retry property in RetryFlowFile: overwriting %s with 1.", value_as_string); + } + catch(const std::out_of_range&) { + logger_->log_error("Narrowing Exception for %s, treating it as non-numerical value", value_as_string); + } + return std::make_pair(1, false); +} + +// Returns true on fail on reuse scenario +bool RetryFlowFile::updateUUIDMarkerAndCheckFailOnReuse(const std::shared_ptr& flow_file) { Review comment: Resolved by moving the function content to `onTrigger` as you recommended. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: users@infra.apache.org