package io.ebeaninternal.server.type;

import io.ebean.text.TextException;
import io.ebeaninternal.server.core.BasicTypeConverter;
import io.ebeanservice.docstore.api.mapping.DocPropertyType;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.math.BigDecimal;
import java.sql.SQLException;
import java.sql.Types;
import java.time.Duration;

ScalarType for java.time.Duration (with seconds precision).
/** * ScalarType for java.time.Duration (with seconds precision). */
public class ScalarTypeDuration extends ScalarTypeBase<Duration> { public ScalarTypeDuration() { super(Duration.class, false, Types.BIGINT); } protected ScalarTypeDuration(int jdbcType) { super(Duration.class, false, jdbcType); } public BigDecimal convertToBigDecimal(Duration value) { return (value == null) ? null : DecimalUtils.toDecimal(value); } public Duration convertFromBigDecimal(BigDecimal value) { return (value == null) ? null : DecimalUtils.toDuration(value); } @Override public void bind(DataBind bind, Duration value) throws SQLException { if (value == null) { bind.setNull(Types.BIGINT); } else { bind.setLong(value.getSeconds()); } } @Override public Duration read(DataReader dataReader) throws SQLException { Long value = dataReader.getLong(); return (value == null) ? null : Duration.ofSeconds(value); } @Override public Object toJdbcType(Object value) { if (value instanceof Long) return value; return ((Duration) value).getSeconds(); } @Override public Duration toBeanType(Object value) { if (value instanceof Duration) return (Duration) value; if (value == null) return null; return Duration.ofSeconds(BasicTypeConverter.toLong(value)); } @Override public Duration readData(DataInput dataInput) throws IOException { if (!dataInput.readBoolean()) { return null; } else { return convertFromBigDecimal(new BigDecimal(dataInput.readUTF())); } } @Override public void writeData(DataOutput dataOutput, Duration value) throws IOException { if (value == null) { dataOutput.writeBoolean(false); } else { ScalarHelp.writeUTF(dataOutput, convertToBigDecimal(value).toString()); } } @Override public String formatValue(Duration v) { return v.toString(); } @Override public Duration parse(String value) { return Duration.parse(value); } @Override public boolean isDateTimeCapable() { return false; } @Override public Duration convertFromMillis(long systemTimeMillis) { throw new TextException("Not Supported"); } @Override public Duration jsonRead(JsonParser parser) throws IOException { return Duration.parse(parser.getValueAsString()); } @Override public void jsonWrite(JsonGenerator writer, Duration value) throws IOException { writer.writeString(value.toString()); } @Override public DocPropertyType getDocType() { return DocPropertyType.KEYWORD; } }