org.yamcs.StreamInitializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yamcs-core Show documentation
There is a newer version: 5.10.7
package org.yamcs;

import java.io.File;
import java.io.FileReader;
import java.io.IOException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.yamcs.StreamConfig.StreamConfigEntry;
import org.yamcs.utils.parser.ParseException;
import org.yamcs.yarch.ExecutionContext;
import org.yamcs.yarch.Tuple;
import org.yamcs.yarch.TupleDefinition;
import org.yamcs.yarch.YarchDatabase;
import org.yamcs.yarch.YarchDatabaseInstance;
import org.yamcs.yarch.streamsql.ResultListener;
import org.yamcs.yarch.streamsql.StreamSqlException;
import org.yamcs.yarch.streamsql.StreamSqlParser;
import org.yamcs.yarch.streamsql.StreamSqlStatement;
import org.yamcs.yarch.streamsql.TokenMgrError;

/**
 * Run at the very beginning of Yamcs startup; creates different streams required for Yamcs operation
 * 
 * There are a number of "hardcoded" stream types that can be created without specifying the schema (because the schema
 * has to match the definition expected by other services).
 * 
 * Additional streams can be created by specifying a file containing StreamSQL commands.
 * 
 * 
 * @author nm
 *
 */
public class StreamInitializer {
    private static final Logger log = LoggerFactory.getLogger(StreamInitializer.class);
    final String yamcsInstance;
    YarchDatabaseInstance ydb;

    public static void createStreams(String yamcsInstance) throws IOException {
        StreamInitializer si = new StreamInitializer(yamcsInstance);
        try {
            si.createStreams();
        } catch (StreamSqlException | ParseException e) {
            throw new ConfigurationException("Cannot create streams", e);
        }
    }

    public StreamInitializer(String yamcsInstance) throws ConfigurationException {
        ydb = YarchDatabase.getInstance(yamcsInstance);
        this.yamcsInstance = yamcsInstance;
    }

    public void createStreams() throws StreamSqlException, ParseException, IOException {
        StreamConfig sc = StreamConfig.getInstance(yamcsInstance);
        for (StreamConfigEntry sce : sc.getEntries()) {
            if (sce.type == StreamConfig.StandardStreamType.CMD_HIST) {
                createStream(sce.name, StandardTupleDefinitions.TC);
            } else if (sce.type == StreamConfig.StandardStreamType.TM) {
                createStream(sce.name, StandardTupleDefinitions.TM);
            } else if (sce.type == StreamConfig.StandardStreamType.PARAM) {
                createStream(sce.name, StandardTupleDefinitions.PARAMETER);
            } else if (sce.type == StreamConfig.StandardStreamType.TC) {
                createStream(sce.name, StandardTupleDefinitions.TC);
            } else if (sce.type == StreamConfig.StandardStreamType.EVENT) {
                createStream(sce.name, StandardTupleDefinitions.EVENT);
            } else if (sce.type == StreamConfig.StandardStreamType.PARAMETER_ALARM) {
                createStream(sce.name, StandardTupleDefinitions.PARAMETER_ALARM);
            } else if (sce.type == StreamConfig.StandardStreamType.EVENT_ALARM) {
                createStream(sce.name, StandardTupleDefinitions.EVENT_ALARM);
            } else if (sce.type == StreamConfig.StandardStreamType.INVALID_TM) {
                createStream(sce.name, StandardTupleDefinitions.INVALID_TM);
            } else if (sce.type == StreamConfig.StandardStreamType.SQL_FILE) {
                loadSqlFile(sce.name); // filename in fact
            } else {
                throw new IllegalArgumentException("Unknown stream type " + sce.type);
            }
        }
    }

    private void createStream(String streamName, TupleDefinition tdef) throws StreamSqlException, ParseException {
        StreamSqlStatement stmt = ydb.createStatement("create stream " + streamName + tdef.getStringDefinition());
        ydb.execute(stmt, new ResultListener() {
            @Override
            public void next(Tuple tuple) {
                // normally create stream does not produce any tuple
            }

            @Override
            public void completeExceptionally(Throwable t) {
                log.error("Error while creating stream {}", streamName, t);
            }

            @Override
            public void complete() {
                log.debug("Created stream {}", streamName);
            }
        });
    }

    private void loadSqlFile(Object o) throws IOException, StreamSqlException, ParseException {
        if (!(o instanceof String)) {
            throw new ConfigurationException("Expected to have a filename to load as SQL File");
        }
        log.debug("Loading SQL File {}", o);
        String filename = (String) o;
        File f = new File(filename);
        var ydb = YarchDatabase.getInstance(yamcsInstance);

        try (FileReader reader = new FileReader(f)) {
            StreamSqlParser parser = new StreamSqlParser(reader);
            StreamSqlStatement stmt;
            while ((stmt = parser.StreamSqlStatement()) != null) {
                StreamSqlStatement stmt1 = stmt;
                stmt1.execute(ydb, new ResultListener() {
                    @Override
                    public void next(Tuple tuple) {
                        // swallow result
                    }

                    @Override
                    public void completeExceptionally(Throwable t) {
                        log.error("Query {} eneded in error: ", stmt1, t);
                    }

                    @Override
                    public void complete() {
                        log.debug("Query {} finished", stmt1);
                    }
                });
            }
        } catch (TokenMgrError e) {
            throw new ParseException(e.getMessage());
        }
    }
}