mirror of https://github.com/apache/nifi.git
NIFI-927: Use a serializable version of NiFiDataPacket in the spark receiver
This commit is contained in:
parent
e4e263c292
commit
72ccc252fe
|
@ -170,18 +170,7 @@ public class NiFiReceiver extends Receiver<NiFiDataPacket> {
|
||||||
StreamUtils.fillBuffer(inStream, data);
|
StreamUtils.fillBuffer(inStream, data);
|
||||||
|
|
||||||
final Map<String, String> attributes = dataPacket.getAttributes();
|
final Map<String, String> attributes = dataPacket.getAttributes();
|
||||||
final NiFiDataPacket NiFiDataPacket = new NiFiDataPacket() {
|
final NiFiDataPacket NiFiDataPacket = new StandardNiFiDataPacket(data, attributes);
|
||||||
@Override
|
|
||||||
public byte[] getContent() {
|
|
||||||
return data;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Map<String, String> getAttributes() {
|
|
||||||
return attributes;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
dataPackets.add(NiFiDataPacket);
|
dataPackets.add(NiFiDataPacket);
|
||||||
dataPacket = transaction.receive();
|
dataPacket = transaction.receive();
|
||||||
} while (dataPacket != null);
|
} while (dataPacket != null);
|
||||||
|
|
|
@ -0,0 +1,43 @@
|
||||||
|
/*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
package org.apache.nifi.spark;
|
||||||
|
|
||||||
|
import java.io.Serializable;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
public class StandardNiFiDataPacket implements NiFiDataPacket, Serializable {
|
||||||
|
private static final long serialVersionUID = 6364005260220243322L;
|
||||||
|
|
||||||
|
private final byte[] content;
|
||||||
|
private final Map<String, String> attributes;
|
||||||
|
|
||||||
|
public StandardNiFiDataPacket(final byte[] content, final Map<String, String> attributes) {
|
||||||
|
this.content = content;
|
||||||
|
this.attributes = attributes;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public byte[] getContent() {
|
||||||
|
return content;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Map<String, String> getAttributes() {
|
||||||
|
return attributes;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue