diff --git a/src/main/java/org/opensearch/ad/transport/ADCancelTaskNodeRequest.java b/src/main/java/org/opensearch/ad/transport/ADCancelTaskNodeRequest.java index e2ab265bd..70cc45a26 100644 --- a/src/main/java/org/opensearch/ad/transport/ADCancelTaskNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/ADCancelTaskNodeRequest.java @@ -13,11 +13,11 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; -public class ADCancelTaskNodeRequest extends BaseNodeRequest { +public class ADCancelTaskNodeRequest extends TransportRequest { private String detectorId; private String detectorTaskId; private String userName; diff --git a/src/main/java/org/opensearch/ad/transport/ADStatsNodeRequest.java b/src/main/java/org/opensearch/ad/transport/ADStatsNodeRequest.java index a4c8ce0e6..a018cf87b 100644 --- a/src/main/java/org/opensearch/ad/transport/ADStatsNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/ADStatsNodeRequest.java @@ -13,14 +13,14 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; /** * ADStatsNodeRequest to get a nodes stat */ -public class ADStatsNodeRequest extends BaseNodeRequest { +public class ADStatsNodeRequest extends TransportRequest { private ADStatsRequest request; /** diff --git a/src/main/java/org/opensearch/ad/transport/ADTaskProfileNodeRequest.java b/src/main/java/org/opensearch/ad/transport/ADTaskProfileNodeRequest.java index 17dc4fede..42f55cfd7 100644 --- a/src/main/java/org/opensearch/ad/transport/ADTaskProfileNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/ADTaskProfileNodeRequest.java @@ -13,11 +13,11 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; -public class ADTaskProfileNodeRequest extends BaseNodeRequest { +public class ADTaskProfileNodeRequest extends TransportRequest { private String detectorId; public ADTaskProfileNodeRequest(StreamInput in) throws IOException { diff --git a/src/main/java/org/opensearch/ad/transport/CronNodeRequest.java b/src/main/java/org/opensearch/ad/transport/CronNodeRequest.java index 50e73add3..25d200895 100644 --- a/src/main/java/org/opensearch/ad/transport/CronNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/CronNodeRequest.java @@ -13,13 +13,13 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; +import org.opensearch.transport.TransportRequest; /** * Delete model represents the request to an individual node */ -public class CronNodeRequest extends BaseNodeRequest { +public class CronNodeRequest extends TransportRequest { public CronNodeRequest() {} diff --git a/src/main/java/org/opensearch/ad/transport/DeleteModelNodeRequest.java b/src/main/java/org/opensearch/ad/transport/DeleteModelNodeRequest.java index 79f7e0da1..a0cfabdd3 100644 --- a/src/main/java/org/opensearch/ad/transport/DeleteModelNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/DeleteModelNodeRequest.java @@ -13,14 +13,14 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; /** * Delete model represents the request to an individual node */ -public class DeleteModelNodeRequest extends BaseNodeRequest { +public class DeleteModelNodeRequest extends TransportRequest { private String adID; diff --git a/src/main/java/org/opensearch/ad/transport/ProfileNodeRequest.java b/src/main/java/org/opensearch/ad/transport/ProfileNodeRequest.java index 62060e60e..97f77aebe 100644 --- a/src/main/java/org/opensearch/ad/transport/ProfileNodeRequest.java +++ b/src/main/java/org/opensearch/ad/transport/ProfileNodeRequest.java @@ -14,15 +14,15 @@ import java.io.IOException; import java.util.Set; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.ad.model.DetectorProfileName; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; /** * Class representing a nodes's profile request */ -public class ProfileNodeRequest extends BaseNodeRequest { +public class ProfileNodeRequest extends TransportRequest { private ProfileRequest request; public ProfileNodeRequest(StreamInput in) throws IOException { diff --git a/src/test/java/org/opensearch/ad/mock/transport/MockADCancelTaskNodeRequest_1_0.java b/src/test/java/org/opensearch/ad/mock/transport/MockADCancelTaskNodeRequest_1_0.java index a6da59dda..b924f4c59 100644 --- a/src/test/java/org/opensearch/ad/mock/transport/MockADCancelTaskNodeRequest_1_0.java +++ b/src/test/java/org/opensearch/ad/mock/transport/MockADCancelTaskNodeRequest_1_0.java @@ -13,11 +13,11 @@ import java.io.IOException; -import org.opensearch.action.support.nodes.BaseNodeRequest; import org.opensearch.common.io.stream.StreamInput; import org.opensearch.common.io.stream.StreamOutput; +import org.opensearch.transport.TransportRequest; -public class MockADCancelTaskNodeRequest_1_0 extends BaseNodeRequest { +public class MockADCancelTaskNodeRequest_1_0 extends TransportRequest { private String detectorId; private String userName;