com.yahoo.elide.datastores.aggregation.timegrains.ISOWeek Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elide-datastore-aggregation Show documentation

Elide Data Store for Aggregation

The newest version!

/*
 * Copyright 2021, Yahoo Inc.
 * Licensed under the Apache License, Version 2.0
 * See LICENSE file in project root for terms.
 */
package com.yahoo.elide.datastores.aggregation.timegrains;

import com.yahoo.elide.core.utils.coerce.converters.ElideTypeConverter;
import com.yahoo.elide.core.utils.coerce.converters.Serde;
import com.yahoo.elide.datastores.aggregation.metadata.enums.TimeGrain;

import java.time.DayOfWeek;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.OffsetDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.Date;

/**
 * Time Grain class for ISOWeek.
 */
public class ISOWeek extends Time {

    public static final String FORMAT = "yyyy-MM-dd";

    public ISOWeek(LocalDateTime date) {
        super(date, true, true, true, false, false, false, getSerializer(TimeGrain.ISOWEEK));
    }

    @ElideTypeConverter(type = ISOWeek.class, name = "ISOWeek")
    static public class ISOWeekSerde implements Serde {
        @Override
        public ISOWeek deserialize(Object val) {
            LocalDateTime date;
            if (val instanceof Date) {
                date = LocalDateTime.ofInstant(((Date) val).toInstant(), ZoneOffset.systemDefault());
            } else {
                LocalDate localDate;
                if (val instanceof OffsetDateTime) {
                    OffsetDateTime offsetDateTime = (OffsetDateTime) val;
                    date = offsetDateTime.toLocalDate().atTime(0, 0);
                } else {
                    localDate = LocalDate.parse(val.toString(), DateTimeFormatter.ISO_LOCAL_DATE);
                    date = localDate.atTime(0, 0);
                }
            }

            if (date.getDayOfWeek() != DayOfWeek.MONDAY) {
                throw new IllegalArgumentException("Date string not a Monday");
            }
            return new ISOWeek(date);
        }

        @Override
        public String serialize(ISOWeek val) {
            return val.serializer.format(val);
        }
    }
}