CloudFilesDaoImpl.java

package org.openstack.atlas.cloudfiles;

import com.rackspacecloud.client.cloudfiles.FilesClient;
import com.rackspacecloud.client.cloudfiles.FilesException;
import com.rackspacecloud.client.cloudfiles.FilesNotFoundException;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.openstack.atlas.auth.AuthUser;
import org.openstack.atlas.config.LbLogsConfigurationKeys;
import org.openstack.atlas.restclients.auth.IdentityClientImpl;
import org.openstack.atlas.util.config.LbConfiguration;
import org.openstack.atlas.util.staticutils.StaticFileUtils;

import javax.activation.FileTypeMap;
import java.io.File;
import java.util.HashMap;

public class CloudFilesDaoImpl implements CloudFilesDao {

    private static final Log LOG = LogFactory.getLog(CloudFilesDaoImpl.class);
    private static final LbConfiguration conf;
    private static final boolean useAdminAuth;

    private FilesClient client;

    private FileTypeMap fileMap = FileTypeMap.getDefaultFileTypeMap();

    static {
        conf = new LbConfiguration();
        useAdminAuth = Boolean.parseBoolean(conf.getString(LbLogsConfigurationKeys.logs_use_service_admin));
    }

    public synchronized void uploadLocalFile(AuthUser user, String containerName, String localFileName, String remoteFileName) throws FilesException {
        File localFile = new File(localFileName);
        if (!localFile.exists()) {
            throw new FilesException("Could not find the local file  " + localFileName + " anymore.", null);
        }

        if (user.isEnabled()) {
            client = new FilesClient(user.getUsername(), user.getAuthKey(), user.getCloudFilesAuthUrl(), null, 5000);
            client.setAuthenticationURL(user.getCloudFilesAuthUrl());

            String fullFilename = localFileName.replaceAll("\\.\\./", "./");
            // the localFilename will always be before the last slash, if we are using
            // slashes (directories)
            String restOfFilename = StaticFileUtils.stripDirectoryFromFileName(fullFilename);

            try {
                try {
                    client.login();
                } catch (Exception e) {
                    // try again to log in, sometimes this fails randomly
                    client.login();
                }
                if (useAdminAuth) { //use the Admin auth token instead of the user's token
                    client.useAlternativeAuth((new IdentityClientImpl()).getAuthToken());
                }
                client.setCurrentRegion(user.getRegion());
                if (!client.containerExists(containerName)) {
                    client.createContainer(containerName);
                }
                try {
                    if (client.getObjectMetaData(containerName, restOfFilename) != null) {
                        throw new FilesException("Could not add file [" + localFileName + "]", null);
                    }
                } catch (FilesNotFoundException e) {
                    //file does not exist, good.
                }
                File file = new File(fullFilename);
                String contentType = fileMap.getContentType(file);

                try {
                    client.storeObjectAs(containerName, file, contentType, remoteFileName, new HashMap());
                } catch (Exception e) {
                    // Files has a tendency to crap out on uploads. Just try again.
                    client.storeObjectAs(containerName, file, contentType, remoteFileName, new HashMap());
                }


            } catch (Exception e) {
                throw new FilesException(fullFilename + "-Failed to upload file:" + e.getMessage(), e);
            }
        } else {
          LOG.info(String.format("User: %s is not enabled and logs will not be processed...", user.getUsername()));
        }
    }


}