[ https://issues.apache.org/jira/browse/MINIFICPP-681?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16694983#comment-16694983 ]
ASF GitHub Bot commented on MINIFICPP-681: ------------------------------------------ Github user phrocker commented on a diff in the pull request: https://github.com/apache/nifi-minifi-cpp/pull/445#discussion_r235471397 --- Diff: libminifi/src/processors/ContentHash.cpp --- @@ -0,0 +1,100 @@ +/** + * @file ContentHash.cpp + * ContentHash class implementation + * + * 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. + */ + +#ifdef OPENSSL_SUPPORT + +#include <algorithm> +#include <iostream> +#include <memory> +#include <string> +#include "processors/ContentHash.h" +#include "core/ProcessContext.h" +#include "core/ProcessSession.h" +#include "core/FlowFile.h" + +namespace org { +namespace apache { +namespace nifi { +namespace minifi { +namespace processors { + +core::Property ContentHash::HashAttribute("Hash Attribute", "Attribute to store checksum to", "Checksum"); +core::Property ContentHash::HashAlgorithm("Hash Algorithm", "Name of the algorithm used to generate checksum", "MD5"); +core::Relationship ContentHash::Success("success", "success operational on the flow record"); + +void ContentHash::initialize() { + //! Set the supported properties + std::set<core::Property> properties; + properties.insert(HashAttribute); + properties.insert(HashAlgorithm); + setSupportedProperties(properties); + //! Set the supported relationships + std::set<core::Relationship> relationships; + relationships.insert(Success); + setSupportedRelationships(relationships); +} + +void ContentHash::onTrigger(core::ProcessContext *context, core::ProcessSession *session) { + std::shared_ptr<core::FlowFile> flowFile = session->get(); + + if (!flowFile) { + return; + } + + ReadCallback cb(flowFile, context); + session->read(flowFile, &cb); + session->transfer(flowFile, Success); +} + +int64_t ContentHash::ReadCallback::process(std::shared_ptr<io::BaseStream> stream) { + std::string attrKey, algoName; + ctx_->getProperty(HashAttribute.getName(), attrKey); + ctx_->getProperty(HashAlgorithm.getName(), algoName); + std::transform(algoName.begin(), algoName.end(), algoName.begin(), ::toupper); + + // Erase '-' to make sha-256 and sha-2 work, too + algoName.erase(std::remove(algoName.begin(), algoName.end(), '-'), algoName.end()); + + // This throws in case algo is not found, but that's fine --- End diff -- Might the code above be less duplicative with a simple if statement here? > Add content hash processor > -------------------------- > > Key: MINIFICPP-681 > URL: https://issues.apache.org/jira/browse/MINIFICPP-681 > Project: NiFi MiNiFi C++ > Issue Type: Improvement > Reporter: Arpad Boda > Assignee: Arpad Boda > Priority: Major > Fix For: 0.6.0 > > > Add a new processor that supports hashing content and add the checksum to the > flowfile as an attribute. -- This message was sent by Atlassian JIRA (v7.6.3#76005)