All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.linkedin.zookeeper.cli.commands.UploadCommand.groovy Maven / Gradle / Ivy

There is a newer version: 1.5.1
Show newest version
/*
 * Copyright 2010-2010 LinkedIn, Inc
 *
 * 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.linkedin.zookeeper.cli.commands

import org.apache.zookeeper.ZooDefs.Ids
import org.apache.zookeeper.CreateMode
import org.linkedin.zookeeper.client.IZKClient

/**
 * Upload a file in zookeeper
 */
public class UploadCommand extends AbstractCommand 
{

  public CliBuilder getCli()
  {
    def cli = new CliBuilder(usage: 'upload -f   ...')
    cli.f(longOpt: 'force', 'force upload')
    return cli
  }

  public void doExecute(IZKClient client, OptionAccessor options)
  {
    boolean force = options.f
    def source = options.arguments()[0]
    def paths = options.arguments()[1..-1]

    log.debug "UPLOAD command"
    log.debug "force=${force}, source=${source}, paths=${paths}, options=${options.arguments()}"

    // execute
    paths.each { path ->
      if (!client.exists(path) || force)
      {
        def buffer = new File(source).readBytes()
        if (client.exists(path))
        {
          client.setByteData(path, buffer)   
          println "${path} - OVERWRITING"
        }
        else
        {
          client.createBytesNodeWithParents(path, buffer, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT)
        }
      }
      else
      {
        println "${path} - SKIPPED (already exists)"
      }
    }
  }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy