RefFieldHandler.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.handler;

import net.tooan.ynpay.third.mongodb.BuguEntity;
import net.tooan.ynpay.third.mongodb.cache.DaoCache;
import net.tooan.ynpay.third.mongodb.cache.FieldsCache;
import net.tooan.ynpay.third.mongodb.lucene.annotations.IndexRefBy;
import net.tooan.ynpay.third.mongodb.mapper.FieldUtil;
import net.tooan.ynpay.third.mongodb.mapper.InternalDao;
import org.apache.lucene.document.Document;

import java.lang.reflect.Field;

/**
 * @author Frank Wen(xbwen@hotmail.com)
 */
public class RefFieldHandler extends AbstractFieldHandler {

    public RefFieldHandler(Object obj, Field field, String prefix) {
        super(obj, field, prefix);
    }

    @Override
    public void handle(Document doc) {
        Object objValue = FieldUtil.get(obj, field);
        if (objValue == null) {
            return;
        }
        BuguEntity entity = (BuguEntity) objValue;
        String refId = entity.getId();
        Class<?> clazz = FieldUtil.getRealType(field);
        InternalDao dao = DaoCache.getInstance().get(clazz);
        Object refObj = dao.findOne(refId);
        if (refObj != null) {
            Field[] fields = FieldsCache.getInstance().get(clazz);
            for (Field f : fields) {
                IndexRefBy irb = f.getAnnotation(IndexRefBy.class);
                if (irb != null) {
                    FieldHandler handler = new RefByFieldHandler(obj.getClass(), refObj, f, prefix);
                    handler.handle(doc);
                }
            }
        }
    }

}