src/main/java/org/hedgecode/snooker/json/JsonCollectionEntity.java

   1 /*
   2  * Copyright (c) 2017. Developed by Hedgecode.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.hedgecode.snooker.json;
  18
  19 import java.io.Serializable;
  20 import java.util.Collections;
  21 import java.util.Comparator;
  22 import java.util.Iterator;
  23 import java.util.LinkedHashMap;
  24 import java.util.LinkedList;
  25 import java.util.List;
  26 import java.util.Map;
  27
  28 import org.hedgecode.snooker.api.CollectionEntity;
  29 import org.hedgecode.snooker.api.IdEntity;
  30
  31 /**
  32  * Abstract Collection Entity to JSON deserialize.
  33  *
  34  * @author Dmitry Samoshin aka gotty
  35  */
  36 public abstract class JsonCollectionEntity<E extends IdEntity>
  37         implements CollectionEntity<E>, Serializable
  38 {
  39     private final Map<Integer, E> entities = new LinkedHashMap<>();
  40
  41     protected JsonCollectionEntity(E[] entities) {
  42         for (E entity : entities) {
  43             if (entity != null)
  44                 this.entities.put(
  45                         entity.getId(), entity
  46                 );
  47         }
  48     }
  49
  50     protected JsonCollectionEntity(List<E> entities) {
  51         for (E entity : entities) {
  52             this.entities.put(
  53                     entity.getId(), entity
  54             );
  55         }
  56     }
  57
  58     @Override
  59     public int size() {
  60         return entities.size();
  61     }
  62
  63     @Override
  64     public boolean isEmpty() {
  65         return entities.isEmpty();
  66     }
  67
  68     @Override
  69     public Iterator<E> iterator() {
  70         return entities.values().iterator();
  71     }
  72
  73     @Override
  74     public E byId(int id) {
  75         return entities.get(id);
  76     }
  77
  78     protected void sort(Comparator<E> comparator) {
  79         List<E> entityList = new LinkedList<>(entities.values());
  80         Collections.sort(
  81                 entityList, comparator
  82         );
  83         entities.clear();
  84         for (E entity : entityList) {
  85             entities.put(entity.getId(), entity);
  86         }
  87     }
  88
  89     @Override
  90     public void sortById() {
  91         sort(
  92                 (entity1, entity2) -> entity1.getId() - entity2.getId()
  93         );
  94     }
  95
  96 }