[
https://issues.apache.org/jira/browse/MINIFICPP-681?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16695857#comment-16695857
]
ASF GitHub Bot commented on MINIFICPP-681:
------------------------------------------
Github user arpadboda commented on a diff in the pull request:
https://github.com/apache/nifi-minifi-cpp/pull/445#discussion_r235710768
--- 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
+ auto algo = HashAlgos.at(algoName);
+
+ const auto& ret_val = algo(stream);
+
+ if (ret_val.second <= 0) {
--- End diff --
Rerturn code was originally introduced to meet the requirements of
readCallback: it's int64_t return value expexts the number of bytes read to be
returned.
The condition here can be removed, we can stamp the empty hash as well.
> 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)
