RefListEncoder.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.encoder;

import net.tooan.ynpay.third.mongodb.BuguEntity;
import net.tooan.ynpay.third.mongodb.annotations.Default;
import net.tooan.ynpay.third.mongodb.annotations.RefList;
import net.tooan.ynpay.third.mongodb.cache.DaoCache;
import net.tooan.ynpay.third.mongodb.mapper.DataType;
import net.tooan.ynpay.third.mongodb.mapper.FieldUtil;
import net.tooan.ynpay.third.mongodb.mapper.InternalDao;
import net.tooan.ynpay.third.mongodb.mapper.ReferenceUtil;

import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.*;
import java.util.Map.Entry;

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

    private RefList refList;

    public RefListEncoder(Object obj, Field field) {
        super(obj, field);
        refList = field.getAnnotation(RefList.class);
    }

    @Override
    public String getFieldName() {
        String fieldName = field.getName();
        String name = refList.name();
        if (!name.equals(Default.NAME)) {
            fieldName = name;
        }
        return fieldName;
    }

    @Override
    public Object encode() {
        Class<?> type = field.getType();
        if (type.isArray()) {
            return encodeArray(type.getComponentType());
        } else {
            return encodeList(type);
        }
    }

    private Object encodeArray(Class<?> clazz) {
        clazz = FieldUtil.getRealType(clazz, field);
        InternalDao dao = DaoCache.getInstance().get(clazz);
        int len = Array.getLength(value);
        List<Object> result = new ArrayList<Object>();
        for (int i = 0; i < len; i++) {
            BuguEntity entity = (BuguEntity) Array.get(value, i);
            if (entity != null) {
                doCascade(dao, entity);
                result.add(ReferenceUtil.toDbReference(refList, entity.getClass(), entity.getId()));
            }
        }
        return result;
    }

    private Object encodeList(Class type) {
        ParameterizedType paramType = (ParameterizedType) field.getGenericType();
        Type[] types = paramType.getActualTypeArguments();
        if (DataType.isList(type)) {
            List<BuguEntity> list = (List<BuguEntity>) value;
            List<Object> result = new ArrayList<Object>();
            Class<?> cls = FieldUtil.getRealType((Class) types[0], field);
            InternalDao dao = DaoCache.getInstance().get(cls);
            for (BuguEntity entity : list) {
                if (entity != null) {
                    doCascade(dao, entity);
                    result.add(ReferenceUtil.toDbReference(refList, entity.getClass(), entity.getId()));
                }
            }
            return result;
        } else if (DataType.isSet(type)) {
            Set<BuguEntity> set = (Set<BuguEntity>) value;
            Set<Object> result = new HashSet<Object>();
            Class<?> cls = FieldUtil.getRealType((Class) types[0], field);
            InternalDao dao = DaoCache.getInstance().get(cls);
            for (BuguEntity entity : set) {
                if (entity != null) {
                    doCascade(dao, entity);
                    result.add(ReferenceUtil.toDbReference(refList, entity.getClass(), entity.getId()));
                }
            }
            return result;
        } else if (DataType.isMap(type)) {
            Map<Object, BuguEntity> map = (Map<Object, BuguEntity>) value;
            Map<Object, Object> result = new HashMap<Object, Object>();
            Class<?> cls = FieldUtil.getRealType((Class) types[1], field);
            InternalDao dao = DaoCache.getInstance().get(cls);
            for (Entry<Object, BuguEntity> entry : map.entrySet()) {
                BuguEntity entity = entry.getValue();
                if (entity != null) {
                    doCascade(dao, entity);
                    result.put(entry.getKey(), ReferenceUtil.toDbReference(refList, entity.getClass(), entity.getId()));
                } else {
                    result.put(entry.getKey(), null);
                }
            }
            return result;
        } else {
            return null;
        }
    }

    private void doCascade(InternalDao dao, BuguEntity entity) {
        if (refList.cascade().toUpperCase().indexOf(Default.CASCADE_CREATE) != -1 || refList.cascade().toUpperCase().indexOf(Default.CASCADE_UPDATE) != -1) {
            dao.save(entity);
        }
    }

}