From 226088cf3d546111444a5104df0bf2ceb97e7136 Mon Sep 17 00:00:00 2001
From: Harshrajsinh Thakor <hrthakor7@gmail.com>
Date: Tue, 6 Jun 2017 23:29:35 -0700
Subject: [PATCH] Refactored the Function<T,Integer> to ToIntFunction<T>.

---
 .../info/ata4/junity/serialize/typetree/TypeTreeV3.java   | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java b/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java
index b5bd2f2e..c79a1e88 100644
--- a/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java
+++ b/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java
@@ -22,7 +22,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicInteger;
-import java.util.function.Function;
+import java.util.function.ToIntFunction;
 
 /**
  *
@@ -167,7 +167,7 @@ private void writeNode(DataWriter out, Node<T> node) throws IOException {
         Map<String, Integer> localMap = new LinkedHashMap<>();
         Map<String, Integer> commonMap = StringTable.commonStrings(revision.major()).inverse();
 
-        Function<String, Integer> addStringOffset = typeName -> {
+        ToIntFunction<String> addStringOffset = typeName -> {
             if (commonMap.containsKey(typeName)) {
                 return commonMap.get(typeName);
             } else if (localMap.containsKey(typeName)) {
@@ -181,8 +181,8 @@ private void writeNode(DataWriter out, Node<T> node) throws IOException {
 
         // apply string offsets
         types.forEach(type -> {
-            type.typeOffset(addStringOffset.apply(type.typeName()));
-            type.nameOffset(addStringOffset.apply(type.fieldName()));
+            type.typeOffset(addStringOffset.applyAsInt(type.typeName()));
+            type.nameOffset(addStringOffset.applyAsInt(type.fieldName()));
         });
 
         out.writeInt(types.size());