Package org.kitesdk.data.spi

Examples of org.kitesdk.data.spi.StorageKey

130
131
132
133
134
135
136
137
138
139
140
141
    PartitionStrategy strategy = new PartitionStrategy.Builder()
        .identity("name")
        .identity("address")
        .build();

    StorageKey key = new StorageKey(strategy);
    key.replaceValues((List) Lists.newArrayList("John Doe", "NY/USA"));

    Assert.assertEquals(
        new Path("name_copy=John+Doe/address_copy=NY%2FUSA"),
        convert.fromKey(key));
  }
View Full Code Here
145
146
147
148
149
150
151
152
153
154
155
156
  public void toDirNameIdentityWithNonString() {
    PartitionStrategy strategy = new PartitionStrategy.Builder()
        .identity("id")
        .build();

    StorageKey expected = new StorageKey(strategy);
    expected.replace(0, 0L);

    Assert.assertEquals("Should convert to schema type",
        expected,
        convert.toKey(new Path("id=0"), new StorageKey(strategy)));
  }
View Full Code Here
66
67
68
69
70
71
72
73
74
75
76
    fileSystem = FileSystem.getLocal(new Configuration());
    keys = Lists.newArrayList();
    for (Object year : Arrays.asList(2012, 2013)) {
      for (Object month : Arrays.asList(9, 10, 11, 12)) {
        for (Object day : Arrays.asList(22, 24, 25)) {
          StorageKey k = new StorageKey.Builder(strategy)
              .add("year", year).add("month", month).add("day", day).build();
          keys.add(k);
        }
      }
    }
View Full Code Here
98
99
100
101
102
103
104
105
106
107
108
109
        .year("timestamp")
        .month("timestamp")
        .day("timestamp")
        .build();

    StorageKey key = new StorageKey(strategy);
    key.replaceValues((List) Lists.newArrayList(2013, 11, 5));

    Assert.assertEquals(
        new Path("year=2013/month=11/day=05"),
        convert.fromKey(key));
  }
View Full Code Here
115
116
117
118
119
120
121
122
123
124
125
        .year("timestamp")
        .month("timestamp")
        .day("timestamp")
        .build();

    StorageKey expected = new StorageKey(strategy);
    expected.replaceValues((List) Lists.newArrayList(2013, 11, 5));

    Assert.assertEquals(expected, convert.toKey(
        new Path("year=2013/month=11/day=5"), new StorageKey(strategy)));
  }
View Full Code Here
77
78
79
80
81
82
83
84
85
86
87
    keys = Lists.newArrayList();
    for (Object year : Arrays.asList(2012, 2013)) {
      for (Object month : Arrays.asList(9, 10, 11, 12)) {
        for (Object day : Arrays.asList(22, 24, 25)) {
          StorageKey k = new StorageKey.Builder(strategy)
              .add("year", year).add("month", month).add("day", day).build();
          keys.add(k);
        }
      }
    }
View Full Code Here
74
75
76
77
78
79
80
81
82
83
84
  public static void createExpectedKeys() {
    keys = Lists.newArrayList();
    for (Object year : Arrays.asList(2012, 2013)) {
      for (Object month : Arrays.asList(9, 10, 11, 12)) {
        for (Object day : Arrays.asList(22, 24, 25)) {
          StorageKey k = new StorageKey.Builder(strategy)
              .add("year", year).add("month", month).add("day", day).build();
          keys.add(k);
        }
      }
    }
View Full Code Here
104
105
106
107
108
109
110
111
112
113
114
        Dataset<Object> dataset = Datasets.load(datasetUri, Object.class);
        if (dataset.getDescriptor().isPartitioned()) {
          partitionStrategy = dataset.getDescriptor().getPartitionStrategy();
          accessor = DataModelUtil.accessor(
              dataset.getType(), dataset.getDescriptor().getSchema());
          key = new StorageKey(partitionStrategy);
        }
        URL schemaUrl = dataset.getDescriptor().getSchemaUrl();
        if (schemaUrl != null) {
          setAvroSchemaUrl(schemaUrl.toExternalForm());
        }
View Full Code Here
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
      if (wrappedWriter instanceof InitializeAccessor) {
        ((InitializeAccessor) wrappedWriter).initialize();
      }
      return wrappedWriter;
    }
    final StorageKey partitionStratKey = new StorageKey(dataset.getDescriptor().getPartitionStrategy());
    // Return a dataset writer that checks on write that an entity is within the
    // range of the view
    AbstractDatasetWriter<E> writer = new AbstractDatasetWriter<E>() {
      private Predicate<StorageKey> keyPredicate = constraints.toKeyPredicate();

      @Override
      public void initialize() {
        if (wrappedWriter instanceof InitializeAccessor) {
          ((InitializeAccessor) wrappedWriter).initialize();
        }
      }

      @Override
      public void write(E entity) {
        StorageKey key = getAccessor().keyFor(entity,
            constraints.getProvidedValues(), partitionStratKey);
        if (!keyPredicate.apply(key)) {
          throw new IllegalArgumentException("View does not contain entity: " + entity);
        }
        wrappedWriter.write(entity);
View Full Code Here
92
93
94
95
96
97
98
99
    private final StorageKey reusableKey;
    private final PathConversion convert;

    public MakeKey(PartitionStrategy strategy, Schema schema) {
      this.partitioners = strategy.getFieldPartitioners();
      this.reusableKey = new StorageKey(strategy);
      this.convert = new PathConversion(schema);
    }
View Full Code Here
TOP

Related Classes of org.kitesdk.data.spi.StorageKey

  • org.kitesdk.data.flume.Log4jAppender
  • org.kitesdk.data.hbase.DaoView
  • org.kitesdk.data.spi.filesystem.FileSystemPartitionIterator$MakeKey
  • org.kitesdk.data.spi.filesystem.PartitionedDatasetWriter
  • org.kitesdk.data.spi.filesystem.TestFileSystemPartitionIterator
  • org.kitesdk.data.spi.filesystem.TestFileSystemPartitionIteratorProvidedPartitioners
  • org.kitesdk.data.spi.filesystem.TestFileSystemPartitionIteratorTolerance
  • org.kitesdk.data.spi.filesystem.TestPathConversion
  • org.kitesdk.data.DatasetException
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.