Search in sources :

Example 1 with INT

use of com.datastax.oss.driver.api.core.type.DataTypes.INT in project java-driver by datastax.

the class CoreGraphDataTypeITBase method should_insert_and_retrieve_nested_UDTS_and_tuples.

@Test
public void should_insert_and_retrieve_nested_UDTS_and_tuples() {
    CqlSession session = session();
    // use CQL to create type for now because DSP-17567 is not in yet, so this is more stable
    session.execute(String.format("CREATE TYPE %s.udt1(a int, b text)", graphName()));
    session.execute(String.format("CREATE TYPE %s.udt2(" + "a int" + ", b text" + ", c frozen<udt1>" + ", mylist list<bigint>" + ", mytuple_withlist tuple<varchar, tuple<bigint, frozen<list<bigint>>>>" + ")", graphName()));
    session.execute(String.format("CREATE TYPE %s.udt3(" + "a list<int>" + ", b set<float>" + ", c map<text, bigint>" + ", d list<frozen<list<double>>>" + ", e set<frozen<set<float>>>" + ", f list<frozen<tuple<int, text>>>" + ")", graphName()));
    UserDefinedType udt1 = session.getMetadata().getKeyspace(graphName()).flatMap(keyspace -> keyspace.getUserDefinedType("udt1")).orElseThrow(IllegalStateException::new);
    UdtValue udtValue1 = udt1.newValue(1, "2");
    UserDefinedType udt2 = session.getMetadata().getKeyspace(graphName()).flatMap(keyspace -> keyspace.getUserDefinedType("udt2")).orElseThrow(IllegalStateException::new);
    TupleType secondNested = tupleOf(BIGINT, listOf(BIGINT));
    TupleType firstNested = tupleOf(TEXT, secondNested);
    UdtValue udtValue2 = udt2.newValue(1, "2", udt1.newValue(3, "4"), ImmutableList.of(5L), firstNested.newValue("6", secondNested.newValue(7L, ImmutableList.of(8L))));
    UserDefinedType udt3 = session.getMetadata().getKeyspace(graphName()).flatMap(keyspace -> keyspace.getUserDefinedType("udt3")).orElseThrow(IllegalStateException::new);
    UdtValue udtValue3 = udt3.newValue(ImmutableList.of(1), ImmutableSet.of(2.1f), ImmutableMap.of("3", 4L), ImmutableList.of(ImmutableList.of(5.1d, 6.1d), ImmutableList.of(7.1d)), ImmutableSet.of(ImmutableSet.of(8.1f), ImmutableSet.of(9.1f)), ImmutableList.of(tupleOf(INT, TEXT).newValue(10, "11")));
    Map<String, Object> properties = ImmutableMap.<String, Object>builder().put("frozen(typeOf('udt1'))", udtValue1).put("frozen(typeOf('udt2'))", udtValue2).put("frozen(typeOf('udt3'))", udtValue3).build();
    int vertexID = 1;
    String vertexLabel = "graphBinaryNestedTypes";
    runTest(properties, vertexLabel, vertexID);
}
Also used : Arrays(java.util.Arrays) INT(com.datastax.oss.driver.api.core.type.DataTypes.INT) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) LineString(com.datastax.dse.driver.api.core.data.geometry.LineString) ByteBuffer(java.nio.ByteBuffer) BIGINT(com.datastax.oss.driver.api.core.type.DataTypes.BIGINT) Point(com.datastax.dse.driver.api.core.data.geometry.Point) BigDecimal(java.math.BigDecimal) ImmutableList(com.google.common.collect.ImmutableList) CqlSession(com.datastax.oss.driver.api.core.CqlSession) Uuids(com.datastax.oss.driver.api.core.uuid.Uuids) Map(java.util.Map) LocalTime(java.time.LocalTime) BigInteger(java.math.BigInteger) DseDataTypes(com.datastax.dse.driver.api.core.type.DseDataTypes) Polygon(com.datastax.dse.driver.api.core.data.geometry.Polygon) ImmutableSet(com.google.common.collect.ImmutableSet) ImmutableMap(com.google.common.collect.ImmutableMap) Geo(com.datastax.dse.driver.api.core.graph.predicates.Geo) TupleType(com.datastax.oss.driver.api.core.type.TupleType) Test(org.junit.Test) Instant(java.time.Instant) UdtValue(com.datastax.oss.driver.api.core.data.UdtValue) Sets(com.google.common.collect.Sets) ZoneId(java.time.ZoneId) UserDefinedType(com.datastax.oss.driver.api.core.type.UserDefinedType) ChronoUnit(java.time.temporal.ChronoUnit) DataTypes.listOf(com.datastax.oss.driver.api.core.type.DataTypes.listOf) LocalDate(java.time.LocalDate) CqlDuration(com.datastax.oss.driver.api.core.data.CqlDuration) DataTypes.tupleOf(com.datastax.oss.driver.api.core.type.DataTypes.tupleOf) TEXT(com.datastax.oss.driver.api.core.type.DataTypes.TEXT) UdtValue(com.datastax.oss.driver.api.core.data.UdtValue) TupleType(com.datastax.oss.driver.api.core.type.TupleType) UserDefinedType(com.datastax.oss.driver.api.core.type.UserDefinedType) LineString(com.datastax.dse.driver.api.core.data.geometry.LineString) CqlSession(com.datastax.oss.driver.api.core.CqlSession) Point(com.datastax.dse.driver.api.core.data.geometry.Point) Test(org.junit.Test)

Example 2 with INT

use of com.datastax.oss.driver.api.core.type.DataTypes.INT in project java-driver by datastax.

the class CoreGraphDataTypeITBase method should_create_and_retrieve_correct_data_with_types.

@Test
public void should_create_and_retrieve_correct_data_with_types() {
    CqlSession session = session();
    // use CQL to create type for now because DSP-17567 is not in yet, so this is more stable
    session.execute(String.format("CREATE TYPE %s.udt_graphbinary(simple text, complex tuple<int, text>, missing text)", graphName()));
    session.execute(String.format("CREATE TYPE %s.udt_graphbinarygeo(point 'PointType', line 'LineStringType', poly 'PolygonType')", graphName()));
    ImmutableMap.Builder<String, Object> properties = ImmutableMap.<String, Object>builder().put("Ascii", "test").put("Bigint", 5L).put("Boolean", true).put("Date", LocalDate.of(2007, 7, 7)).put("Decimal", BigDecimal.valueOf(2.3)).put("Double", 4.5d).put("Float", 4.8f).put("Int", 45).put("Smallint", (short) 1).put("Text", "test").put("Time", LocalTime.now(ZoneId.systemDefault())).put("Timeuuid", Uuids.timeBased()).put("Timestamp", Instant.now().truncatedTo(ChronoUnit.MILLIS)).put("Uuid", java.util.UUID.randomUUID()).put("Varint", BigInteger.valueOf(3234)).put("Blob", ByteBuffer.wrap(new byte[] { 1, 2, 3 })).put("Tinyint", (byte) 38).put("listOf(Int)", Arrays.asList(2, 3, 4)).put("setOf(Int)", Sets.newHashSet(2, 3, 4)).put("mapOf(Int, Text)", ImmutableMap.of(2, "two", 4, "four")).put("Duration", CqlDuration.newInstance(1, 2, 3)).put("LineString", Geo.lineString(1, 2, 3, 4, 5, 6)).put("Point", Geo.point(3, 4)).put("Polygon", Geo.polygon(Geo.point(3, 4), Geo.point(5, 4), Geo.point(6, 6))).put("tupleOf(Int, Text)", tupleOf(INT, TEXT).newValue(5, "Bar")).put("typeOf('udt_graphbinary')", session.getMetadata().getKeyspace(graphName()).flatMap(keyspace -> keyspace.getUserDefinedType("udt_graphbinary")).orElseThrow(IllegalStateException::new).newValue("some text", tupleOf(INT, TEXT).newValue(5, "Bar"), "some missing text")).put("typeOf('udt_graphbinarygeo')", session.getMetadata().getKeyspace(graphName()).flatMap(keyspaceMetadata -> keyspaceMetadata.getUserDefinedType("udt_graphbinarygeo")).orElseThrow(IllegalStateException::new).newValue(Point.fromCoordinates(3.3, 4.4), LineString.fromPoints(Point.fromCoordinates(1, 1), Point.fromCoordinates(2, 2), Point.fromCoordinates(3, 3)), Polygon.fromPoints(Point.fromCoordinates(3, 4), Point.fromCoordinates(5, 4), Point.fromCoordinates(6, 6))));
    TupleType tuple = tupleOf(DseDataTypes.POINT, DseDataTypes.LINE_STRING, DseDataTypes.POLYGON);
    tuple.attach(session.getContext());
    properties.put("tupleOf(Point, LineString, Polygon)", tuple.newValue(Point.fromCoordinates(3.3, 4.4), LineString.fromPoints(Point.fromCoordinates(1, 1), Point.fromCoordinates(2, 2), Point.fromCoordinates(3, 3)), Polygon.fromPoints(Point.fromCoordinates(3, 4), Point.fromCoordinates(5, 4), Point.fromCoordinates(6, 6))));
    int vertexID = 1;
    String vertexLabel = "graphBinaryAllTypes";
    runTest(properties.build(), vertexLabel, vertexID);
}
Also used : Arrays(java.util.Arrays) INT(com.datastax.oss.driver.api.core.type.DataTypes.INT) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) LineString(com.datastax.dse.driver.api.core.data.geometry.LineString) ByteBuffer(java.nio.ByteBuffer) BIGINT(com.datastax.oss.driver.api.core.type.DataTypes.BIGINT) Point(com.datastax.dse.driver.api.core.data.geometry.Point) BigDecimal(java.math.BigDecimal) ImmutableList(com.google.common.collect.ImmutableList) CqlSession(com.datastax.oss.driver.api.core.CqlSession) Uuids(com.datastax.oss.driver.api.core.uuid.Uuids) Map(java.util.Map) LocalTime(java.time.LocalTime) BigInteger(java.math.BigInteger) DseDataTypes(com.datastax.dse.driver.api.core.type.DseDataTypes) Polygon(com.datastax.dse.driver.api.core.data.geometry.Polygon) ImmutableSet(com.google.common.collect.ImmutableSet) ImmutableMap(com.google.common.collect.ImmutableMap) Geo(com.datastax.dse.driver.api.core.graph.predicates.Geo) TupleType(com.datastax.oss.driver.api.core.type.TupleType) Test(org.junit.Test) Instant(java.time.Instant) UdtValue(com.datastax.oss.driver.api.core.data.UdtValue) Sets(com.google.common.collect.Sets) ZoneId(java.time.ZoneId) UserDefinedType(com.datastax.oss.driver.api.core.type.UserDefinedType) ChronoUnit(java.time.temporal.ChronoUnit) DataTypes.listOf(com.datastax.oss.driver.api.core.type.DataTypes.listOf) LocalDate(java.time.LocalDate) CqlDuration(com.datastax.oss.driver.api.core.data.CqlDuration) DataTypes.tupleOf(com.datastax.oss.driver.api.core.type.DataTypes.tupleOf) TEXT(com.datastax.oss.driver.api.core.type.DataTypes.TEXT) TupleType(com.datastax.oss.driver.api.core.type.TupleType) LineString(com.datastax.dse.driver.api.core.data.geometry.LineString) CqlSession(com.datastax.oss.driver.api.core.CqlSession) ImmutableMap(com.google.common.collect.ImmutableMap) Point(com.datastax.dse.driver.api.core.data.geometry.Point) Test(org.junit.Test)

Aggregations

LineString (com.datastax.dse.driver.api.core.data.geometry.LineString)2 Point (com.datastax.dse.driver.api.core.data.geometry.Point)2 Polygon (com.datastax.dse.driver.api.core.data.geometry.Polygon)2 Geo (com.datastax.dse.driver.api.core.graph.predicates.Geo)2 DseDataTypes (com.datastax.dse.driver.api.core.type.DseDataTypes)2 CqlSession (com.datastax.oss.driver.api.core.CqlSession)2 CqlDuration (com.datastax.oss.driver.api.core.data.CqlDuration)2 UdtValue (com.datastax.oss.driver.api.core.data.UdtValue)2 BIGINT (com.datastax.oss.driver.api.core.type.DataTypes.BIGINT)2 INT (com.datastax.oss.driver.api.core.type.DataTypes.INT)2 TEXT (com.datastax.oss.driver.api.core.type.DataTypes.TEXT)2 DataTypes.listOf (com.datastax.oss.driver.api.core.type.DataTypes.listOf)2 DataTypes.tupleOf (com.datastax.oss.driver.api.core.type.DataTypes.tupleOf)2 TupleType (com.datastax.oss.driver.api.core.type.TupleType)2 UserDefinedType (com.datastax.oss.driver.api.core.type.UserDefinedType)2 Uuids (com.datastax.oss.driver.api.core.uuid.Uuids)2 ImmutableList (com.google.common.collect.ImmutableList)2 ImmutableMap (com.google.common.collect.ImmutableMap)2 ImmutableSet (com.google.common.collect.ImmutableSet)2 Sets (com.google.common.collect.Sets)2