VirtualNodeGroovyRule.java

package com.taobao.tddl.rule.impl;

import com.taobao.tddl.rule.virtualnode.VirtualNodeMap;

/**
 * @author <a href="junyu@taobao.com">junyu</a>
 * @version 1.0
 * @since 1.6
 * @date 2011-7-14 02:08:56
 */
public abstract class VirtualNodeGroovyRule extends GroovyRule<String> {

    private final VirtualNodeMap vNodeMap;

    public VirtualNodeGroovyRule(String expression, VirtualNodeMap vNodeMap){
        super(expression);
        this.vNodeMap = vNodeMap;
    }

    public VirtualNodeGroovyRule(String expression, VirtualNodeMap vNodeMap, String extraPackagesStr){
        super(expression, extraPackagesStr);
        this.vNodeMap = vNodeMap;
    }

    protected String map(String key) {
        return this.vNodeMap.getValue(key);
    }
}