com.astamuse.asta4d.data.InjectTrace Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of asta4d-core Show documentation
core functionalities of asta4d framework, including template and snippt implemention
There is a newer version: 1.2-M2
package com.astamuse.asta4d.data;

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map.Entry;

import com.astamuse.asta4d.Context;

@SuppressWarnings("rawtypes")
public class InjectTrace {

    private static class TraceMap extends HashMap {

        /**
         * 
         */
        private static final long serialVersionUID = 1L;

        Object targetInstance;

    }

    private static class InstanceTraceList extends LinkedList {

        /**
         * 
         */
        private static final long serialVersionUID = 1L;

    }

    private static final String InstanceTraceListSaveKey = InjectTrace.class.getName() + "#InstanceTraceListSaveKey";

    public static final void saveInstanceInjectionTraceInfo(Object instance, Method setter, ContextDataHolder valueHolder) {
        saveInstanceInjectionTraceInfoInner(instance, createTraceKey(setter), valueHolder);
    }

    public static final void saveInstanceInjectionTraceInfo(Object instance, Field field, ContextDataHolder valueHolder) {
        saveInstanceInjectionTraceInfoInner(instance, createTraceKey(field), valueHolder);
    }

    public static final void saveInstanceInjectionTraceInfo(Object instance, String propertyName, ContextDataHolder valueHolder) {
        saveInstanceInjectionTraceInfoInner(instance, "pn:" + propertyName, valueHolder);
    }

    public static final void saveMethodInjectionTraceInfo(Method method, int parameterIndex, ContextDataHolder valueHolder) {
        saveInstanceInjectionTraceInfoInner(null, createTraceKey(method, parameterIndex), valueHolder);
    }

    private static final void saveInstanceInjectionTraceInfoInner(Object instance, String traceKey, ContextDataHolder valueHolder) {
        Context context = Context.getCurrentThreadContext();
        InstanceTraceList traceList = context.getData(InstanceTraceListSaveKey);
        if (traceList == null) {
            traceList = new InstanceTraceList();
            context.setData(InstanceTraceListSaveKey, traceList);
        }
        synchronized (traceList) {
            TraceMap traceMap = null;
            for (TraceMap map : traceList) {
                if (map.targetInstance == instance) {
                    traceMap = map;
                    break;
                }
            }
            if (traceMap == null) {
                traceMap = new TraceMap();
                traceMap.targetInstance = instance;
                traceList.add(traceMap);
            }

            traceMap.put(traceKey, valueHolder);
        }
    }

    public static final ContextDataHolder getInstanceInjectionTraceInfo(Object instance, Method setter) {
        return getInstanceInjectionTraceInfoInner(instance, createTraceKey(setter));
    }

    public static final ContextDataHolder getInstanceInjectionTraceInfo(Object instance, Field field) {
        return getInstanceInjectionTraceInfoInner(instance, createTraceKey(field));
    }

    public static final ContextDataHolder getInstanceInjectionTraceInfo(Object instance, String propertyName) {
        return getInstanceInjectionTraceInfoInner(instance, "pn:" + propertyName);
    }

    public static final ContextDataHolder getMethodInjectionTraceInfo(Method method, int parameterIndex) {
        return getInstanceInjectionTraceInfoInner(null, createTraceKey(method, parameterIndex));
    }

    private static final ContextDataHolder getInstanceInjectionTraceInfoInner(Object instance, String traceKey) {
        Context context = Context.getCurrentThreadContext();
        InstanceTraceList traceList = context.getData(InstanceTraceListSaveKey);
        if (traceList == null) {
            return null;
        }
        synchronized (traceList) {
            TraceMap traceMap = null;
            for (TraceMap map : traceList) {
                if (map.targetInstance == instance) {
                    traceMap = map;
                    break;
                }
            }
            if (traceMap == null) {
                return null;
            }

            return traceMap.get(traceKey);
        }
    }

    private static final String createTraceKey(Method m) {
        return m.getDeclaringClass().getName() + ":" + m.toString();
    }

    private static final String createTraceKey(Field f) {
        return f.getDeclaringClass().getName() + ":" + f.toString();
    }

    private static final String createTraceKey(Method m, int parameterIndex) {
        return m.getDeclaringClass().getName() + ":" + m.toString() + ":" + parameterIndex;
    }

    public static final List retrieveTraceList() {
        Context context = Context.getCurrentThreadContext();
        return context.getData(InstanceTraceListSaveKey);
    }

    public static final void restoreTraceList(List restoreList) {
        if (restoreList == null) {
            return;
        }
        for (TraceMap map : (InstanceTraceList) restoreList) {
            for (Entry entry : map.entrySet()) {
                saveInstanceInjectionTraceInfo(map.targetInstance, entry.getKey(), entry.getValue());
            }
        }
    }

}