[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r305080155
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,328 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r305072769
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,328 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r305029180
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,328 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304690002
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,310 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304689926
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,310 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304689943
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,310 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.mutable.MutableInt;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.ISE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635888
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,274 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.RE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import javax.annotation.Nullable;
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List shuffleDataLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ // This can be null until IntermediaryDataManager is started.
+ @Nullable
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635858
##
File path:
indexing-service/src/test/java/org/apache/druid/indexing/worker/IntermediaryDataManagerManualAddAndDeleteTest.java
##
@@ -0,0 +1,147 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.amazonaws.util.StringUtils;
+import com.google.common.collect.ImmutableList;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.NoopIndexingServiceClient;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.Intervals;
+import org.apache.druid.segment.loading.StorageLocationConfig;
+import org.apache.druid.timeline.DataSegment;
+import org.apache.druid.timeline.partition.NumberedShardSpec;
+import org.joda.time.Interval;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Comparator;
+import java.util.List;
+
+public class IntermediaryDataManagerManualAddAndDeleteTest
+{
+ @Rule
+ public TemporaryFolder tempDir = new TemporaryFolder();
+
+ private IntermediaryDataManager intermediaryDataManager;
+
+ @Before
+ public void setup() throws IOException
+ {
+final WorkerConfig workerConfig = new WorkerConfig();
+final TaskConfig taskConfig = new TaskConfig(
+null,
+null,
+null,
+null,
+null,
+false,
+null,
+null,
+ImmutableList.of(new StorageLocationConfig(tempDir.newFolder(), null,
null))
+);
+final IndexingServiceClient indexingServiceClient = new
NoopIndexingServiceClient();
+intermediaryDataManager = new IntermediaryDataManager(workerConfig,
taskConfig, indexingServiceClient);
+intermediaryDataManager.start();
+ }
+
+ @After
+ public void teardown() throws InterruptedException
+ {
+intermediaryDataManager.stop();
+ }
+
+ @Test
+ public void testAddSegment() throws IOException
Review comment:
Hmm, removed.
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
-
To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org
For additional commands, e-mail: commits-h...@druid.apache.org
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635841
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/http/ShuffleResource.java
##
@@ -0,0 +1,116 @@
+/*
+ * 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.druid.indexing.worker.http;
+
+import com.google.common.io.ByteStreams;
+import com.google.inject.Inject;
+import com.sun.jersey.spi.container.ResourceFilters;
+import org.apache.druid.indexing.worker.IntermediaryDataManager;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.StringUtils;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.server.http.security.TaskShuffleResourceFilter;
+import org.joda.time.Interval;
+
+import javax.ws.rs.DELETE;
+import javax.ws.rs.GET;
+import javax.ws.rs.Path;
+import javax.ws.rs.PathParam;
+import javax.ws.rs.Produces;
+import javax.ws.rs.QueryParam;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.core.Response.Status;
+import javax.ws.rs.core.StreamingOutput;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.List;
+
+@Path("/druid/worker/v1/shuffle")
+@ResourceFilters(TaskShuffleResourceFilter.class)
+public class ShuffleResource
+{
+ private static final Logger log = new Logger(ShuffleResource.class);
+
+ private final IntermediaryDataManager intermediaryDataManager;
+
+ @Inject
+ public ShuffleResource(IntermediaryDataManager intermediaryDataManager)
+ {
+this.intermediaryDataManager = intermediaryDataManager;
+ }
+
+ @GET
+ @Path("/task/{supervisorTaskId}/partition")
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response getPartition(
+ @PathParam("supervisorTaskId") String supervisorTaskId,
+ @QueryParam("dataSource") String dataSource,
+ @QueryParam("startTime") String startTime,
+ @QueryParam("endTime") String endTime,
+ @QueryParam("partitionId") int partitionId
+ )
+ {
+final Interval interval = new Interval(DateTimes.of(startTime),
DateTimes.of(endTime));
+final List partitionFiles =
intermediaryDataManager.findPartitionFiles(
+supervisorTaskId,
+interval,
+partitionId
+);
+
+if (partitionFiles.isEmpty()) {
+ final String errorMessage = StringUtils.format(
+ "Can't find the partition for supervisor[%s], interval[%s], and
partitionId[%s]",
+ supervisorTaskId,
+ interval,
+ partitionId
+ );
+ return Response.status(Status.NOT_FOUND).entity(errorMessage).build();
+} else {
+ return Response.ok(
+ (StreamingOutput) output -> {
+for (File partitionFile : partitionFiles) {
+ try (final FileInputStream fileInputStream = new
FileInputStream(partitionFile)) {
+ByteStreams.copy(fileInputStream, output);
+ }
+}
+ }
+ ).build();
+}
+ }
+
+ @DELETE
+ @Path("/task/{supervisorTaskId}")
+ public Response deletePartitions(
+ @PathParam("supervisorTaskId") String supervisorTaskId,
+ @QueryParam("dataSource") String dataSource
Review comment:
Hmm, I just removed dataSource and changed to use `StateResourceFilter`
instead. Added the below javadoc.
> We use StateResourceFilter here because it performs an admin-like
authorization and all endpoints here are supposed to be used for only internal
communcation. Another possible alternate could be performing datasource-level
authorization as in TaskResourceFilter. However, datasource information is not
available in middleManagers or indexers yet which makes hard to use it. We
could develop a new ResourceFileter in the future if needed.
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635812
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635785
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635872 ## File path: server/src/main/java/org/apache/druid/server/http/security/TaskShuffleResourceFilter.java ## @@ -0,0 +1,72 @@ +/* + * 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.druid.server.http.security; + +import com.google.common.base.Preconditions; +import com.google.inject.Inject; +import com.sun.jersey.spi.container.ContainerRequest; +import org.apache.druid.server.security.Access; +import org.apache.druid.server.security.AuthorizationUtils; +import org.apache.druid.server.security.AuthorizerMapper; +import org.apache.druid.server.security.ForbiddenException; +import org.apache.druid.server.security.Resource; +import org.apache.druid.server.security.ResourceAction; +import org.apache.druid.server.security.ResourceType; + +/** + * This resource filter is used for data shuffle between native parallel index tasks. See ShuffleResource for details. + * + * It currently performs the authorization check for DATASOURCE, but ideally, it should be the authorization check + * for task data. This issue should be addressed in the future. Review comment: Removed this class now in favor of using `StateResourceFilter`. This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services - To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org For additional commands, e-mail: commits-h...@druid.apache.org
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304635767
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304107124
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304107095
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304107143
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/http/ShuffleResource.java
##
@@ -0,0 +1,116 @@
+/*
+ * 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.druid.indexing.worker.http;
+
+import com.google.common.io.ByteStreams;
+import com.google.inject.Inject;
+import com.sun.jersey.spi.container.ResourceFilters;
+import org.apache.druid.indexing.worker.IntermediaryDataManager;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.StringUtils;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.server.http.security.TaskShuffleResourceFilter;
+import org.joda.time.Interval;
+
+import javax.ws.rs.DELETE;
+import javax.ws.rs.GET;
+import javax.ws.rs.Path;
+import javax.ws.rs.PathParam;
+import javax.ws.rs.Produces;
+import javax.ws.rs.QueryParam;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.core.Response.Status;
+import javax.ws.rs.core.StreamingOutput;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.List;
+
+@Path("/druid/worker/v1/shuffle")
+@ResourceFilters(TaskShuffleResourceFilter.class)
+public class ShuffleResource
+{
+ private static final Logger log = new Logger(ShuffleResource.class);
+
+ private final IntermediaryDataManager intermediaryDataManager;
+
+ @Inject
+ public ShuffleResource(IntermediaryDataManager intermediaryDataManager)
+ {
+this.intermediaryDataManager = intermediaryDataManager;
+ }
+
+ @GET
+ @Path("/task/{supervisorTaskId}/partition")
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response getPartition(
+ @PathParam("supervisorTaskId") String supervisorTaskId,
+ @QueryParam("dataSource") String dataSource,
+ @QueryParam("startTime") String startTime,
+ @QueryParam("endTime") String endTime,
+ @QueryParam("partitionId") int partitionId
+ )
+ {
+final Interval interval = new Interval(DateTimes.of(startTime),
DateTimes.of(endTime));
+final List partitionFiles =
intermediaryDataManager.findPartitionFiles(
+supervisorTaskId,
+interval,
+partitionId
+);
+
+if (partitionFiles.isEmpty()) {
+ final String errorMessage = StringUtils.format(
+ "Can't find the partition for supervisor[%s], interval[%s], and
partitionId[%s]",
+ supervisorTaskId,
+ interval,
+ partitionId
+ );
+ return Response.status(Status.NOT_FOUND).entity(errorMessage).build();
+} else {
+ return Response.ok(
+ (StreamingOutput) output -> {
+for (File partitionFile : partitionFiles) {
+ try (final FileInputStream fileInputStream = new
FileInputStream(partitionFile)) {
+ByteStreams.copy(fileInputStream, output);
+ }
+}
+ }
+ ).build();
+}
+ }
+
+ @DELETE
+ @Path("/task/{supervisorTaskId}")
+ public Response deletePartitions(
+ @PathParam("supervisorTaskId") String supervisorTaskId,
+ @QueryParam("dataSource") String dataSource
Review comment:
It's.. used in `TaskShuffleResourceFilter` to check authorization. I know
this is weird and, ideally, the resourceFilter should check authorization for
task data instead of dataSource. However, the current security system only
supports dataSource-level authorization. Added a comment.
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
-
To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org
For additional commands, e-mail: commits-h...@druid.apache.org
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304107048
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/common/config/TaskConfig.java
##
@@ -89,6 +94,13 @@ public TaskConfig(
this.directoryLockTimeout = directoryLockTimeout == null
? DEFAULT_DIRECTORY_LOCK_TIMEOUT
: directoryLockTimeout;
+if (intermediarySegmentsLocations == null) {
+ this.intermediarySegmentsLocations = Collections.singletonList(
+ new StorageLocationConfig(new
File(System.getProperty("java.io.tmpdir"), "intermediary-segments"), null, null)
Review comment:
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
-
To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org
For additional commands, e-mail: commits-h...@druid.apache.org
[GitHub] [incubator-druid] jihoonson commented on a change in pull request #8088: Add intermediary data server for shuffle
jihoonson commented on a change in pull request #8088: Add intermediary data
server for shuffle
URL: https://github.com/apache/incubator-druid/pull/8088#discussion_r304107067
##
File path:
indexing-service/src/main/java/org/apache/druid/indexing/worker/IntermediaryDataManager.java
##
@@ -0,0 +1,270 @@
+/*
+ * 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.druid.indexing.worker;
+
+import com.google.common.collect.Iterators;
+import com.google.common.io.Files;
+import com.google.inject.Inject;
+import org.apache.commons.io.FileUtils;
+import org.apache.druid.client.indexing.IndexingServiceClient;
+import org.apache.druid.client.indexing.TaskStatus;
+import org.apache.druid.guice.ManageLifecycle;
+import org.apache.druid.indexing.common.config.TaskConfig;
+import org.apache.druid.indexing.worker.config.WorkerConfig;
+import org.apache.druid.java.util.common.DateTimes;
+import org.apache.druid.java.util.common.IOE;
+import org.apache.druid.java.util.common.concurrent.Execs;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
+import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
+import org.apache.druid.java.util.common.logger.Logger;
+import org.apache.druid.segment.loading.StorageLocation;
+import org.apache.druid.timeline.DataSegment;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.joda.time.Period;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
+
+/**
+ * This class manages intermediary segments for data shuffle between native
parallel index tasks.
+ * In native parallel indexing, phase 1 tasks store segment files in local
storage of middleManagers
+ * and phase 2 tasks read those files via HTTP.
+ *
+ * The directory where segment files are placed is structured as
+ * {@link
StorageLocation#path}/supervisorTaskId/startTimeOfSegment/endTimeOfSegment/partitionIdOfSegment.
+ *
+ * This class provides interfaces to store, find, and remove segment files.
+ * It also has a self-cleanup mechanism to clean up stale segment files. It
periodically checks the last access time
+ * per supervisorTask and removes its all segment files if the supervisorTask
is not running anymore.
+ */
+@ManageLifecycle
+public class IntermediaryDataManager
+{
+ private static final Logger log = new Logger(IntermediaryDataManager.class);
+
+ private final long intermediaryPartitionDiscoveryPeriodSec;
+ private final long intermediaryPartitionCleanupPeriodSec;
+ private final Period intermediaryPartitionTimeout;
+ private final List intermediarySegmentsLocations;
+ private final IndexingServiceClient indexingServiceClient;
+
+ // supervisorTaskId -> time to check supervisorTask status
+ // This time is initialized when a new supervisorTask is found and updated
whenever a partition is accessed for
+ // the supervisor.
+ private final ConcurrentHashMap supervisorTaskCheckTimes =
new ConcurrentHashMap<>();
+
+ // supervisorTaskId -> cyclic iterator of storage locations
+ private final Map> locationIterators = new
HashMap<>();
+
+ // The overlord is supposed to send a cleanup request as soon as the
supervisorTask is finished in parallel indexing,
+ // but middleManager or indexer could miss the request. This executor is to
automatically clean up unused intermediary
+ // partitions.
+ private ScheduledExecutorService supervisorTaskChecker;
+
+ @Inject
+ public IntermediaryDataManager(
+ WorkerConfig workerConfig,
+ TaskConfig taskConfig,
+ IndexingServiceClient indexingServiceClient
+ )
+ {
+this.intermediaryPartitionDiscoveryPeriodSec =
workerConfig.getIntermediaryPartitionDiscoveryPeriodSec();
+this.intermediaryPartitionCleanupPeriodSec =
workerConfig.getIntermediaryPartitionCleanupPeriodSec();
+
