DeleteCascadeTask.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.mapper;

import net.tooan.ynpay.third.mongodb.BuguEntity;
import net.tooan.ynpay.third.mongodb.BuguQuery;
import net.tooan.ynpay.third.mongodb.cache.DaoCache;

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

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

    private List<Field> refFields;
    private List<Field> refListFields;
    private BuguEntity entity;

    public DeleteCascadeTask(List<Field> refFields, List<Field> refListFields, BuguEntity entity) {
        this.refFields = refFields;
        this.refListFields = refListFields;
        this.entity = entity;
    }

    @Override
    public void run() {
        for (Field f : refFields) {
            processRef(f);
        }
        for (Field f : refListFields) {
            processRefList(f);
        }
    }

    private void processRef(Field f) {
        Object value = FieldUtil.get(entity, f);
        if (value != null) {
            Class<?> type = f.getType();
            InternalDao dao = DaoCache.getInstance().get(type);
            dao.remove(value);
        }
    }

    private void processRefList(Field f) {
        Object value = FieldUtil.get(entity, f);
        if (value == null) {
            return;
        }
        List<String> idList = null;
        Class<?> clazz = null;
        Class<?> type = f.getType();
        if (type.isArray()) {
            clazz = type.getComponentType();
            idList = getArrayIds(value);
        } else {
            ParameterizedType paramType = (ParameterizedType) f.getGenericType();
            Type[] types = paramType.getActualTypeArguments();
            int len = types.length;
            if (len == 1) {
                clazz = (Class) types[0];
                idList = getListIds(value, type);
            } else if (len == 2) {
                clazz = (Class) types[1];
                idList = getMapIds(value);
            }
        }
        if (clazz == null) {
            return;
        }
        InternalDao dao = DaoCache.getInstance().get(clazz);
        BuguQuery query = dao.query().in(Operator.ID, idList);
        dao.remove(query);
    }

    private List<String> getArrayIds(Object value) {
        int len = Array.getLength(value);
        List<String> idList = new ArrayList<String>();
        for (int i = 0; i < len; i++) {
            Object item = Array.get(value, i);
            if (item != null) {
                BuguEntity ent = (BuguEntity) item;
                idList.add(ent.getId());
            }
        }
        return idList;
    }

    private List<String> getListIds(Object value, Class type) {
        List<String> idList = new ArrayList<String>();
        if (DataType.isList(type)) {
            List<BuguEntity> list = (List<BuguEntity>) value;
            for (BuguEntity ent : list) {
                if (ent != null) {
                    idList.add(ent.getId());
                }
            }
        } else if (DataType.isSet(type)) {
            Set<BuguEntity> set = (Set<BuguEntity>) value;
            for (BuguEntity ent : set) {
                if (ent != null) {
                    idList.add(ent.getId());
                }
            }
        }
        return idList;
    }

    private List<String> getMapIds(Object value) {
        Map<Object, BuguEntity> map = (Map<Object, BuguEntity>) value;
        Collection<BuguEntity> values = map.values();
        List<String> idList = new ArrayList<String>();
        for (BuguEntity ent : values) {
            if (ent != null) {
                idList.add(ent.getId());
            }
        }
        return idList;
    }

}