| /** |
| * Licensed to jclouds, Inc. (jclouds) under one or more |
| * contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. jclouds licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.jclouds.blobstore; |
| |
| import static com.google.common.base.Preconditions.checkNotNull; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.util.Collections; |
| import java.util.Date; |
| import java.util.Map; |
| import java.util.concurrent.ConcurrentHashMap; |
| import java.util.concurrent.ConcurrentMap; |
| import javax.inject.Provider; |
| import javax.inject.Inject; |
| import javax.ws.rs.core.HttpHeaders; |
| import javax.ws.rs.core.UriBuilder; |
| |
| import com.google.common.base.Preconditions; |
| import com.google.common.base.Supplier; |
| import com.google.common.base.Throwables; |
| import com.google.common.collect.Multimaps; |
| |
| import org.jclouds.blobstore.domain.Blob; |
| import org.jclouds.blobstore.domain.Blob.Factory; |
| import org.jclouds.blobstore.domain.MutableBlobMetadata; |
| import org.jclouds.blobstore.options.ListContainerOptions; |
| import org.jclouds.blobstore.util.BlobStoreUtils; |
| import org.jclouds.crypto.Crypto; |
| import org.jclouds.crypto.CryptoStreams; |
| import org.jclouds.date.DateService; |
| import org.jclouds.domain.Location; |
| import org.jclouds.http.HttpUtils; |
| import org.jclouds.io.ContentMetadataCodec; |
| import org.jclouds.io.MutableContentMetadata; |
| import org.jclouds.io.Payload; |
| import org.jclouds.io.Payloads; |
| import org.jclouds.io.payloads.ByteArrayPayload; |
| import org.jclouds.io.payloads.DelegatingPayload; |
| |
| public class TransientStorageStrategy implements LocalStorageStrategy { |
| private final ConcurrentMap<String, ConcurrentMap<String, Blob>> containerToBlobs = new ConcurrentHashMap<String, ConcurrentMap<String, Blob>>(); |
| private final ConcurrentMap<String, Location> containerToLocation = new ConcurrentHashMap<String, Location>(); |
| private final Supplier<Location> defaultLocation; |
| private final DateService dateService; |
| private final Factory blobFactory; |
| private final Crypto crypto; |
| private final ContentMetadataCodec contentMetadataCodec; |
| private final Provider<UriBuilder> uriBuilders; |
| |
| @Inject |
| TransientStorageStrategy(final Supplier<Location> defaultLocation, |
| DateService dateService, Factory blobFactory, Crypto crypto, |
| ContentMetadataCodec contentMetadataCodec, |
| Provider<UriBuilder> uriBuilders) { |
| this.defaultLocation = defaultLocation; |
| this.dateService = dateService; |
| this.blobFactory = blobFactory; |
| this.crypto = crypto; |
| this.contentMetadataCodec = contentMetadataCodec; |
| this.uriBuilders = uriBuilders; |
| } |
| |
| @Override |
| public Iterable<String> getAllContainerNames() { |
| return containerToBlobs.keySet(); |
| } |
| |
| @Override |
| public boolean containerExists(final String containerName) { |
| return containerToBlobs.containsKey(containerName); |
| } |
| |
| @Override |
| public void clearContainer(final String containerName) { |
| containerToBlobs.get(containerName).clear(); |
| } |
| |
| @Override |
| public void clearContainer(String container, ListContainerOptions options) { |
| // TODO implement options |
| clearContainer(container); |
| } |
| |
| @Override |
| public boolean createContainerInLocation(final String containerName, final Location location) { |
| ConcurrentMap<String, Blob> origValue = containerToBlobs.putIfAbsent( |
| containerName, new ConcurrentHashMap<String, Blob>()); |
| if (origValue != null) { |
| return false; |
| } |
| containerToLocation.put(containerName, location != null ? location : defaultLocation.get()); |
| return true; |
| } |
| |
| @Override |
| public void deleteContainer(final String containerName) { |
| containerToBlobs.remove(containerName); |
| } |
| |
| @Override |
| public boolean blobExists(final String containerName, final String blobName) { |
| Map<String, Blob> map = containerToBlobs.get(containerName); |
| return map != null && map.containsKey(blobName); |
| } |
| |
| @Override |
| public Blob getBlob(final String containerName, final String blobName) { |
| Map<String, Blob> map = containerToBlobs.get(containerName); |
| return map == null ? null : map.get(blobName); |
| } |
| |
| @Override |
| public String putBlob(final String containerName, final Blob blob) throws IOException { |
| Blob newBlob = createUpdatedCopyOfBlobInContainer(containerName, blob); |
| Map<String, Blob> map = containerToBlobs.get(containerName); |
| map.put(newBlob.getMetadata().getName(), newBlob); |
| Payloads.calculateMD5(newBlob, crypto.md5()); |
| String eTag = CryptoStreams.hex(newBlob.getPayload().getContentMetadata().getContentMD5()); |
| return eTag; |
| } |
| |
| @Override |
| public void removeBlob(final String containerName, final String blobName) { |
| Map<String, Blob> map = containerToBlobs.get(containerName); |
| if (map != null) |
| map.remove(blobName); |
| } |
| |
| @Override |
| public Iterable<String> getBlobKeysInsideContainer(final String containerName) { |
| return containerToBlobs.get(containerName).keySet(); |
| } |
| |
| @Override |
| public Location getLocation(final String containerName) { |
| return containerToLocation.get(containerName); |
| } |
| |
| @Override |
| public String getSeparator() { |
| return "/"; |
| } |
| |
| private Blob createUpdatedCopyOfBlobInContainer(String containerName, Blob in) { |
| checkNotNull(in, "blob"); |
| checkNotNull(in.getPayload(), "blob.payload"); |
| ByteArrayPayload payload = (in.getPayload() instanceof ByteArrayPayload) ? ByteArrayPayload.class.cast(in |
| .getPayload()) : null; |
| if (payload == null) |
| payload = (in.getPayload() instanceof DelegatingPayload) ? (DelegatingPayload.class.cast(in.getPayload()) |
| .getDelegate() instanceof ByteArrayPayload) ? ByteArrayPayload.class.cast(DelegatingPayload.class |
| .cast(in.getPayload()).getDelegate()) : null : null; |
| try { |
| if (payload == null || !(payload instanceof ByteArrayPayload)) { |
| MutableContentMetadata oldMd = in.getPayload().getContentMetadata(); |
| ByteArrayOutputStream out = new ByteArrayOutputStream(); |
| in.getPayload().writeTo(out); |
| payload = (ByteArrayPayload) Payloads.calculateMD5(Payloads.newPayload(out.toByteArray())); |
| HttpUtils.copy(oldMd, payload.getContentMetadata()); |
| } else { |
| if (payload.getContentMetadata().getContentMD5() == null) |
| Payloads.calculateMD5(in, crypto.md5()); |
| } |
| } catch (IOException e) { |
| Throwables.propagate(e); |
| } |
| Blob blob = blobFactory.create(BlobStoreUtils.copy(in.getMetadata())); |
| blob.setPayload(payload); |
| blob.getMetadata().setContainer(containerName); |
| blob.getMetadata().setUri( |
| uriBuilders.get().scheme("mem").host(containerName).path(in.getMetadata().getName()).build()); |
| blob.getMetadata().setLastModified(new Date()); |
| String eTag = CryptoStreams.hex(payload.getContentMetadata().getContentMD5()); |
| blob.getMetadata().setETag(eTag); |
| // Set HTTP headers to match metadata |
| blob.getAllHeaders().replaceValues(HttpHeaders.LAST_MODIFIED, |
| Collections.singleton(dateService.rfc822DateFormat(blob.getMetadata().getLastModified()))); |
| blob.getAllHeaders().replaceValues(HttpHeaders.ETAG, Collections.singleton(eTag)); |
| copyPayloadHeadersToBlob(payload, blob); |
| blob.getAllHeaders().putAll(Multimaps.forMap(blob.getMetadata().getUserMetadata())); |
| return blob; |
| } |
| |
| private void copyPayloadHeadersToBlob(Payload payload, Blob blob) { |
| blob.getAllHeaders().putAll(contentMetadataCodec.toHeaders(payload.getContentMetadata())); |
| } |
| } |