import (
	"context"
	"log"
	"time"

	dataxr "github.com/koolay/datax-runner"
)

type StdoutLog struct {
}

type StderrLog struct {
}

func (lg *StdoutLog) Write(text string) {
	log.Println("[Stdout]", text)
}

func (lg *StderrLog) Write(text string) {
	log.Println("[Stderr]", text)
}

func main() {
	datax := dataxr.NewDataX(dataxr.Config{
		Debug:      true,
		Xms:        "512m",
		Xmx:        "512m",
		Loglevel:   "error",
		DataxHome:  "./datax",
		Mode:       "",
		Jobid:      "1",
		ConfigFile: "./datax_stream_job.json",
	}, &StdoutLog{}, &StderrLog{})

	ctx := context.Background()
	pid, err := datax.Exec(ctx, "java")
	if err != nil {
		log.Fatal(err)
	}

	log.Println("pid", pid)

	err = datax.Wait(ctx, 30*time.Second)
	if err != nil {
		log.Fatal(err)
	}
}