org.openmrs.module.encounteralerts.advice.PurgeDefinitionAdvisor.java Source code

Introduction

Here is the source code for org.openmrs.module.encounteralerts.advice.PurgeDefinitionAdvisor.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */

package org.openmrs.module.encounteralerts.advice;

import java.lang.reflect.Method;

import org.aopalliance.aop.Advice;
import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.openmrs.api.context.Context;
import org.openmrs.module.encounteralerts.api.EncounterAlertsService;
import org.openmrs.module.reporting.definition.service.SerializedDefinitionService;
import org.openmrs.module.reporting.query.encounter.definition.EncounterQuery;
import org.springframework.aop.Advisor;
import org.springframework.aop.support.StaticMethodMatcherPointcutAdvisor;

public class PurgeDefinitionAdvisor extends StaticMethodMatcherPointcutAdvisor implements Advisor {

    @Override
    public boolean matches(Method method, Class targetClass) {
        return (SerializedDefinitionService.class.isAssignableFrom(targetClass)
                && method.getName().equals("purgeDefinition"));
    }

    public Advice getAdvice() {
        return new PurgeDefinitionAdvice();
    }

    private class PurgeDefinitionAdvice implements MethodInterceptor {

        @Override
        public Object invoke(MethodInvocation invocation) throws Throwable {
            // Identify definition
            for (Object arg : invocation.getArguments()) {
                if (EncounterQuery.class.isAssignableFrom(arg.getClass())) {
                    EncounterAlertsService eaService = Context.getService(EncounterAlertsService.class);
                    eaService.retireAlertsWithQuery((EncounterQuery) arg);
                }
            }
            return invocation.proceed();
        }

    }
}