From 650279e781a2ce2454bfc7107a96b626ca73f7d8 Mon Sep 17 00:00:00 2001
From: Alexey Kudravtsev <cdr@intellij.com>
Date: Thu, 7 Mar 2019 02:06:39 +0300
Subject: [PATCH] more notnull

---
 .../com/intellij/util/containers/NotNullList.java    | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/platform/util/src/com/intellij/util/containers/NotNullList.java b/platform/util/src/com/intellij/util/containers/NotNullList.java
index ea0407f32e05..9177a1e2efc0 100644
--- a/platform/util/src/com/intellij/util/containers/NotNullList.java
+++ b/platform/util/src/com/intellij/util/containers/NotNullList.java
@@ -35,6 +35,7 @@ public class NotNullList<E> extends ArrayList<E> {
 
   public NotNullList(@NotNull Collection<? extends E> c) {
     super(c);
+    checkNotNullCollection(c);
   }
 
   @Override
@@ -58,14 +59,20 @@ public class NotNullList<E> extends ArrayList<E> {
     return super.set(index, element);
   }
 
-  private void checkNotNullCollection(Collection<? extends E> c) {
+  @Override
+  @NotNull
+  public E get(int index) {
+    return super.get(index);
+  }
+
+  private void checkNotNullCollection(@NotNull Collection<? extends E> c) {
     for (E e : c) {
       if (e == null) throw new IllegalArgumentException("null element in the collection: "+c);
     }
   }
 
   @Override
-  public boolean addAll(int index, Collection<? extends E> c) {
+  public boolean addAll(int index, @NotNull Collection<? extends E> c) {
     checkNotNullCollection(c);
     return super.addAll(index, c);
   }
@@ -76,6 +83,7 @@ public class NotNullList<E> extends ArrayList<E> {
     final List<E> subList = super.subList(fromIndex, toIndex);
     return new AbstractList<E>() {
       @Override
+      @NotNull
       public E get(int index) {
         return subList.get(index);
       }