Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.interceptor;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.cache.Cache;

import org.springframework.cache.CacheManager;

import org.springframework.stereotype.Component;

import org.springframework.web.servlet.HandlerInterceptor;

import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

@Component

public class PostInterceptor implements HandlerInterceptor {

    private static final Logger LOGGER = LogManager.getLogger(RoleInterceptor.class);

    @Autowired

    private CacheManager timeoutCacheManager;

    @Override

    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object arg2, Exception arg3)

            throws Exception {

    }

    @Override

    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object arg2, ModelAndView arg3)

            throws Exception {

        LOGGER.info("request is received after : " + request.getRequestURL().toString());

    }

    @Override

    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        //LOGGER.info("request is received in post interceptor before [{}] with method [{}]", request.getRequestURL().toString(), request.getMethod());

        if ("POST".equalsIgnoreCase(request.getMethod())) {

            String idempotencyKey = request.getHeader("IdempotencyKey");

            if (idempotencyKey == null || idempotencyKey.isEmpty()) {

//                response.sendError(HttpServletResponse.SC_BAD_REQUEST, "Missing idempotency token in header");

                return true;

            }

            Cache cache = timeoutCacheManager.getCache("IdempotencyKey");

            if (cache != null) {

                if (cache.get(idempotencyKey) != null) {

                    response.setStatus(HttpServletResponse.SC_OK);

                    response.getWriter().write("Duplicate request. Idempotency Key already processed.");

                    return false;

                } else {

                    cache.put(idempotencyKey, "PROCESSED");

                    return true;

                }

            }

        }

        return true;

    }

}