草庐IT

go - 如何附加文件(io.Reader)？

coder 2023-06-28 原文 

func SimpleUploader(r *http.Request, w http.ResponseWriter) {
    // temp folder path
    chunkDirPath := "./creatives/.uploads/" + userUUID
    // create folder
    err = os.MkdirAll(chunkDirPath, 02750)

    // Get file handle from multipart request
    var file io.Reader
    mr, err := r.MultipartReader()

    var fileName string
    // Read multipart body until the "file" part
    for {
        part, err := mr.NextPart()
        if err == io.EOF {
            break
        }
        if part.FormName() == "file" {
            file = part
            fileName = part.FileName()
            fmt.Println(fileName)
            break
        }
    }

    // Create files
    tempFile := chunkDirPath + "/" + fileName
    dst, err := os.Create(tempFile)

    defer dst.Close()

    buf := make([]byte, 1024*1024)
    file.Read(buf)
    // write/save buffer to disk
    ioutil.WriteFile(tempFile, buf, os.ModeAppend)
    if http.DetectContentType(buf) != "video/mp4" {
        response, _ := json.Marshal(&Response{"File upload cancelled"})
        settings.WriteResponse(w, http.StatusInternalServerError, response)
        return
    }

    // joinedFile := io.MultiReader(bytes.NewReader(buf), file)
    _, err = io.Copy(dst, file)
    if err != nil {
        settings.LogError(err, methodName, "Error copying file")
    }

    response, _ := json.Marshal(&Response{"File uploaded successfully"})
    settings.WriteResponse(w, http.StatusInternalServerError, response)
}

我正在上传视频文件。 在上传整个文件之前，我想做一些检查，所以我将前 1mb 保存到一个文件中:

buf := make([]byte, 1024*1024)
file.Read(buf)
// write/save buffer to disk
ioutil.WriteFile(tempFile, buf, os.ModeAppend)

然后，如果检查通过，我想上传文件的其余部分 dst 是用于保存第一个 1 mb 的同一个文件 所以基本上我试图附加到文件:

_, err = io.Copy(dst, file)

上传的文件大小正确，但文件已损坏(无法播放视频)。

我还尝试了什么？ : 加入两个读者并保存到一个新文件。但是使用这种方法，文件大小会增加 1 MB 并且已损坏。

joinedFile := io.MultiReader(bytes.NewReader(buf), file)
_, err = io.Copy(newDst, joinedFile)

请帮忙。

最佳答案

您基本上通过执行 os.Create 和 ioutil.WriteFile 打开文件两次

问题在于 os.Create 的返回值 (dst) 就像指向该文件开头的指针。 WriteFile 不会移动 dst 指向的位置。

您基本上是在执行 WriteFile，然后在 WriteFile 写入的第一组字节之上执行 io.Copy。

尝试先执行 WriteFile(带有 Create 标志)，然后 os.OpenFile(而不是 os.Create)带有 Append 标志的同一文件将剩余字节追加到末尾。

另外，允许客户端给你文件名是非常危险的，因为它可能是 ../../.bashrc(例如)，你会用用户决定上传的任何内容覆盖你的 shell init .

如果您自己计算文件名会更安全，如果您需要记住用户选择的文件名，请将其存储在您的数据库中，甚至是稍后加载的 metadata.json 类型的文件中。

关于go - 如何附加文件(io.Reader)？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36895714/

ReadergofilesectionWriteFile

有关go - 如何附加文件(io.Reader)？的更多相关文章

  1. ruby - 如何使用 Nokogiri 的 xpath 和 at_xpath 方法 - 2

    我正在学习如何使用Nokogiri，根据这段代码我遇到了一些问题:require'rubygems'require'mechanize'post_agent=WWW::Mechanize.newpost_page=post_agent.get('http://www.vbulletin.org/forum/showthread.php?t=230708')puts"\nabsolutepathwithtbodygivesnil"putspost_page.parser.xpath('/html/body/div/div/div/div/div/table/tbody/tr/td/div

  2. ruby - 如何从 ruby​​ 中的字符串运行任意对象方法？ - 2

    总的来说，我对ruby​​还比较陌生，我正在为我正在创建的对象编写一些rspec测试用例。许多测试用例都非常基础，我只是想确保正确填充和返回值。我想知道是否有办法使用循环结构来执行此操作。不必为我要测试的每个方法都设置一个assertEquals。例如:describeitem,"TestingtheItem"doit"willhaveanullvaluetostart"doitem=Item.new#HereIcoulddotheitem.name.shouldbe_nil#thenIcoulddoitem.category.shouldbe_nilendend但我想要一些方法来使用

  3. ruby - 使用 RubyZip 生成 ZIP 文件时设置压缩级别 - 2

    我有一个Ruby程序，它使用rubyzip压缩XML文件的目录树。gem。我的问题是文件开始变得很重，我想提高压缩级别，因为压缩时间不是问题。我在rubyzipdocumentation中找不到一种为创建的ZIP文件指定压缩级别的方法。有人知道如何更改此设置吗？是否有另一个允许指定压缩级别的Ruby库？ 最佳答案 这是我通过查看ruby​​zip内部创建的代码。level=Zlib::BEST_COMPRESSIONZip::ZipOutputStream.open(zip_file)do|zip|Dir.glob("**/*")d

  4. ruby - 其他文件中的 Rake 任务 - 2

    我试图在一个项目中使用rake，如果我把所有东西都放到Rakefile中，它会很大并且很难读取/找到东西，所以我试着将每个命名空间放在lib/rake中它自己的文件中，我添加了这个到我的rake文件的顶部:Dir['#{File.dirname(__FILE__)}/lib/rake/*.rake'].map{|f|requiref}它加载文件没问题，但没有任务。我现在只有一个.rake文件作为测试，名为“servers.rake”，它看起来像这样:namespace:serverdotask:testdoputs"test"endend所以当我运行rakeserver:testid时

  5. ruby-on-rails - 在 Rails 中将文件大小字符串转换为等效千字节 - 2

    我的目标是转换表单输入，例如“100兆字节”或“1GB”，并将其转换为我可以存储在数据库中的文件大小(以千字节为单位)。目前，我有这个:defquota_convert@regex=/([0-9]+)(.*)s/@sizes=%w{kilobytemegabytegigabyte}m=self.quota.match(@regex)if@sizes.include?m[2]eval("self.quota=#{m[1]}.#{m[2]}")endend这有效，但前提是输入是倍数(“gigabytes”，而不是“gigabyte”)并且由于使用了eval看起来疯狂不安全。所以，功能正常，

  6. python - 如何使用 Ruby 或 Python 创建一系列高音调和低音调的蜂鸣声？ - 2

    关闭。这个问题是opinion-based.它目前不接受答案。想要改进这个问题？更新问题，以便editingthispost可以用事实和引用来回答它.关闭4年前。Improvethisquestion我想在固定时间创建一系列低音和高音调的哔哔声。例如:在150毫秒时发出高音调的蜂鸣声在151毫秒时发出低音调的蜂鸣声200毫秒时发出低音调的蜂鸣声250毫秒的高音调蜂鸣声有没有办法在Ruby或Python中做到这一点？我真的不在乎输出编码是什么(.wav、.mp3、.ogg等等)，但我确实想创建一个输出文件。

  7. ruby-on-rails - Rails 3 中的多个路由文件 - 2

    Rails2.3可以选择随时使用RouteSet#add_configuration_file添加更多路由。是否可以在Rails3项目中做同样的事情？ 最佳答案 在config/application.rb中:config.paths.config.routes在Rails3.2(也可能是Rails3.1)中，使用:config.paths["config/routes"] 关于ruby-on-rails-Rails3中的多个路由文件，我们在StackOverflow上找到一个类似的问题

  8. ruby-on-rails - 如何验证 update_all 是否实际在 Rails 中更新 - 2

    给定这段代码defcreate@upgrades=User.update_all(["role=?","upgraded"],:id=>params[:upgrade])redirect_toadmin_upgrades_path,:notice=>"Successfullyupgradeduser."end我如何在该操作中实际验证它们是否已保存或未重定向到适当的页面和消息？ 最佳答案 在Rails3中，update_all不返回任何有意义的信息，除了已更新的记录数(这可能取决于您的DBMS是否返回该信息)。http://ar.ru

  9. ruby-on-rails - 'compass watch' 是如何工作的/它是如何与 rails 一起使用的 - 2

    我在我的项目目录中完成了compasscreate.和compassinitrails。几个问题:我已将我的.sass文件放在public/stylesheets中。这是放置它们的正确位置吗？当我运行compasswatch时，它不会自动编译这些.sass文件。我必须手动指定文件:compasswatchpublic/stylesheets/myfile.sass等。如何让它自动运行？文件ie.css、print.css和screen.css已放在stylesheets/compiled。如何在编译后不让它们重新出现的情况下删除它们？我自己编译的.sass文件编译成compiled/t

  10. ruby - 将差异补丁应用于字符串/文件 - 2

    对于具有离线功能的智能手机应用程序，我正在为Xml文件创建单向文本同步。我希望我的服务器将增量/差异(例如GNU差异补丁)发送到目标设备。这是计划:Time=0Server:hasversion_1ofXmlfile(~800kiB)Client:hasversion_1ofXmlfile(~800kiB)Time=1Server:hasversion_1andversion_2ofXmlfile(each~800kiB)computesdeltaoftheseversions(=patch)(~10kiB)sendspatchtoClient(~10kiBtransferred)Cl

随机推荐