com.github.epheatt.kafka.connect.morphlines.Records.java Source code

Introduction

Here is the source code for com.github.epheatt.kafka.connect.morphlines.Records.java

Source

/**
 * Copyright  2016 Jeremy Custenborder (jcustenborder@gmail.com)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.github.epheatt.kafka.connect.morphlines;

import com.google.common.collect.ImmutableMap;
import org.apache.kafka.common.record.TimestampType;
import org.apache.kafka.connect.data.Schema;
import org.apache.kafka.connect.data.SchemaBuilder;
import org.apache.kafka.connect.data.Struct;
import org.apache.kafka.connect.sink.SinkRecord;

import java.util.Arrays;
import java.util.List;
import java.util.Map;

class Records {

    public static class TestCase {
        SinkRecord record;
    }

    public static class StringTestCase extends TestCase {
        String string;
    }

    public static class StructTestCase extends TestCase {
        Struct struct;
    }

    public static class MapTestCase extends TestCase {
        Map<String, Object> map;
    }

    static MapTestCase map() {
        MapTestCase testCase = new MapTestCase();

        testCase.map = ImmutableMap.of("firstName", "example", "lastName", "user", "email",
                "example.user@example.com", "age", 27);
        testCase.record = new SinkRecord("testing", 1, null, null, null, testCase.map, 1L, 1484897702123L,
                TimestampType.CREATE_TIME);

        return testCase;
    }

    static StructTestCase struct() {
        StructTestCase testCase = new StructTestCase();

        Schema schema = SchemaBuilder.struct().name("Testing").field("firstName", Schema.OPTIONAL_STRING_SCHEMA)
                .field("lastName", Schema.OPTIONAL_STRING_SCHEMA).field("email", Schema.OPTIONAL_STRING_SCHEMA)
                .field("age", Schema.OPTIONAL_INT32_SCHEMA).build();
        testCase.struct = new Struct(schema).put("firstName", "example").put("lastName", "user")
                .put("email", "example.user@example.com").put("age", 27);
        testCase.record = new SinkRecord("testing", 1, null, null, schema, testCase.struct, 2L, 1484897702123L,
                TimestampType.CREATE_TIME);

        return testCase;
    }

    static StringTestCase string() {
        StringTestCase testCase = new StringTestCase();

        Schema schema = SchemaBuilder.string().name("Testing").build();
        testCase.string = "{\"firstName\":\"example\",\"lastName\":\"user\",\"email\":\"example.user@example.com\",\"age\":27}";
        testCase.record = new SinkRecord("testing", 1, null, null, schema, testCase.string, 3L, 1484897702123L,
                TimestampType.CREATE_TIME);

        return testCase;
    }

    static List<SinkRecord> records() {
        return Arrays.asList(string().record, struct().record, map().record);
    }
}