/*
 * Copyright 2015, TeamDev Ltd. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.spine3.gae.datastore.entityutils;

import com.google.appengine.api.datastore.Entity;
import com.google.protobuf.Message;
import org.spine3.base.CommandRequest;
import org.spine3.base.EventRecord;

import java.util.HashMap;

/**
 * Holds Entity Converters and provides an API for them.
 *
 * @author Mikhayil Mikhaylov
 */
public class EntityConverters {

    private static HashMap<ClassName, EntityConverter> converters = new HashMap<ClassName, EntityConverter>() {{
        put(ClassName.of(EventRecord.class), new EventRecordEntityConverter());
        put(ClassName.of(CommandRequest.class), new CommandRequestEntityConverter());
    }};

    public static Entity convert(Message message) {
        final ClassName messageClassName = ClassName.of(message.getClass());
        if (!converters.containsKey(messageClassName)) {
            throw new IllegalArgumentException("Unknown message type");
        }

        return converters.get(messageClassName).convert(message);
    }

}
@@ -1107,16 +1107,63 @@
 Entity;%0A
+import com.google.common.collect.ImmutableMap;%0A
 import c
@@ -1190,16 +1190,16 @@
 essage;%0A
-
 import o
@@ -1283,20 +1283,16 @@
 va.util.
-Hash
 Map;%0A%0A/*
@@ -1439,20 +1439,16 @@
  static 
-Hash
 Map%3CClas
@@ -1492,15 +1492,28 @@
 new 
-HashMap
+ImmutableMap.Builder
 %3CCla
@@ -1542,11 +1542,8 @@
 r%3E()
- %7B%7B
 %0A   
@@ -1539,32 +1539,37 @@
 rter%3E()%0A        
+    .
 put(ClassName.of
@@ -1573,43 +1573,49 @@
 .of(
-EventRecord.class), new EventRecord
+CommandRequest.class), new CommandRequest
 Enti
@@ -1624,25 +1624,24 @@
 Converter())
-;
 %0A        put
@@ -1637,16 +1637,21 @@
         
+    .
 put(Clas
@@ -1663,49 +1663,43 @@
 .of(
-CommandRequest.class), new CommandRequest
+EventRecord.class), new EventRecord
 Enti
@@ -1716,16 +1716,29 @@
 r())
-;
 %0A
+  
     
-%7D%7D
+      .build()
 ;%0A%0A 
@@ -1976,27 +1976,53 @@
 (%22Un
-known message type%22
+supported message class: %22 + messageClassName
 );%0A 

Make EntityConverters use ImmutableMap inside.