コード例 #1
0
ファイル: FileUploader.java プロジェクト: ankit-singh/cloud
  /**
   * @param rootDir
   * @param filePath
   * @param clientName
   * @param tcpSocket
   * @return
   */
  public static boolean sendFile(
      String rootDir, String filePath, String clientName, Socket tcpSocket) {
    boolean pushSuccess = false;
    Logger.Log("Root Dir :" + rootDir);
    Logger.Log("File Path:" + filePath);
    Logger.Log("Client Name :" + clientName);
    rootDir = Utilities.getFormattedDir(rootDir);
    filePath = Utilities.getFormattedFilePath(filePath);
    try {
      File file = new File(rootDir + filePath);
      FileInputStream fileInputStream = new FileInputStream(file);
      DataOutputStream socketOutputStream = new DataOutputStream(tcpSocket.getOutputStream());
      long startTime = System.currentTimeMillis();
      // Step 1: prepare the destination for upload
      // Step 2: Confirm and upload file
      byte[] buffer = new byte[BUFFER_SIZE];
      int read;
      int readTotal = 0;
      while ((read = fileInputStream.read(buffer)) != -1) {
        socketOutputStream.write(buffer, 0, read);
        readTotal += read;
      }

      long endTime = System.currentTimeMillis();
      System.out.println(readTotal + " bytes written in " + (endTime - startTime) + " ms.");
      pushSuccess = true;
      fileInputStream.close();
    } catch (Exception e) {
      Logger.Log(e.getMessage());
      e.printStackTrace();
    }
    return pushSuccess;
  }
コード例 #2
0
ファイル: FileUploader.java プロジェクト: ankit-singh/cloud
 /**
  * @param rootDir
  * @param filePath
  * @param clientName
  * @param dest
  * @return
  */
 public static boolean pushFile(
     String rootDir, String filePath, String clientName, ServerDetails dest) {
   boolean pushSuccess = false;
   Logger.Log("Root Dir :" + rootDir);
   Logger.Log("File Path:" + filePath);
   Logger.Log("Client Name :" + clientName);
   rootDir = Utilities.getFormattedDir(rootDir);
   filePath = Utilities.getFormattedFilePath(filePath);
   try {
     Socket tcpSocket = new Socket(dest.getIp(), dest.getPort());
     File file = new File(rootDir + filePath);
     FileInputStream fileInputStream = new FileInputStream(file);
     DataOutputStream socketOutputStream = new DataOutputStream(tcpSocket.getOutputStream());
     DataInputStream socketInputStream = new DataInputStream(tcpSocket.getInputStream());
     long startTime = System.currentTimeMillis();
     String md5 = MD5Checksum.getMD5Checksum(rootDir + filePath);
     // Step 1: prepare the destination for upload
     String request =
         IConstants.PUSH
             + IConstants.DELIMITER
             + clientName
             + IConstants.DELIMITER
             + filePath
             + IConstants.DELIMITER
             + md5;
     Logger.Log("Upload Request :" + request);
     socketOutputStream.writeInt(request.length());
     socketOutputStream.writeBytes(request);
     // Step 2: Confirm and upload file
     int opcode = socketInputStream.readInt();
     if (opcode == IConstants.OK) {
       byte[] buffer = new byte[BUFFER_SIZE];
       int read;
       int readTotal = 0;
       while ((read = fileInputStream.read(buffer)) != -1) {
         socketOutputStream.write(buffer, 0, read);
         readTotal += read;
       }
       socketInputStream.close();
       socketOutputStream.close();
       long endTime = System.currentTimeMillis();
       System.out.println(readTotal + " bytes written in " + (endTime - startTime) + " ms.");
       pushSuccess = true;
     } else if (opcode == IConstants.NO_ACTION) {
       Logger.Log("UPLOAD NOT REQUIRED");
     }
     fileInputStream.close();
   } catch (Exception e) {
     Logger.Log(e.getMessage());
     e.printStackTrace();
   }
   return pushSuccess;
 }