/*
 * Copyright (c) 2016-2018 Flux Capacitor.
 *
 * 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 io.fluxcapacitor.javaclient.persisting.caching;

import io.fluxcapacitor.common.api.tracking.MessageBatch;
import io.fluxcapacitor.javaclient.tracking.BatchInterceptor;
import io.fluxcapacitor.javaclient.tracking.Tracker;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

import java.util.function.Consumer;

@Slf4j
@RequiredArgsConstructor
public class CacheInvalidatingInterceptor implements BatchInterceptor {
    private final Cache cache;
    private volatile int[] lastSegment;

    @Override
    public Consumer<MessageBatch> intercept(Consumer<MessageBatch> consumer, Tracker tracker) {
        return batch -> {
            if (shouldInvalidateCache(batch.getSegment())) {
                log.info("Consumer segment changed. Invalidating event model caches.");
                try {
                    cache.invalidateAll();
                } catch (Exception e) {
                    log.error("Failed to invalidate event model cache", e);
                }
            }
            if (batch.getSegment()[0] != batch.getSegment()[1]) {
                lastSegment = batch.getSegment();
            }
            consumer.accept(batch);
        };
    }

    private boolean shouldInvalidateCache(int[] newSegment) {
        return lastSegment != null
                && newSegment[0] != newSegment[1]
                && (newSegment[0] > lastSegment[0] || newSegment[1] < lastSegment[1]);
    }
}

@@ -1088,22 +1088,32 @@
 ate 
-volatile 
+final ThreadLocal%3C
 int%5B%5D
+%3E
  las
@@ -1120,16 +1120,38 @@
 tSegment
+ = new ThreadLocal%3C%3E()
 ;%0A%0A    @
@@ -1392,16 +1392,29 @@
  changed
+ (tracker %7B%7D)
 . Invali
@@ -1440,16 +1440,25 @@
 caches.%22
+, tracker
 );%0A     
@@ -1758,19 +1758,21 @@
 tSegment
- = 
+.set(
 batch.ge
@@ -1781,16 +1781,17 @@
 egment()
+)
 ;%0A      
@@ -1922,22 +1922,21 @@
         
-return
+int%5B%5D
  lastSeg
@@ -1944,31 +1944,68 @@
 ent 
-!
 = 
-null%0A               
+this.lastSegment.get();%0A        return lastSegment != null
  && 
@@ -2034,16 +2034,17 @@
 gment%5B1%5D
+ 
 %0A       

Fix in CacheInvalidatingInterceptor when configuring a consumer to use multiple thread