net.hasor.dataql.binder.DataQLModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hasor-dataql Show documentation
the hasor DataQL project.
There is a newer version: 4.2.5
/*
 * Copyright 2008-2009 the original author or authors.
 *
 * 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.hasor.dataql.binder;
import net.hasor.core.*;
import net.hasor.dataql.Query;
import net.hasor.dataql.domain.compiler.QIL;
import net.hasor.dataql.domain.compiler.QueryCompiler;
import net.hasor.dataql.domain.parser.ParseException;
import net.hasor.dataql.runtime.QueryRuntime;
import net.hasor.dataql.udfs.collection.First;
import net.hasor.dataql.udfs.collection.Foreach;
import net.hasor.dataql.udfs.collection.Last;
import net.hasor.dataql.udfs.collection.Limit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;
/**
 * 提供 DataQL 初始化功能。
 * @version : 2017-6-08
 * @author 赵永春 ([email protected])
 */
public class DataQLModule implements Module {
    protected Logger logger = LoggerFactory.getLogger(getClass());
    public void loadModule(ApiBinder apiBinder) throws Throwable {
        //
        // .初始化 DataQL
        final QueryRuntime runtime = new QueryRuntime();
        apiBinder.bindType(DataQL.class).toInstance(new DataQL() {
            @Override
            public Query createQuery(String qlString) throws ParseException {
                QIL queryType = QueryCompiler.compilerQuery(qlString);
                return runtime.createEngine(queryType).newQuery();
            }
        });
        //
        // .启动过程
        Hasor.addStartListener(apiBinder.getEnvironment(), new EventListener() {
            @Override
            public void onEvent(String event, Object eventData) throws Throwable {
                AppContext appContext = (AppContext) eventData;
                //
                List udfList = appContext.findBindingBean(DefineUDF.class);
                for (DefineUDF define : udfList) {
                    String defineName = define.getName();
                    runtime.addShareUDF(defineName, define);
                }
            }
        });
        //
        // .UDFs(内置集合函数)
        DataApiBinder dataBinder = apiBinder.tryCast(DataApiBinder.class);
        if (dataBinder == null) {
            return;
        }
        dataBinder.addUDF("foreach", new Foreach());
        dataBinder.addUDF("first", new First());
        dataBinder.addUDF("last", new Last());
        dataBinder.addUDF("limit", new Limit());
    }
}