How to name file when saveAsTextFile in spark?

后端 未结 3 1410
别跟我提以往
别跟我提以往 2021-02-20 17:45

When saving as a textfile in spark version 1.5.1 I use: rdd.saveAsTextFile(\'\').

But if I want to find the file in that direcotry, how d

相关标签:
3条回答
  • 2021-02-20 18:31

    The correct answer to this question is that saveAsTextFile does not allow you to name the actual file.

    The reason for this is that the data is partitioned and within the path given as a parameter to the call to saveAsTextFile(...), it will treat that as a directory and then write one file per partition.

    You can call rdd.coalesce(1).saveAsTextFile('/some/path/somewhere') and it will create /some/path/somewhere/part-0000.txt.

    If you need more control than this, you will need to do an actual file operation on your end after you do a rdd.collect().

    Notice, this will pull all data into one executor so you may run into memory issues. That's the risk you take.

    0 讨论(0)
  • 2021-02-20 18:38

    It's not possible to name the file as @nod said. However, it's possible to rename the file right afterward. An example using PySpark:

    sc._jsc.hadoopConfiguration().set(
        "mapred.output.committer.class",
        "org.apache.hadoop.mapred.FileOutputCommitter")
    URI = sc._gateway.jvm.java.net.URI
    Path = sc._gateway.jvm.org.apache.hadoop.fs.Path
    FileSystem = sc._gateway.jvm.org.apache.hadoop.fs.FileSystem
    fs = FileSystem.get(URI("s3://{bucket_name}"), sc._jsc.hadoopConfiguration())
    file_path = "s3://{bucket_name}/processed/source={source_name}/year={partition_year}/week={partition_week}/"
    # remove data already stored if necessary
    fs.delete(Path(file_path))
    
    df.saveAsTextFile(file_path, compressionCodecClass="org.apache.hadoop.io.compress.GzipCodec")
    
    # rename created file
    created_file_path = fs.globStatus(Path(file_path + "part*.gz"))[0].getPath()
    fs.rename(
        created_file_path,
        Path(file_path + "{desired_name}.jl.gz"))
    
    0 讨论(0)
  • 2021-02-20 18:47

    As I said in my comment above, the documentation with examples can be found here. And quoting the description of the method saveAsTextFile:

    Save this RDD as a text file, using string representations of elements.

    In the following example I save a simple RDD into a file, then I load it and print its content.

    samples = sc.parallelize([
        ("abonsanto@fakemail.com", "Alberto", "Bonsanto"),
        ("mbonsanto@fakemail.com", "Miguel", "Bonsanto"),
        ("stranger@fakemail.com", "Stranger", "Weirdo"),
        ("dbonsanto@fakemail.com", "Dakota", "Bonsanto")
    ])
    
    print samples.collect()
    
    samples.saveAsTextFile("folder/here.txt")
    read_rdd = sc.textFile("folder/here.txt")
    
    read_rdd.collect()
    

    The output will be

    ('abonsanto@fakemail.com', 'Alberto', 'Bonsanto')
    ('mbonsanto@fakemail.com', 'Miguel', 'Bonsanto')
    ('stranger@fakemail.com', 'Stranger', 'Weirdo')
    ('dbonsanto@fakemail.com', 'Dakota', 'Bonsanto')
    
    [u"('abonsanto@fakemail.com', 'Alberto', 'Bonsanto')",
     u"('mbonsanto@fakemail.com', 'Miguel', 'Bonsanto')",
     u"('stranger@fakemail.com', 'Stranger', 'Weirdo')",
     u"('dbonsanto@fakemail.com', 'Dakota', 'Bonsanto')"]
    

    Let's take a look using a Unix-based terminal.

    usr@host:~/folder/here.txt$ cat *
    ('abonsanto@fakemail.com', 'Alberto', 'Bonsanto')
    ('mbonsanto@fakemail.com', 'Miguel', 'Bonsanto')
    ('stranger@fakemail.com', 'Stranger', 'Weirdo')
    ('dbonsanto@fakemail.com', 'Dakota', 'Bonsanto')
    
    0 讨论(0)
提交回复
热议问题