ars.file.query.AbstractQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ars-core Show documentation
Ars framework core
The newest version!
package ars.file.query;

import java.util.Map;
import java.util.List;
import java.util.Date;
import java.util.Arrays;
import java.util.Iterator;
import java.util.Map.Entry;
import java.util.LinkedList;
import java.util.Collection;
import java.util.Collections;

import ars.util.Beans;
import ars.util.Strings;
import ars.file.Describe;
import ars.file.Describe.Property;
import ars.file.query.Queries.Order;
import ars.file.query.Queries.Condition;
import ars.file.query.Queries.Like.Position;

/**
 * 文件查询集合抽象实现
 *
 * @author wuyongqiang
 */
public abstract class AbstractQuery implements Query {
    protected final String workingDirectory; // 工作目录

    private String path; // 查询操作相对路径
    private boolean loaded; // 集合是否已加载
    private boolean spread; // 是否展开
    private List orders = new LinkedList(); // 排序条件
    private List describes = Collections.emptyList(); // 缓存数据
    private List conditions = new LinkedList(); // 查询条件

    public AbstractQuery(String workingDirectory) {
        if (workingDirectory == null) {
            throw new IllegalArgumentException("WorkingDirectory must not be null");
        }
        this.workingDirectory = Strings.getRealPath(workingDirectory);
    }

    /**
     * 执行文件查询
     *
     * @param path       文件查询相对路径
     * @param spread     是否展开
     * @param conditions 查询条件数组
     * @return 文件描述列表
     */
    protected abstract List execute(String path, boolean spread, Condition... conditions);

    @Override
    public Iterator iterator() {
        try {
            return this.list().iterator();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public Query path(String path) {
        this.path = path;
        return this;
    }

    @Override
    public Query eq(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.eq(property, value));
        }
        return this;
    }

    @Override
    public Query ne(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.ne(property, value));
        }
        return this;
    }

    @Override
    public Query gt(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.gt(property, value));
        }
        return this;
    }

    @Override
    public Query ge(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.ge(property, value));
        }
        return this;
    }

    @Override
    public Query lt(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.lt(property, value));
        }
        return this;
    }

    @Override
    public Query le(Property property, Object value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.le(property, value));
        }
        return this;
    }

    @Override
    public Query between(Property property, Object low, Object high) {
        if (property != null && low != null && high != null) {
            this.conditions.add(Queries.between(property, low, high));
        }
        return this;
    }

    @Override
    public Query start(Property property, String value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.like(property, value, Position.BEGIN));
        }
        return this;
    }

    @Override
    public Query end(Property property, String value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.like(property, value, Position.END));
        }
        return this;
    }

    @Override
    public Query like(Property property, String value) {
        if (property != null && value != null) {
            this.conditions.add(Queries.like(property, value, Position.ANY));
        }
        return this;
    }

    @SuppressWarnings("unchecked")
    @Override
    public Query custom(String key, Object value) {
        if (key == null || value == null) {
            return this;
        }
        if (key.equalsIgnoreCase(ORDER)) {
            Collection collection = value instanceof Collection ? (Collection) value
                : value instanceof String[] ? Arrays.asList((String[]) value) : Arrays.asList(value.toString());
            for (String order : collection) {
                boolean desc = order.charAt(0) == '-';
                String property = desc || order.charAt(0) == '+' ? order.substring(1) : order;
                if (desc) {
                    this.desc(Describe.Property.valueOf(property.toUpperCase()));
                } else {
                    this.asc(Describe.Property.valueOf(property.toUpperCase()));
                }
            }
        } else if (!key.startsWith(MARK)) {
            int index = key.indexOf(MARK);
            String name = index > 0 ? key.substring(0, index) : key; // 属性名称
            String feature = index > 0 ? key.substring(index + MARK.length()) : EQ; // 特性查询类型
            Property property = null;
            try {
                property = Describe.Property.valueOf(name.toUpperCase());
            } catch (Exception e) {
                return this;
            }
            if (feature.equalsIgnoreCase(START)) {
                this.start(property, value.toString());
            } else if (feature.equalsIgnoreCase(END)) {
                this.end(property, value.toString());
            } else if (feature.equalsIgnoreCase(LIKE)) {
                this.like(property, value.toString());
            } else {
                Class type = property == Describe.Property.NAME ? String.class
                    : property == Describe.Property.SIZE ? int.class
                    : property == Describe.Property.MODIFIED ? Date.class : boolean.class;
                value = Beans.toObject(type, value);
                if (feature.equalsIgnoreCase(GE)) {
                    this.ge(property, value);
                } else if (feature.equalsIgnoreCase(GT)) {
                    this.gt(property, value);
                } else if (feature.equalsIgnoreCase(LE)) {
                    this.le(property, value);
                } else if (feature.equalsIgnoreCase(LT)) {
                    this.lt(property, value);
                } else if (feature.equalsIgnoreCase(EQ)) {
                    this.eq(property, value);
                } else if (feature.equalsIgnoreCase(NE)) {
                    this.ne(property, value);
                }
            }
        }
        return this;
    }

    @Override
    public Query custom(Map parameters) {
        if (parameters != null && !parameters.isEmpty()) {
            for (Entry entry : parameters.entrySet()) {
                this.custom(entry.getKey(), entry.getValue());
            }
        }
        return this;
    }

    @Override
    public Query asc(Property... properties) {
        if (properties != null && properties.length > 0) {
            for (Property property : properties) {
                this.orders.add(Queries.asc(property));
            }
        }
        return this;
    }

    @Override
    public Query desc(Property... properties) {
        if (properties != null && properties.length > 0) {
            for (Property property : properties) {
                this.orders.add(Queries.desc(property));
            }
        }
        return this;
    }

    @Override
    public Query spread(boolean spread) {
        this.spread = spread;
        return this;
    }

    @Override
    public List list() {
        if (!this.loaded) {
            this.describes = this.execute(this.path, this.spread, this.conditions.toArray(new Condition[0]));
            if (!this.orders.isEmpty()) {
                Queries.sort(this.describes, this.orders.toArray(new Order[0]));
            }
            this.loaded = true;
        }
        return this.describes;
    }

}