/*
* 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.controller.swap;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.atomic.AtomicLong;
import org.apache.nifi.controller.repository.FlowFileRecord;
import org.apache.nifi.controller.repository.claim.ContentClaim;
import org.apache.nifi.controller.repository.claim.ResourceClaim;
import org.apache.nifi.controller.repository.claim.ResourceClaimManager;
import org.apache.nifi.controller.repository.claim.StandardContentClaim;
import org.apache.nifi.flowfile.FlowFile;
public class MockFlowFile implements FlowFileRecord {
private static final AtomicLong idGenerator = new AtomicLong(0L);
private final long id;
private final long entryDate = System.currentTimeMillis();
private final long lastQueueDate = System.currentTimeMillis();
private final Map<String, String> attributes;
private final long size;
private final ContentClaim contentClaim;
public MockFlowFile(final Map<String, String> attributes, final long size, final ResourceClaimManager claimManager) {
this(attributes, size, createContentClaim(String.valueOf(idGenerator.get()), claimManager));
}
public MockFlowFile(final Map<String, String> attributes, final long size, final ContentClaim contentClaim) {
this(idGenerator.getAndIncrement(), attributes, size, contentClaim);
}
public MockFlowFile(final long id, final Map<String, String> attributes, final long size, final ContentClaim contentClaim) {
this.id = id;
this.attributes = new HashMap<>(attributes);
this.size = size;
this.contentClaim = contentClaim;
}
public static ContentClaim createContentClaim(final String id, final ResourceClaimManager claimManager) {
final ResourceClaim resourceClaim = claimManager.newResourceClaim("container", "section", id, false, false);
claimManager.incrementClaimantCount(resourceClaim);
return new StandardContentClaim(resourceClaim, 3L);
}
@Override
public long getId() {
return id;
}
@Override
public long getEntryDate() {
return entryDate;
}
@Override
public long getLineageStartDate() {
return entryDate;
}
@Override
public Long getLastQueueDate() {
return lastQueueDate;
}
@Override
public boolean isPenalized() {
return false;
}
@Override
public String getAttribute(String key) {
return attributes.get(key);
}
@Override
public long getSize() {
return size;
}
@Override
public Map<String, String> getAttributes() {
return Collections.unmodifiableMap(attributes);
}
@Override
public int compareTo(final FlowFile o) {
return Long.compare(id, o.getId());
}
@Override
public long getPenaltyExpirationMillis() {
return -1L;
}
@Override
public ContentClaim getContentClaim() {
return contentClaim;
}
@Override
public long getContentClaimOffset() {
return 1;
}
@Override
public long getLineageStartIndex() {
return 0;
}
@Override
public long getQueueDateIndex() {
return 0;
}
public static void resetIdGenerator() {
idGenerator.set(0L);
}
}