com.github.vivekkothari.river.service.IEnricher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of data-river-core Show documentation

There is a newer version: 2.0.0

package com.github.vivekkothari.river.service;

import com.github.vivekkothari.river.bean.MessageValue;
import lombok.SneakyThrows;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * @author vivek.kothari on 30/05/16.
 */
public interface IEnricher {

    ThreadLocal dtf = ThreadLocal.withInitial(() -> new SimpleDateFormat("yyyy-MM-dd hh:mm:ss"));
    ThreadLocal df = ThreadLocal.withInitial(() -> new SimpleDateFormat("yyyy-MM-dd"));

    /**
     * Enriches the incoming {@link MessageValue} and returns the enriched {@link MessageValue}.
     *
     * @param messageValue
     *         incoming messageValue
     *
     * @return
     */
    MessageValue enrich(final MessageValue messageValue);

    /**
     * Returns the {@link Date} on which the record was first created.
     *
     * @param messageValue
     *         incoming messageValue
     *
     * @return by default returns the "created_at" field in the incoming message.
     */
    @SneakyThrows
    default Date getRecordCreationDate(final MessageValue messageValue) {
        return (Date) messageValue.getData()
                                  .get("created_at");
    }

    /**
     * Returns the {@code id} using which it needs to be indexed in ES
     *
     * @param messageValue
     *         incoming messageValue
     *
     * @return by default returns the "id" field in the incoming message.
     */
    default String recordId(final MessageValue messageValue) {
        return String.valueOf(messageValue.getData()
                                          .get("id"));
    }

}