RefEntityChangedListener.java

/*
 * Copyright (c) www.bugull.com
 * 
 * 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 net.tooan.ynpay.third.mongodb.lucene.backend;

import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import net.tooan.ynpay.third.mongodb.BuguEntity;
import net.tooan.ynpay.third.mongodb.annotations.Default;
import net.tooan.ynpay.third.mongodb.annotations.Ref;
import net.tooan.ynpay.third.mongodb.annotations.RefList;
import net.tooan.ynpay.third.mongodb.cache.DaoCache;
import net.tooan.ynpay.third.mongodb.cache.FieldsCache;
import net.tooan.ynpay.third.mongodb.lucene.BuguIndex;
import net.tooan.ynpay.third.mongodb.lucene.annotations.IndexRef;
import net.tooan.ynpay.third.mongodb.lucene.annotations.IndexRefList;
import net.tooan.ynpay.third.mongodb.mapper.InternalDao;
import net.tooan.ynpay.third.mongodb.mapper.ReferenceUtil;

import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.List;
import java.util.Set;

/**
 * @author Frank Wen(xbwen@hotmail.com)
 */
@SuppressWarnings("unchecked")
public class RefEntityChangedListener {

    private Set<Class<?>> refBySet;

    public RefEntityChangedListener(Set<Class<?>> refBySet) {
        this.refBySet = refBySet;
    }

    public void entityChange(Class<?> refClass, String id) {
        for (Class<?> cls : refBySet) {
            Field[] fields = FieldsCache.getInstance().get(cls);
            for (Field f : fields) {
                processField(refClass, id, cls, f);
            }
        }
    }

    private void processField(Class<?> refClass, String id, Class<?> cls, Field f) {
        boolean match = false;
        String fieldName = f.getName();
        Ref ref = f.getAnnotation(Ref.class);
        if (ref != null && f.getType().equals(refClass) && f.getAnnotation(IndexRef.class) != null) {
            match = true;
            String name = ref.name();
            if (!name.equals(Default.NAME)) {
                fieldName = name;
            }
        } else {
            RefList refList = f.getAnnotation(RefList.class);
            if (refList != null && f.getAnnotation(IndexRefList.class) != null) {
                Class<?> c = null;
                Class<?> type = f.getType();
                if (type.isArray()) {
                    c = type.getComponentType();  //for Array
                } else {
                    ParameterizedType paramType = (ParameterizedType) f.getGenericType();
                    Type[] types = paramType.getActualTypeArguments();
                    if (types.length == 1) {
                        c = (Class) types[0];  //for List, Set
                    } else {
                        c = (Class) types[1];  //for Map
                    }
                }
                if (c != null && c.equals(refClass)) {
                    match = true;
                    String name = refList.name();
                    if (!name.equals(Default.NAME)) {
                        fieldName = name;
                    }
                }
            }
        }
        if (match) {
            InternalDao dao = DaoCache.getInstance().get(cls);
            Object refObj = ReferenceUtil.toDbReference(cls, fieldName, refClass, id);
            DBObject query = new BasicDBObject(fieldName, refObj);
            List<BuguEntity> list = dao.findForLucene(query);
            for (BuguEntity o : list) {
                IndexUpdateTask task = new IndexUpdateTask(o);
                BuguIndex.getInstance().getExecutor().execute(task);
            }
        }
    }

}