| /* |
| * 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.iotdb.confignode.manager.pipe.event; |
| |
| import org.apache.iotdb.commons.pipe.event.EnrichedEvent; |
| import org.apache.iotdb.commons.pipe.event.PipeSnapshotEvent; |
| import org.apache.iotdb.commons.pipe.pattern.PipePattern; |
| import org.apache.iotdb.commons.pipe.task.meta.PipeTaskMeta; |
| import org.apache.iotdb.confignode.consensus.request.ConfigPhysicalPlanType; |
| import org.apache.iotdb.confignode.manager.pipe.resource.snapshot.PipeConfigNodeSnapshotResourceManager; |
| import org.apache.iotdb.confignode.persistence.schema.CNSnapshotFileType; |
| import org.apache.iotdb.tsfile.utils.ReadWriteIOUtils; |
| |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import java.io.File; |
| import java.nio.ByteBuffer; |
| import java.util.Arrays; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.EnumMap; |
| import java.util.HashSet; |
| import java.util.Map; |
| import java.util.Objects; |
| import java.util.Set; |
| import java.util.stream.Collectors; |
| |
| public class PipeConfigRegionSnapshotEvent extends PipeSnapshotEvent { |
| |
| private static final Logger LOGGER = LoggerFactory.getLogger(PipeConfigRegionSnapshotEvent.class); |
| private String snapshotPath; |
| // This will only be filled in when the snapshot is a schema info file. |
| private String templateFilePath; |
| private static final Map<CNSnapshotFileType, Set<Short>> |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP = new EnumMap<>(CNSnapshotFileType.class); |
| private CNSnapshotFileType fileType; |
| |
| static { |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.put( |
| CNSnapshotFileType.ROLE, |
| Collections.unmodifiableSet( |
| new HashSet<>( |
| Arrays.asList( |
| ConfigPhysicalPlanType.CreateRole.getPlanType(), |
| ConfigPhysicalPlanType.GrantRole.getPlanType())))); |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.put( |
| CNSnapshotFileType.USER, |
| Collections.unmodifiableSet( |
| new HashSet<>( |
| Arrays.asList( |
| ConfigPhysicalPlanType.CreateUser.getPlanType(), |
| ConfigPhysicalPlanType.GrantUser.getPlanType())))); |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.put( |
| CNSnapshotFileType.USER_ROLE, |
| Collections.singleton(ConfigPhysicalPlanType.GrantRoleToUser.getPlanType())); |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.put( |
| CNSnapshotFileType.SCHEMA, |
| Collections.unmodifiableSet( |
| new HashSet<>( |
| Arrays.asList( |
| ConfigPhysicalPlanType.CreateDatabase.getPlanType(), |
| ConfigPhysicalPlanType.SetTTL.getPlanType(), |
| ConfigPhysicalPlanType.CreateSchemaTemplate.getPlanType(), |
| ConfigPhysicalPlanType.CommitSetSchemaTemplate.getPlanType())))); |
| } |
| |
| public PipeConfigRegionSnapshotEvent() { |
| // Used for deserialization |
| this(null, null, null); |
| } |
| |
| public PipeConfigRegionSnapshotEvent( |
| String snapshotPath, String templateFilePath, CNSnapshotFileType type) { |
| this(snapshotPath, templateFilePath, type, null, null, null); |
| } |
| |
| public PipeConfigRegionSnapshotEvent( |
| String snapshotPath, |
| String templateFilePath, |
| CNSnapshotFileType type, |
| String pipeName, |
| PipeTaskMeta pipeTaskMeta, |
| PipePattern pattern) { |
| super(pipeName, pipeTaskMeta, pattern, PipeConfigNodeSnapshotResourceManager.getInstance()); |
| this.snapshotPath = snapshotPath; |
| this.templateFilePath = Objects.nonNull(templateFilePath) ? templateFilePath : ""; |
| this.fileType = type; |
| } |
| |
| public File getSnapshotFile() { |
| return new File(snapshotPath); |
| } |
| |
| public File getTemplateFile() { |
| return !templateFilePath.isEmpty() ? new File(templateFilePath) : null; |
| } |
| |
| public CNSnapshotFileType getFileType() { |
| return fileType; |
| } |
| |
| @Override |
| public boolean internallyIncreaseResourceReferenceCount(String holderMessage) { |
| try { |
| snapshotPath = resourceManager.increaseSnapshotReference(snapshotPath); |
| if (!templateFilePath.isEmpty()) { |
| templateFilePath = resourceManager.increaseSnapshotReference(templateFilePath); |
| } |
| return true; |
| } catch (Exception e) { |
| LOGGER.warn( |
| String.format( |
| "Increase reference count for snapshot %s error. Holder Message: %s", |
| snapshotPath, holderMessage), |
| e); |
| return false; |
| } |
| } |
| |
| @Override |
| public boolean internallyDecreaseResourceReferenceCount(String holderMessage) { |
| try { |
| resourceManager.decreaseSnapshotReference(snapshotPath); |
| if (!templateFilePath.isEmpty()) { |
| resourceManager.decreaseSnapshotReference(templateFilePath); |
| } |
| return true; |
| } catch (Exception e) { |
| LOGGER.warn( |
| String.format( |
| "Decrease reference count for snapshot %s error. Holder Message: %s", |
| snapshotPath, holderMessage), |
| e); |
| return false; |
| } |
| } |
| |
| @Override |
| public EnrichedEvent shallowCopySelfAndBindPipeTaskMetaForProgressReport( |
| String pipeName, |
| PipeTaskMeta pipeTaskMeta, |
| PipePattern pattern, |
| long startTime, |
| long endTime) { |
| return new PipeConfigRegionSnapshotEvent( |
| snapshotPath, templateFilePath, fileType, pipeName, pipeTaskMeta, pattern); |
| } |
| |
| @Override |
| public ByteBuffer serializeToByteBuffer() { |
| ByteBuffer result = |
| ByteBuffer.allocate( |
| 2 * Byte.BYTES |
| + 2 * Integer.BYTES |
| + snapshotPath.getBytes().length |
| + templateFilePath.getBytes().length); |
| ReadWriteIOUtils.write(PipeConfigSerializableEventType.CONFIG_SNAPSHOT.getType(), result); |
| ReadWriteIOUtils.write(fileType.getType(), result); |
| ReadWriteIOUtils.write(snapshotPath, result); |
| ReadWriteIOUtils.write(templateFilePath, result); |
| return result; |
| } |
| |
| @Override |
| public void deserializeFromByteBuffer(ByteBuffer buffer) { |
| fileType = CNSnapshotFileType.deserialize(ReadWriteIOUtils.readByte(buffer)); |
| snapshotPath = ReadWriteIOUtils.readString(buffer); |
| templateFilePath = ReadWriteIOUtils.readString(buffer); |
| } |
| |
| /////////////////////////////// Type parsing /////////////////////////////// |
| |
| public static boolean needTransferSnapshot(Set<ConfigPhysicalPlanType> listenedTypeSet) { |
| final Set<Short> types = |
| SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.values().stream() |
| .flatMap(Collection::stream) |
| .collect(Collectors.toSet()); |
| types.retainAll( |
| listenedTypeSet.stream() |
| .map(ConfigPhysicalPlanType::getPlanType) |
| .collect(Collectors.toSet())); |
| return !types.isEmpty(); |
| } |
| |
| public void confineTransferredTypes(Set<ConfigPhysicalPlanType> listenedTypeSet) { |
| final Set<Short> types = |
| new HashSet<>(SNAPSHOT_FILE_TYPE_2_CONFIG_PHYSICAL_PLAN_TYPE_MAP.get(fileType)); |
| types.retainAll( |
| listenedTypeSet.stream() |
| .map(ConfigPhysicalPlanType::getPlanType) |
| .collect(Collectors.toSet())); |
| transferredTypes = types; |
| } |
| |
| public static Set<ConfigPhysicalPlanType> getConfigPhysicalPlanTypeSet(String sealTypes) { |
| return sealTypes.isEmpty() |
| ? Collections.emptySet() |
| : Arrays.stream(sealTypes.split(",")) |
| .map( |
| typeValue -> |
| ConfigPhysicalPlanType.convertToConfigPhysicalPlanType( |
| Short.parseShort(typeValue))) |
| .collect(Collectors.toSet()); |
| } |
| |
| /////////////////////////////// Object /////////////////////////////// |
| |
| @Override |
| public String toString() { |
| return String.format( |
| "PipeConfigRegionSnapshotEvent{snapshotPath=%s, templateFilePath=%s, fileType=%s}", |
| snapshotPath, templateFilePath, fileType) |
| + " - " |
| + super.toString(); |
| } |
| } |