/*
* Licensed 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.apache.karaf.cellar.cloud;
import org.apache.karaf.cellar.core.discovery.DiscoveryService;
import org.jclouds.blobstore.BlobStore;
import org.jclouds.blobstore.BlobStoreContext;
import org.jclouds.blobstore.BlobStoreContextFactory;
import org.jclouds.blobstore.domain.Blob;
import org.jclouds.blobstore.domain.StorageMetadata;
import org.jclouds.blobstore.options.ListContainerOptions;
import org.joda.time.DateTime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
public class BlobStoreDiscoveryService implements DiscoveryService {
private static final transient Logger LOGGER = LoggerFactory.getLogger(BlobStoreDiscoveryService.class);
private String provider;
private String identity;
private String credential;
private String container;
private Integer validityPeriod = 60;
private String ipAddress = getIpAddress();
BlobStoreContext context;
private BlobStore blobStore;
/**
* Constructor
*/
public BlobStoreDiscoveryService() {
LOGGER.debug("CELLAR CLOUD: blob store discovery service initialized");
}
public void init() {
try {
if (blobStore == null) {
if (context == null) {
context = new BlobStoreContextFactory().createContext(provider, identity, credential);
}
blobStore = context.getBlobStore();
signIn();
}
} catch (Exception ex) {
LOGGER.error("CELLAR CLOUD: error while initializing blob store discovery service", ex);
}
}
public void destroy() {
signOut();
context.close();
}
public void update(Map<String, Object> properties) {
LOGGER.debug("CELLAR CLOUD: updating properties");
}
/**
* Returns a {@link Set} of Ips.
*
* @return
*/
public Set<String> discoverMembers() {
refresh();
Set<String> members = new HashSet<String>();
ListContainerOptions opt = new ListContainerOptions();
for (StorageMetadata md : blobStore.list(container, opt)) {
String ip = md.getName();
Object obj = readBlob(container, ip);
//Check if ip hasn't been updated recently.
if (obj instanceof DateTime) {
DateTime registeredTime = (DateTime) obj;
if (registeredTime != null && registeredTime.plusSeconds(validityPeriod).isAfterNow()) {
members.add(ip);
} else {
blobStore.removeBlob(container, ip);
}
}
}
return members;
}
/**
* Sign In member to the {@link DiscoveryService}.
*/
public void signIn() {
DateTime now = new DateTime();
createBlob(container, ipAddress, now);
}
/**
* Refresh member to the {@link DiscoveryService}.
*/
public void refresh() {
DateTime now = new DateTime();
createBlob(container, ipAddress, now);
}
/**
* Sing out member to the {@link DiscoveryService}.
*/
public void signOut() {
if (blobStore.blobExists(container, ipAddress)) {
blobStore.removeBlob(container, ipAddress);
} else {
LOGGER.debug("CELLAR CLOUD: unable to find the IP address of the current node in the blob store");
}
}
/**
* Reads from a {@link BlobStore}. It returns an Object.
*
* @param container
* @param blobName
* @return
*/
protected Object readBlob(String container, String blobName) {
Object result = null;
ObjectInputStream ois = null;
blobStore.createContainerInLocation(null, container);
InputStream is = blobStore.getBlob(container, blobName).getPayload().getInput();
try {
ois = new ObjectInputStream(is);
result = ois.readObject();
} catch (IOException e) {
LOGGER.error("CELLAR CLOUD: failed to read blob", e);
} catch (ClassNotFoundException e) {
LOGGER.error("CELLAR CLOUD: failed to read blob", e);
} finally {
if (ois != null) {
try {
ois.close();
} catch (IOException e) {
}
}
if (is != null) {
try {
is.close();
} catch (IOException e) {
}
}
}
return result;
}
public void createBlob(String container, String name, Object data) {
Blob blob;
if (blobStore != null) {
if (!blobStore.containerExists(container)) {
blobStore.createContainerInLocation(null, container);
}
if (blobStore.blobExists(container, name)) {
blob = blobStore.getBlob(container, name);
} else {
blob = blobStore.blobBuilder(name).build();
}
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ObjectOutputStream oos = null;
try {
oos = new ObjectOutputStream(baos);
oos.writeObject(data);
blob.setPayload(baos.toByteArray());
blobStore.putBlob(container, blob);
} catch (IOException e) {
LOGGER.error("CELLAR CLOUD: failed to write blob", e);
} finally {
if (oos != null) {
try {
oos.close();
} catch (IOException e) {
}
}
if (baos != null) {
try {
baos.close();
} catch (IOException e) {
}
}
}
}
}
/**
* Get an instance of InetAddress for the local computer
* and return its string representation.
*/
protected String getIpAddress() {
try {
return InetAddress.getLocalHost().getHostAddress();
} catch (UnknownHostException ex) {
LOGGER.error("CELLAR CLOUD: unable to determine IP address for current node", ex);
return null;
}
}
public String getProvider() {
return provider;
}
public void setProvider(String provider) {
this.provider = provider;
}
public String getIdentity() {
return identity;
}
public void setIdentity(String identity) {
this.identity = identity;
}
public String getCredential() {
return credential;
}
public void setCredential(String credential) {
this.credential = credential;
}
public String getContainer() {
return container;
}
public void setContainer(String container) {
this.container = container;
}
public Integer getValidityPeriod() {
return validityPeriod;
}
public void setValidityPeriod(Integer validityPeriod) {
this.validityPeriod = validityPeriod;
}
}