mirror of
https://github.com/wassname/ray.git
synced 2026-07-06 05:16:30 +08:00
This reverts commit 1b1466748f.
This commit is contained in:
+2
-2
@@ -15,7 +15,7 @@ import io.ray.streaming.runtime.core.graph.executiongraph.ExecutionGraph;
|
||||
import io.ray.streaming.runtime.core.graph.executiongraph.ExecutionJobVertex;
|
||||
import io.ray.streaming.runtime.core.graph.executiongraph.ExecutionVertex;
|
||||
import io.ray.streaming.runtime.core.resource.ResourceType;
|
||||
import io.ray.streaming.runtime.master.context.JobMasterRuntimeContext;
|
||||
import io.ray.streaming.runtime.master.JobRuntimeContext;
|
||||
import io.ray.streaming.runtime.master.graphmanager.GraphManager;
|
||||
import io.ray.streaming.runtime.master.graphmanager.GraphManagerImpl;
|
||||
import java.util.HashMap;
|
||||
@@ -34,7 +34,7 @@ public class ExecutionGraphTest extends BaseUnitTest {
|
||||
public void testBuildExecutionGraph() {
|
||||
Map<String, String> jobConf = new HashMap<>();
|
||||
StreamingConfig streamingConfig = new StreamingConfig(jobConf);
|
||||
GraphManager graphManager = new GraphManagerImpl(new JobMasterRuntimeContext(streamingConfig));
|
||||
GraphManager graphManager = new GraphManagerImpl(new JobRuntimeContext(streamingConfig));
|
||||
JobGraph jobGraph = buildJobGraph();
|
||||
jobGraph.getJobConfig().put("streaming.task.resource.cpu.limitation.enable", "true");
|
||||
|
||||
|
||||
+1
-1
@@ -36,7 +36,7 @@ public class UnionStreamTest {
|
||||
streamSource1
|
||||
.union(streamSource2, streamSource3)
|
||||
.sink((SinkFunction<Integer>) value -> {
|
||||
LOG.info("UnionStreamTest, sink: {}", value);
|
||||
LOG.info("UnionStreamTest: {}", value);
|
||||
try {
|
||||
if (!Files.exists(Paths.get(sinkFileName))) {
|
||||
Files.createFile(Paths.get(sinkFileName));
|
||||
|
||||
+1
-1
@@ -14,7 +14,7 @@ public class JobMasterTest {
|
||||
Assert.assertNull(jobMaster.getGraphManager());
|
||||
Assert.assertNull(jobMaster.getResourceManager());
|
||||
Assert.assertNull(jobMaster.getJobMasterActor());
|
||||
Assert.assertFalse(jobMaster.init(false));
|
||||
Assert.assertFalse(jobMaster.init());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
+3
-3
@@ -7,7 +7,7 @@ import io.ray.streaming.runtime.BaseUnitTest;
|
||||
import io.ray.streaming.runtime.config.StreamingConfig;
|
||||
import io.ray.streaming.runtime.config.global.CommonConfig;
|
||||
import io.ray.streaming.runtime.core.resource.Container;
|
||||
import io.ray.streaming.runtime.master.context.JobMasterRuntimeContext;
|
||||
import io.ray.streaming.runtime.master.JobRuntimeContext;
|
||||
import io.ray.streaming.runtime.util.RayUtils;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
@@ -44,8 +44,8 @@ public class ResourceManagerTest extends BaseUnitTest {
|
||||
Map<String, String> conf = new HashMap<String, String>();
|
||||
conf.put(CommonConfig.JOB_NAME, "testApi");
|
||||
StreamingConfig config = new StreamingConfig(conf);
|
||||
JobMasterRuntimeContext jobMasterRuntimeContext = new JobMasterRuntimeContext(config);
|
||||
ResourceManager resourceManager = new ResourceManagerImpl(jobMasterRuntimeContext);
|
||||
JobRuntimeContext jobRuntimeContext = new JobRuntimeContext(config);
|
||||
ResourceManager resourceManager = new ResourceManagerImpl(jobRuntimeContext);
|
||||
|
||||
// test register container
|
||||
List<Container> containers = resourceManager.getRegisteredContainers();
|
||||
|
||||
+2
-2
@@ -9,7 +9,7 @@ import io.ray.streaming.runtime.core.graph.ExecutionGraphTest;
|
||||
import io.ray.streaming.runtime.core.graph.executiongraph.ExecutionGraph;
|
||||
import io.ray.streaming.runtime.core.resource.Container;
|
||||
import io.ray.streaming.runtime.core.resource.ResourceType;
|
||||
import io.ray.streaming.runtime.master.context.JobMasterRuntimeContext;
|
||||
import io.ray.streaming.runtime.master.JobRuntimeContext;
|
||||
import io.ray.streaming.runtime.master.graphmanager.GraphManager;
|
||||
import io.ray.streaming.runtime.master.graphmanager.GraphManagerImpl;
|
||||
import io.ray.streaming.runtime.master.resourcemanager.ResourceAssignmentView;
|
||||
@@ -64,7 +64,7 @@ public class PipelineFirstStrategyTest extends BaseUnitTest {
|
||||
|
||||
Map<String, String> jobConf = new HashMap<>();
|
||||
StreamingConfig streamingConfig = new StreamingConfig(jobConf);
|
||||
GraphManager graphManager = new GraphManagerImpl(new JobMasterRuntimeContext(streamingConfig));
|
||||
GraphManager graphManager = new GraphManagerImpl(new JobRuntimeContext(streamingConfig));
|
||||
JobGraph jobGraph = ExecutionGraphTest.buildJobGraph();
|
||||
ExecutionGraph executionGraph = ExecutionGraphTest.buildExecutionGraph(graphManager, jobGraph);
|
||||
ResourceAssignmentView assignmentView = strategy.assignResource(containers, executionGraph);
|
||||
|
||||
+1
-1
@@ -9,7 +9,7 @@ import io.ray.streaming.api.function.impl.FlatMapFunction;
|
||||
import io.ray.streaming.api.function.impl.ReduceFunction;
|
||||
import io.ray.streaming.api.stream.DataStreamSource;
|
||||
import io.ray.streaming.runtime.BaseUnitTest;
|
||||
import io.ray.streaming.runtime.transfer.channel.ChannelId;
|
||||
import io.ray.streaming.runtime.transfer.ChannelId;
|
||||
import io.ray.streaming.runtime.util.EnvUtil;
|
||||
import io.ray.streaming.util.Config;
|
||||
import java.io.File;
|
||||
|
||||
+5
-5
@@ -6,11 +6,11 @@ import io.ray.api.Ray;
|
||||
import io.ray.runtime.functionmanager.JavaFunctionDescriptor;
|
||||
import io.ray.streaming.runtime.config.StreamingWorkerConfig;
|
||||
import io.ray.streaming.runtime.transfer.ChannelCreationParametersBuilder;
|
||||
import io.ray.streaming.runtime.transfer.ChannelId;
|
||||
import io.ray.streaming.runtime.transfer.DataMessage;
|
||||
import io.ray.streaming.runtime.transfer.DataReader;
|
||||
import io.ray.streaming.runtime.transfer.DataWriter;
|
||||
import io.ray.streaming.runtime.transfer.TransferHandler;
|
||||
import io.ray.streaming.runtime.transfer.channel.ChannelId;
|
||||
import io.ray.streaming.runtime.transfer.message.DataMessage;
|
||||
import io.ray.streaming.util.Config;
|
||||
import java.lang.management.ManagementFactory;
|
||||
import java.nio.ByteBuffer;
|
||||
@@ -104,7 +104,7 @@ class ReaderWorker extends Worker {
|
||||
new JavaFunctionDescriptor(Worker.class.getName(), "onWriterMessage", "([B)V"),
|
||||
new JavaFunctionDescriptor(Worker.class.getName(), "onWriterMessageSync", "([B)[B"));
|
||||
StreamingWorkerConfig workerConfig = new StreamingWorkerConfig(conf);
|
||||
dataReader = new DataReader(inputQueueList, inputActors, new HashMap<>(), workerConfig);
|
||||
dataReader = new DataReader(inputQueueList, inputActors, workerConfig);
|
||||
|
||||
// Should not GetBundle in RayCall thread
|
||||
Thread readThread = new Thread(Ray.wrapRunnable(new Runnable() {
|
||||
@@ -124,7 +124,7 @@ class ReaderWorker extends Worker {
|
||||
|
||||
int checkPointId = 1;
|
||||
for (int i = 0; i < msgCount * inputQueueList.size(); ++i) {
|
||||
DataMessage dataMessage = (DataMessage) dataReader.read(100);
|
||||
DataMessage dataMessage = dataReader.read(100);
|
||||
|
||||
if (dataMessage == null) {
|
||||
LOGGER.error("dataMessage is null");
|
||||
@@ -232,7 +232,7 @@ class WriterWorker extends Worker {
|
||||
new JavaFunctionDescriptor(Worker.class.getName(), "onReaderMessage", "([B)V"),
|
||||
new JavaFunctionDescriptor(Worker.class.getName(), "onReaderMessageSync", "([B)[B"));
|
||||
StreamingWorkerConfig workerConfig = new StreamingWorkerConfig(conf);
|
||||
dataWriter = new DataWriter(outputQueueList, outputActors, new HashMap<>(), workerConfig);
|
||||
dataWriter = new DataWriter(outputQueueList, outputActors, workerConfig);
|
||||
Thread writerThread = new Thread(Ray.wrapRunnable(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ import static org.testng.Assert.assertEquals;
|
||||
|
||||
|
||||
import io.ray.streaming.runtime.BaseUnitTest;
|
||||
import io.ray.streaming.runtime.transfer.channel.ChannelId;
|
||||
import io.ray.streaming.runtime.util.EnvUtil;
|
||||
import org.testng.annotations.Test;
|
||||
|
||||
|
||||
@@ -3,4 +3,4 @@ log4j.rootLogger=INFO, stdout
|
||||
log4j.appender.stdout=org.apache.log4j.ConsoleAppender
|
||||
log4j.appender.stdout.Target=System.out
|
||||
log4j.appender.stdout.layout=org.apache.log4j.PatternLayout
|
||||
log4j.appender.stdout.layout.ConversionPattern=%d{yyyy-MM-dd HH:mm:ss,SS} %-4p %c{1}:%L [%t] - %m%n
|
||||
log4j.appender.stdout.layout.ConversionPattern=%d{yyyy-MM-dd HH:mm:ss} %-5p %c{1}:%L - %m%n
|
||||
|
||||
Reference in New Issue
Block a user