se.carlengstrom.internetonastick.job.AppendLineFileToMarkovJob.java Source code

Introduction

Here is the source code for se.carlengstrom.internetonastick.job.AppendLineFileToMarkovJob.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package se.carlengstrom.internetonastick.job;

import com.fasterxml.jackson.databind.ObjectMapper;
import se.carlengstrom.internetonastick.model.Markov;
import se.carlengstrom.internetonastick.model.builders.MarkovBuilder;

import java.io.IOException;

/**
 *
 * @author Eng
 */
public class AppendLineFileToMarkovJob extends Job {

    public enum Delimiter {
        LINE, PERIOD
    }

    private final String directory;
    private final Delimiter delimiter;

    public AppendLineFileToMarkovJob(Markov m, String directory, Delimiter delimiter) {
        super(m);
        this.directory = directory;
        this.delimiter = delimiter;
    }

    @Override
    public void jobRun() throws IOException {
        if (delimiter == Delimiter.LINE) {
            MarkovBuilder.appendMarkovFromLineDelimitedText(getMarkov(), directory + "/source.txt", this);
        } else if (delimiter == Delimiter.PERIOD) {
            MarkovBuilder.appendMarkovFromPeriodDelimitedText(getMarkov(), directory + "/source.txt", this);
        }

        final ObjectMapper mapper = new ObjectMapper();
        mapper.writeValueAsString(getMarkov());
    }
}