hdfs_write.c
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:2k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. #include "hdfs.h" 
  19. int main(int argc, char **argv) {
  20.     if (argc != 4) {
  21.         fprintf(stderr, "Usage: hdfs_write <filename> <filesize> <buffersize>n");
  22.         exit(-1);
  23.     }
  24.     
  25.     hdfsFS fs = hdfsConnect("default", 0);
  26.     if (!fs) {
  27.         fprintf(stderr, "Oops! Failed to connect to hdfs!n");
  28.         exit(-1);
  29.     } 
  30.  
  31.     const char* writeFileName = argv[1];
  32.     tSize fileTotalSize = strtoul(argv[2], NULL, 10);
  33.     tSize bufferSize = strtoul(argv[3], NULL, 10);
  34.    
  35.     hdfsFile writeFile = hdfsOpenFile(fs, writeFileName, O_WRONLY, bufferSize, 0, 0);
  36.     if (!writeFile) {
  37.         fprintf(stderr, "Failed to open %s for writing!n", writeFileName);
  38.         exit(-2);
  39.     }
  40.     // data to be written to the file
  41.     char* buffer = malloc(sizeof(char) * bufferSize);
  42.     if(buffer == NULL) {
  43.         return -2;
  44.     }
  45.     int i = 0;
  46.     for (i=0; i < bufferSize; ++i) {
  47.         buffer[i] = 'a' + (i%26);
  48.     }
  49.     
  50.     // write to the file
  51.     tSize nrRemaining;
  52.     for (nrRemaining = fileTotalSize; nrRemaining > 0; nrRemaining -= bufferSize ) {
  53.         int curSize = ( bufferSize < nrRemaining ) ? bufferSize : (int)nrRemaining; 
  54.         hdfsWrite(fs, writeFile, (void*)buffer, curSize); 
  55.     }
  56.     free(buffer);
  57.     hdfsCloseFile(fs, writeFile);
  58.     hdfsDisconnect(fs);
  59.     return 0;
  60. }
  61. /**
  62.  * vim: ts=4: sw=4: et:
  63.  */