io.stargate.sgv2.graphql.schema.graphqlfirst.fetchers.deployed.TimestampParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sgv2-graphqlapi Show documentation

There is a newer version: 2.0.0-ALPHA-17

package io.stargate.sgv2.graphql.schema.graphqlfirst.fetchers.deployed;

import graphql.schema.DataFetchingEnvironment;
import java.time.ZonedDateTime;
import java.time.format.DateTimeParseException;
import java.util.Optional;

public class TimestampParser {

  public static Optional parse(
      Optional cqlTimestampArgumentName, DataFetchingEnvironment environment) {
    return cqlTimestampArgumentName
        .filter(environment::containsArgument)
        .map(
            name -> {
              Object argument = environment.getArgument(name);
              if (argument instanceof Long) {
                return (Long) argument;
              } else if (argument instanceof String) {
                return parseString(((String) argument));
              } else {
                // Can't happen per the types allowed for the argument
                throw new AssertionError("Unexpected timestamp type");
              }
            });
  }

  private static long parseString(String spec) {
    try {
      ZonedDateTime dateTime = ZonedDateTime.parse(spec);
      return dateTime.toEpochSecond() * 1_000_000 + dateTime.getNano() / 1000;
    } catch (DateTimeParseException e2) {
      throw new IllegalArgumentException(
          String.format(
              "Can't parse Timeout '%s' (expected an ISO 8601 zoned date time string)", spec));
    }
  }
}