com.azure.resourcemanager.datafactory.models.AvroWriteSettings Maven / Gradle / Ivy

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

package com.azure.resourcemanager.datafactory.models;

import com.azure.core.annotation.Fluent;
import com.azure.json.JsonReader;
import com.azure.json.JsonToken;
import com.azure.json.JsonWriter;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * Avro write settings.
 */
@Fluent
public final class AvroWriteSettings extends FormatWriteSettings {
    /*
     * The write setting type.
     */
    private String type = "AvroWriteSettings";

    /*
     * Top level record name in write result, which is required in AVRO spec.
     */
    private String recordName;

    /*
     * Record namespace in the write result.
     */
    private String recordNamespace;

    /*
     * Limit the written file's row count to be smaller than or equal to the specified count. Type: integer (or
     * Expression with resultType integer).
     */
    private Object maxRowsPerFile;

    /*
     * Specifies the file name pattern _. when copy from non-file based store
     * without partitionOptions. Type: string (or Expression with resultType string).
     */
    private Object fileNamePrefix;

    /**
     * Creates an instance of AvroWriteSettings class.
     */
    public AvroWriteSettings() {
    }

    /**
     * Get the type property: The write setting type.
     * 
     * @return the type value.
     */
    @Override
    public String type() {
        return this.type;
    }

    /**
     * Get the recordName property: Top level record name in write result, which is required in AVRO spec.
     * 
     * @return the recordName value.
     */
    public String recordName() {
        return this.recordName;
    }

    /**
     * Set the recordName property: Top level record name in write result, which is required in AVRO spec.
     * 
     * @param recordName the recordName value to set.
     * @return the AvroWriteSettings object itself.
     */
    public AvroWriteSettings withRecordName(String recordName) {
        this.recordName = recordName;
        return this;
    }

    /**
     * Get the recordNamespace property: Record namespace in the write result.
     * 
     * @return the recordNamespace value.
     */
    public String recordNamespace() {
        return this.recordNamespace;
    }

    /**
     * Set the recordNamespace property: Record namespace in the write result.
     * 
     * @param recordNamespace the recordNamespace value to set.
     * @return the AvroWriteSettings object itself.
     */
    public AvroWriteSettings withRecordNamespace(String recordNamespace) {
        this.recordNamespace = recordNamespace;
        return this;
    }

    /**
     * Get the maxRowsPerFile property: Limit the written file's row count to be smaller than or equal to the specified
     * count. Type: integer (or Expression with resultType integer).
     * 
     * @return the maxRowsPerFile value.
     */
    public Object maxRowsPerFile() {
        return this.maxRowsPerFile;
    }

    /**
     * Set the maxRowsPerFile property: Limit the written file's row count to be smaller than or equal to the specified
     * count. Type: integer (or Expression with resultType integer).
     * 
     * @param maxRowsPerFile the maxRowsPerFile value to set.
     * @return the AvroWriteSettings object itself.
     */
    public AvroWriteSettings withMaxRowsPerFile(Object maxRowsPerFile) {
        this.maxRowsPerFile = maxRowsPerFile;
        return this;
    }

    /**
     * Get the fileNamePrefix property: Specifies the file name pattern
     * <fileNamePrefix>_<fileIndex>.<fileExtension> when copy from non-file based store without
     * partitionOptions. Type: string (or Expression with resultType string).
     * 
     * @return the fileNamePrefix value.
     */
    public Object fileNamePrefix() {
        return this.fileNamePrefix;
    }

    /**
     * Set the fileNamePrefix property: Specifies the file name pattern
     * <fileNamePrefix>_<fileIndex>.<fileExtension> when copy from non-file based store without
     * partitionOptions. Type: string (or Expression with resultType string).
     * 
     * @param fileNamePrefix the fileNamePrefix value to set.
     * @return the AvroWriteSettings object itself.
     */
    public AvroWriteSettings withFileNamePrefix(Object fileNamePrefix) {
        this.fileNamePrefix = fileNamePrefix;
        return this;
    }

    /**
     * Validates the instance.
     * 
     * @throws IllegalArgumentException thrown if the instance is not valid.
     */
    @Override
    public void validate() {
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public JsonWriter toJson(JsonWriter jsonWriter) throws IOException {
        jsonWriter.writeStartObject();
        jsonWriter.writeStringField("type", this.type);
        jsonWriter.writeStringField("recordName", this.recordName);
        jsonWriter.writeStringField("recordNamespace", this.recordNamespace);
        jsonWriter.writeUntypedField("maxRowsPerFile", this.maxRowsPerFile);
        jsonWriter.writeUntypedField("fileNamePrefix", this.fileNamePrefix);
        if (additionalProperties() != null) {
            for (Map.Entry additionalProperty : additionalProperties().entrySet()) {
                jsonWriter.writeUntypedField(additionalProperty.getKey(), additionalProperty.getValue());
            }
        }
        return jsonWriter.writeEndObject();
    }

    /**
     * Reads an instance of AvroWriteSettings from the JsonReader.
     * 
     * @param jsonReader The JsonReader being read.
     * @return An instance of AvroWriteSettings if the JsonReader was pointing to an instance of it, or null if it was
     * pointing to JSON null.
     * @throws IOException If an error occurs while reading the AvroWriteSettings.
     */
    public static AvroWriteSettings fromJson(JsonReader jsonReader) throws IOException {
        return jsonReader.readObject(reader -> {
            AvroWriteSettings deserializedAvroWriteSettings = new AvroWriteSettings();
            Map additionalProperties = null;
            while (reader.nextToken() != JsonToken.END_OBJECT) {
                String fieldName = reader.getFieldName();
                reader.nextToken();

                if ("type".equals(fieldName)) {
                    deserializedAvroWriteSettings.type = reader.getString();
                } else if ("recordName".equals(fieldName)) {
                    deserializedAvroWriteSettings.recordName = reader.getString();
                } else if ("recordNamespace".equals(fieldName)) {
                    deserializedAvroWriteSettings.recordNamespace = reader.getString();
                } else if ("maxRowsPerFile".equals(fieldName)) {
                    deserializedAvroWriteSettings.maxRowsPerFile = reader.readUntyped();
                } else if ("fileNamePrefix".equals(fieldName)) {
                    deserializedAvroWriteSettings.fileNamePrefix = reader.readUntyped();
                } else {
                    if (additionalProperties == null) {
                        additionalProperties = new LinkedHashMap<>();
                    }

                    additionalProperties.put(fieldName, reader.readUntyped());
                }
            }
            deserializedAvroWriteSettings.withAdditionalProperties(additionalProperties);

            return deserializedAvroWriteSettings;
        });
    }
}