com.spikeify.aerospikeql.QueryUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aerospike-ql Show documentation
SQL wrapper for Aerospike database
The newest version!
package com.spikeify.aerospikeql;

import com.aerospike.client.Language;
import com.aerospike.client.task.RegisterTask;
import com.spikeify.Spikeify;
import com.spikeify.aerospikeql.generate.CodeGenerator;
import com.spikeify.aerospikeql.parse.ParserException;
import com.spikeify.aerospikeql.parse.QueryFields;
import com.spikeify.aerospikeql.parse.QueryParser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.PrintWriter;

public class QueryUtils {

	private static final Logger log = LoggerFactory.getLogger(QueryUtils.class);

	private final String udfFolder;
	private final Spikeify sfy;

	public QueryUtils(Spikeify sfy,
	                  String udfFolder) {
		this.sfy = sfy;
		if(udfFolder == null){
			udfFolder = "udf/";
		}
		this.udfFolder = udfFolder;
	}

	public QueryUtils(Spikeify sfy) {
		this.sfy = sfy;
		udfFolder = "udf/";
	}

	/**
	 * Unregister udf and remove it
	 * @param queryName - name of the query
	 */
	public void removeUdf(String queryName) {
		try {
			String clientUdfPath = udfFolder + queryName + ".lua";
			log.info("Deleting local LUA file '{}'", clientUdfPath);
			// delete local UDF file
			File file = new File(clientUdfPath);
			boolean wasDeleted = file.delete();
			if (wasDeleted){
				log.info("{} was deleted", queryName);
			}
		} catch (Exception e) {
			log.error("Error deleting file ", e);
		}
		unregisterUdf(queryName + ".lua");
	}

	/**
	 * Save and register query with query name
	 * @param queryName - name of the query
	 * @param query - SQL query
	 * @return - Query fields data structure
	 */
	public QueryFields addUdf(String queryName, String query) {
		QueryFields queryFields = getQueryFiels(query);

		if (queryFields != null) {
			String code = CodeGenerator.generateCode(queryFields);
			log.info("Lua code: {}", code);

			saveUdfFile(queryName, code);
			registerUdf(queryName);
		}
		return queryFields;

	}

	protected QueryFields getQueryFiels(String query){
		QueryFields queryFields;
		try {
			queryFields = QueryParser.parseQuery(query);
		} catch (ParserException e) {
			log.error(e.getMessage());
			return null;
		}
		return queryFields;
	}

	private void registerUdf(String udfName) {
		String clientUdfPath = udfFolder + udfName + ".lua";

		log.info("Registering UDF: {}, {}", udfName, clientUdfPath);
		if (udfName != null) {
			RegisterTask task = sfy.getClient().register(null, clientUdfPath, udfName + ".lua", Language.LUA);
			task.waitTillComplete();
		} else {
			log.error("Missing udfName field");
		}
	}

	private void saveUdfFile(String udfName, String content) {
		// recreate LUA file
		String clientUdfPath = udfFolder + udfName + ".lua";
		log.info("Creating file: {}", clientUdfPath);
		File luaFile = new File(clientUdfPath);
		if (luaFile.getParentFile() != null) {
			boolean wasCreated = luaFile.getParentFile().mkdirs();
			if(wasCreated){
				log.info("Parent directory for UDFs was created.");
			}
		}

		PrintWriter printWriter = null;
		try {

			printWriter = new PrintWriter(luaFile);
			printWriter.println(content);

		} catch (Exception e) {
			log.error("Error saving file: ", e);
		} finally {
			if (printWriter != null) {
				printWriter.flush();
				printWriter.close();
			}
		}

	}

	private void unregisterUdf(String udfName) {
		log.info("Unregistering LUA: {}", udfName);
		// remove UDF - update AerospikeClient version
		sfy.getClient().removeUdf(null, udfName);
	}

	public Class findEntity(String entitiesPackageName, String query) throws ParserException {
		Class clazz;
		QueryFields queryFields = QueryParser.parseQuery(query);
		String setName = queryFields.getSet();
		if (!entitiesPackageName.endsWith(".")) {
			entitiesPackageName = entitiesPackageName + ".";

		}
		try {
			clazz = Class.forName(entitiesPackageName + setName);
		} catch (ClassNotFoundException e) {
			log.error(e.getMessage());
			return null;
		}
		return clazz;
	}
}