golang copy

司空儿 · · 690 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

package main

import (

    "fmt"

    "io"

    "os"

    "path/filepath"

    "strconv"

)

var BUFFERSIZE int64

func Copy1(src, dst string, BUFFERSIZE int64) error {

    sourceFileStat, err := os.Stat(src)

    if err != nil {

        return err

    }

    if !sourceFileStat.Mode().IsRegular() {

        return fmt.Errorf("%s is not a regular file", src)

    }

    source, err := os.Open(src)

    if err != nil {

        return err

    }

    defer source.Close()

    _, err = os.Stat(dst)

    if err == nil {

        return fmt.Errorf("file %s already exists", dst)

    }

    destination, err := os.Create(dst)

    if err != nil {

        return err

    }

    defer destination.Close()

    if err != nil {

        panic(err)

    }

    buf := make([]byte, BUFFERSIZE)

    for {

        n, err := source.Read(buf)

        if err != nil && err != io.EOF {

            return err

        }

        if n == 0 {

            break

        }

        if _, err := destination.Write(buf[:n]); err != nil {

            return err

        }

    }

    return err

}

func Copy(src, dst string) (int64, error) {

    sourceFileStat, err := os.Stat(src)

    if err != nil {

        return 0, err

    }

    if !sourceFileStat.Mode().IsRegular() {

        return 0, fmt.Errorf("%s is not a regular file", src)

    }

    source, err := os.Open(src)

    if err != nil {

            return 0, err

    }

    defer source.Close()

    destination, err := os.Create(dst)

    if err != nil {

        return 0, err

    }

    defer destination.Close()

    nBytes, err := io.Copy(destination, source)

    return nBytes, err

}

func main() {

    if len(os.Args) != 4 {

        fmt.Printf("usage: %s source destination BUFFERSIZE\n",

            filepath.Base(os.Args[0]))

        os.Exit(1)

    }

    source := os.Args[1]

    destination := os.Args[2]

    BUFFERSIZE, _ = strconv.ParseInt(os.Args[3], 10, 64)

    fmt.Printf("Copying %s to %s\n", source, destination)

    err := Copy1(source, destination, BUFFERSIZE)

    if err != nil {

        fmt.Printf("File copying failed: %q\n", err)

    }

}


作者:holdtom
链接:https://www.imooc.com/article/271446
来源:慕课网

有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:司空儿

查看原文:golang copy

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:701969077

690 次点击  
加入收藏 微博
下一篇:Golang中的陷阱
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传