NutMessageLoader.java

package org.nutz.mvc.impl;

import java.io.Reader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.nutz.lang.Lang;
import org.nutz.lang.segment.CharSegment;
import org.nutz.lang.segment.Segment;
import org.nutz.lang.util.MultiLineProperties;
import org.nutz.log.Log;
import org.nutz.log.Logs;
import org.nutz.mvc.MessageLoader;
import org.nutz.mvc.Mvcs;
import org.nutz.resource.NutResource;
import org.nutz.resource.Scans;

public class NutMessageLoader implements MessageLoader {

    private static final Log log = Logs.get();

    public Map<String, Map<String, Object>> load(String refer) {
        Map<String, Map<String, Object>> re = new HashMap<String, Map<String, Object>>();
        List<NutResource> allnrs = Scans.me().scan(refer, "^.+[.]properties$");
        if (log.isDebugEnabled())
            log.debugf("Load Messages in %s resource : [%s]", allnrs.size(), allnrs);
        // 求取路径的最大长度
        int max = 0;
        for (NutResource nr : allnrs) {
            String[] nms = nr.getName().split("[\\\\/]");
            max = Math.max(max, nms.length);
        }

        // 根据第二级目录，编制列表
        Map<String, List<NutResource>> map = new HashMap<String, List<NutResource>>();
        for (NutResource nr : allnrs) {
            String langType;
            String resName = nr.getName();
            if (resName.contains("/"))
                langType = resName.substring(0, resName.indexOf('/'));
            else if (resName.contains("\\"))
                langType = resName.substring(0, resName.indexOf('\\'));
            else
                langType = Mvcs.DEFAULT_MSGS;
            // 按语言类型编制
            List<NutResource> list = map.get(langType);
            if (null == list) {
                list = new ArrayList<NutResource>(10);
                map.put(langType, list);
            }
            list.add(nr);
        }
        // 根据语言的分类，依次构建字符串 Map
        try {
            for (Entry<String, List<NutResource>> entry : map.entrySet()) {
                List<NutResource> nrs = entry.getValue();
                String langType = entry.getKey();
                // 循环读取该语言的文件夹
                for (NutResource nr : nrs) {
                    // 读取多行属性
                    MultiLineProperties p = new MultiLineProperties();
                    Reader r = nr.getReader();
                    p.load(r);
                    r.close();

                    // 获取当前语言的 Map
                    Map<String, Object> msgs = re.get(langType);
                    if (null == msgs) {
                        msgs = new NutMessageMap();
                        re.put(langType, msgs);
                    }

                    // 将本地化字符串增加到当前语言
                    for (String key : p.keySet()) {
                        String str = p.get(key);
                        Segment seg = (new CharSegment()).valueOf(str);
                        if (seg.keys().isEmpty())
                            msgs.put(key, str);
                        else
                            msgs.put(key, seg);
                    }

                } // ~ 内部循环结束
            }
        }
        catch (Exception e) {
            throw Lang.wrapThrow(e);
        }
        // 看看有没有默认的,没有的话,取第一个为默认
        // TODO 这段代码应该在 1.b.46 之后的某一个版本删掉
        if (!re.containsKey(Mvcs.DEFAULT_MSGS)) {
            if (re.size() > 0) {
                String first_lang = re.keySet().iterator().next();
                re.put(Mvcs.DEFAULT_MSGS, re.get(first_lang));
            }
        }

        if (log.isDebugEnabled())
            log.debugf("Message Loaded, size = %s", re.size());

        // 返回结果
        return re;
    }

}