added tests and licenses

kisaga · kisaga · commit 00bfeeaa1b5d · 2022-06-18T18:27:38.000+03:00
diff --git a/src/main/java/com/twitter/clientlib/exceptions/AuthenticationException.java b/src/main/java/com/twitter/clientlib/exceptions/AuthenticationException.java
@@ -1,3 +1,20 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.exceptions;
 
 public class AuthenticationException extends RuntimeException {
diff --git a/src/main/java/com/twitter/clientlib/exceptions/EmptyStreamTimeoutException.java b/src/main/java/com/twitter/clientlib/exceptions/EmptyStreamTimeoutException.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.exceptions;
 
 public class EmptyStreamTimeoutException extends RuntimeException {
diff --git a/src/main/java/com/twitter/clientlib/exceptions/StreamException.java b/src/main/java/com/twitter/clientlib/exceptions/StreamException.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.exceptions;
 
 public class StreamException extends RuntimeException{
diff --git a/src/main/java/com/twitter/clientlib/exceptions/TooManyRequestsException.java b/src/main/java/com/twitter/clientlib/exceptions/TooManyRequestsException.java
@@ -0,0 +1,30 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.exceptions;
+
+public class TooManyRequestsException extends RuntimeException{
+
+    public TooManyRequestsException(String message) {
+        super(message);
+    }
+
+    public TooManyRequestsException(String message, Throwable cause) {
+        super(message, cause);
+    }
+}
diff --git a/src/main/java/com/twitter/clientlib/query/StreamQueryParameters.java b/src/main/java/com/twitter/clientlib/query/StreamQueryParameters.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query;
 
 import com.twitter.clientlib.query.model.*;
diff --git a/src/main/java/com/twitter/clientlib/query/model/Expansion.java b/src/main/java/com/twitter/clientlib/query/model/Expansion.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/query/model/MediaField.java b/src/main/java/com/twitter/clientlib/query/model/MediaField.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/query/model/PlaceField.java b/src/main/java/com/twitter/clientlib/query/model/PlaceField.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/query/model/PollField.java b/src/main/java/com/twitter/clientlib/query/model/PollField.java
@@ -1,3 +1,20 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/query/model/TweetField.java b/src/main/java/com/twitter/clientlib/query/model/TweetField.java
@@ -1,3 +1,20 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/query/model/UserField.java b/src/main/java/com/twitter/clientlib/query/model/UserField.java
@@ -1,3 +1,20 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.query.model;
 
 import java.util.Arrays;
diff --git a/src/main/java/com/twitter/clientlib/stream/TweetsStreamExecutor.java b/src/main/java/com/twitter/clientlib/stream/TweetsStreamExecutor.java
@@ -14,9 +14,6 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 
-NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
-https://openapi-generator.tech
-Do not edit the class manually.
 */
 
 
diff --git a/src/main/java/com/twitter/clientlib/stream/TweetsStreamListener.java b/src/main/java/com/twitter/clientlib/stream/TweetsStreamListener.java
@@ -14,9 +14,6 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 
-NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
-https://openapi-generator.tech
-Do not edit the class manually.
 */
 
 
diff --git a/src/main/java/com/twitter/clientlib/stream/TwitterStream.java b/src/main/java/com/twitter/clientlib/stream/TwitterStream.java
@@ -1,3 +1,21 @@
+/*
+Copyright 2020 Twitter, Inc.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed 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 com.twitter.clientlib.stream;
 
 import com.twitter.clientlib.ApiClient;
@@ -6,12 +24,14 @@
 import com.twitter.clientlib.api.TweetsApi;
 import com.twitter.clientlib.exceptions.AuthenticationException;
 import com.twitter.clientlib.exceptions.StreamException;
+import com.twitter.clientlib.exceptions.TooManyRequestsException;
 import com.twitter.clientlib.query.StreamQueryParameters;
 
 import okio.BufferedSource;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import javax.annotation.PreDestroy;
 import java.util.LinkedList;
 import java.util.List;
 
@@ -39,6 +59,10 @@ public void addListener(TweetsStreamListener listener) {
         listeners.add(listener);
     }
 
+    public void removeListener(TweetsStreamListener listener) {
+        listeners.remove(listener);
+    }
+
     public void sampleStream(StreamQueryParameters streamParameters) {
         try {
             BufferedSource streamResult = tweets.sampleStream(streamParameters);
@@ -50,14 +74,25 @@ public void sampleStream(StreamQueryParameters streamParameters) {
                 logger.error("Authentication didn't work");
                 throw new AuthenticationException("Not authenticated. Please check the credentials", e);
             }
-            throw new StreamException("An exception occurred during stream execution", e);
+            if(e.getCode() == 429){
+                /* for this error twitter indicates that to implement a reconnection mechanism
+                *  see: https://developer.twitter.com/en/docs/twitter-api/tweets/volume-streams/integrate/handling-disconnections
+                */
+                throw new TooManyRequestsException("Too many requests. Service responded with 429 status code");
+            }
+            throw new StreamException("An exception occurred during stream execution ",e);
         }
     }
 
     public void shutdown() {
         executor.shutdown();
     }
 
+    @PreDestroy
+    public void preDestroy() {
+        executor.shutdown();
+    }
+
     private void initBasePath() {
         String basePath = System.getenv("TWITTER_API_BASE_PATH");
         apiClient.setBasePath(basePath != null ? basePath : "https://api.twitter.com");
diff --git a/src/test/java/com/twitter/clientlib/stream/TwitterStreamTest.java b/src/test/java/com/twitter/clientlib/stream/TwitterStreamTest.java
