// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// Code generated by Microsoft (R) AutoRest Code Generator.

package com.azure.analytics.synapse.spark.models;

import com.azure.core.annotation.Fluent;
import com.fasterxml.jackson.annotation.JsonProperty;

/** The SparkStatement model. */
@Fluent
public final class SparkStatement {
    /*
     * The id property.
     */
    @JsonProperty(value = "id", required = true)
    private int id;

    /*
     * The code property.
     */
    @JsonProperty(value = "code")
    private String code;

    /*
     * The state property.
     */
    @JsonProperty(value = "state")
    private LivyStatementStates state;

    /*
     * The output property.
     */
    @JsonProperty(value = "output")
    private SparkStatementOutput output;

    /**
     * Get the id property: The id property.
     *
     * @return the id value.
     */
    public int getId() {
        return this.id;
    }

    /**
     * Set the id property: The id property.
     *
     * @param id the id value to set.
     * @return the SparkStatement object itself.
     */
    public SparkStatement setId(int id) {
        this.id = id;
        return this;
    }

    /**
     * Get the code property: The code property.
     *
     * @return the code value.
     */
    public String getCode() {
        return this.code;
    }

    /**
     * Set the code property: The code property.
     *
     * @param code the code value to set.
     * @return the SparkStatement object itself.
     */
    public SparkStatement setCode(String code) {
        this.code = code;
        return this;
    }

    /**
     * Get the state property: The state property.
     *
     * @return the state value.
     */
    public LivyStatementStates getState() {
        return this.state;
    }

    /**
     * Set the state property: The state property.
     *
     * @param state the state value to set.
     * @return the SparkStatement object itself.
     */
    public SparkStatement setState(LivyStatementStates state) {
        this.state = state;
        return this;
    }

    /**
     * Get the output property: The output property.
     *
     * @return the output value.
     */
    public SparkStatementOutput getOutput() {
        return this.output;
    }

    /**
     * Set the output property: The output property.
     *
     * @param output the output value to set.
     * @return the SparkStatement object itself.
     */
    public SparkStatement setOutput(SparkStatementOutput output) {
        this.output = output;
        return this;
    }
}
