com.yahoo.yqlplus.example.apis.Programs.java Source code

Introduction

Here is the source code for com.yahoo.yqlplus.example.apis.Programs.java

Source

/*
 * Copyright (c) 2017 Yahoo Inc.
 * Licensed under the terms of the Apache version 2.0 license.
 * See LICENSE file for terms.
 */

package com.yahoo.yqlplus.example.apis;

import java.io.IOException;
import java.io.UncheckedIOException;
import java.util.HashMap;
import java.util.Map;

import org.antlr.v4.runtime.RecognitionException;
import org.apache.commons.io.IOUtils;

import com.google.inject.Guice;
import com.google.inject.Injector;
import com.yahoo.yqlplus.engine.CompiledProgram;
import com.yahoo.yqlplus.engine.YQLPlusCompiler;
import com.yahoo.yqlplus.language.parser.ProgramCompileException;

public class Programs {

    // map of path and program
    private static Map<String, CompiledProgram> programMap;
    static {
        initProgramMap();
    }

    private static void initProgramMap() {
        programMap = new HashMap<>();
        try {
            compilePrograms();
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
    }

    public static void compilePrograms() throws IOException {
        String weatherProgram = IOUtils.toString(Programs.class.getResourceAsStream("/programs/weather.program"),
                "UTF-8");
        Injector injector = Guice.createInjector(new SourceModule());
        YQLPlusCompiler compiler = injector.getInstance(YQLPlusCompiler.class);
        try {
            programMap.put("/weather", compiler.compile(weatherProgram));
        } catch (RecognitionException | IOException e) {
            throw new ProgramCompileException("Compiling program: " + weatherProgram + " failed ", e);
        }
    }

    public static CompiledProgram getProgram(String path) {
        return programMap.get(path);
    }

    public static String runProgram(String path, Map<String, Object> params) throws IOException {
        try {
            Object results = programMap.get(path).run(params, true).getResult("weather").get().getResult();
            return HttpUtil.getGson().toJson(results);
        } catch (Exception e) {
            throw new IOException("Executing program failed", e);
        }
    }
}