package main

import (

    "fmt"

    "io"

    "io/ioutil"

    "log"

    "net/http"

    "os"

    "reflect"

)

const (

    UPLOAD_DIR = "./uploads"

)

func uploadHandler(w http.ResponseWriter, r *http.Request) {

    if r.Method == "GET" {

        io.WriteString(w, "<html><form method=\"POST\" action=\"/upload\" "+

            " enctype=\"multipart/form-data\">"+

            "Choose an image to upload: <input name=\"image\" type=\"file\" />"+

            "<input type=\"submit\" value=\"Upload\" />"+

            "</form></html>")

        return

    }

    if r.Method == "POST" {

        f, h, err := r.FormFile("image")

        if err != nil {

            http.Error(w, err.Error(), http.StatusInternalServerError)

            return

        }

        filename := h.Filename

        fmt.Println("filename:", filename)

        defer f.Close() //文件流句柄

        t, err := os.Create(UPLOAD_DIR + "/" + filename)

        fmt.Println("filepath:", UPLOAD_DIR+"/"+filename)

        if err != nil {

            http.Error(w, err.Error(), http.StatusInternalServerError)

            return

        }

        defer t.Close() //关闭临时文件句柄

        if _, err := io.Copy(t, f); err != nil {

            http.Error(w, err.Error(), http.StatusInternalServerError)

            return

        }

        http.Redirect(w, r, "/view?id="+filename, http.StatusFound)

    }

}

func viewHandler(w http.ResponseWriter, r *http.Request) {

    imageId := r.FormValue("id")

    imagePath := UPLOAD_DIR + "/" + imageId

    if exists := isExists(imagePath); !exists {

        http.NotFound(w, r)

        return

    }

    w.Header().Set("Content-Type", "image") //HTTP响应头输出格式

    http.ServeFile(w, r, imagePath)

}

func isExists(path string) bool {

    _, err := os.Stat(path)

    if err == nil {

        return true

    }

    return os.IsExist(err)

}

func listHandler(w http.ResponseWriter, r *http.Request) {

    fileInfoArr, err := ioutil.ReadDir("./uploads")

    if err != nil {

        http.Error(w, err.Error(), http.StatusInternalServerError)

        return

    }

    var listHtml string

    for _, fileInfo := range fileInfoArr {

        imgid := fileInfo.Name

        fmt.Println("type:", reflect.TypeOf(imgid))

        listHtml += "<li><a href=\"/view?id=" + imgid() + "\">" + imgid() + "</a></li>"

    }

    io.WriteString(w, "<html><body><ol>"+listHtml+"</ol></body></html>")

}

func init() { //main函数之前执行

    fmt.Println("init-----")

}

func safeHandler(fn http.HandlerFunc) http.HandlerFunc { //异常的处理

    return func(w http.ResponseWriter, r *http.Request) {

        defer func() {

            if e, ok := recover().(error); ok {

                http.Error(w, err.Error(), http.StatusInternalServerError)

                log.Println("WARN: panic in %v - %v", fn, e)

                log.Println(string(debug.Stack()))

            }

        }()

        fn(w, r)

    }

}

//处理静态资源

func staticDirHandler(mux *http.ServeMux, prefix string, staticDir string, flags int) {

    mux.HandleFunc(prefix, func(w http.ResponseWriter, r *http.Request) {

        file := staticDir + r.URL.Path[len(prefix)-:]

        if (flags & ListDir) ==  {

            if exists := isExists(file); !exists {

                http.NotFound(w, r)

                return

            }

        }

        http.ServeFile(w, r, file)

    })

}

func main() {

    mux := http.NewServeMux()

    staticDirHandler(mux, "/assets/", "./public", )

    http.HandleFunc("/", safeHandler(listHandler))

    http.HandleFunc("/view", safeHandler(viewHandler))

    http.HandleFunc("/upload", safeHandler(uploadHandler))

    err := http.ListenAndServe(":8080", nil)

    if err != nil {

        log.Fatal("ListenAndServe: ", err.Error())

    }

}

go 文件上传的更多相关文章

  1. jquery.uploadify文件上传组件

    1.jquery.uploadify简介 在ASP.NET中上传的控件有很多,比如.NET自带的FileUpload,以及SWFUpload,Uploadify等等,尤其后面两个控件的用户体验比较好, ...

  2. 11、Struts2 的文件上传和下载

    文件上传 表单准备 要想使用 HTML 表单上传一个或多个文件 须把 HTML 表单的 enctype 属性设置为 multipart/form-data 须把 HTML 表单的method 属性设置 ...

  3. Java FtpClient 实现文件上传服务

    一.Ubuntu 安装 Vsftpd 服务 1.安装 sudo apt-get install vsftpd 2.添加用户(uftp) sudo useradd -d /home/uftp -s /b ...

  4. 小兔Java教程 - 三分钟学会Java文件上传

    今天群里正好有人问起了Java文件上传的事情,本来这是Java里面的知识点,而我目前最主要的精力还是放在了JS的部分.不过反正也不麻烦,我就专门开一贴来聊聊Java文件上传的基本实现方法吧. 话不多说 ...

  5. ,net core mvc 文件上传

    工作用到文件上传的功能,在这个分享下 ~~ Controller: public class PictureController : Controller { private IHostingEnvi ...

  6. Web开发安全之文件上传安全

    很长一段时间像我这种菜鸡搞一个网站第一时间反应就是找上传,找上传.借此机会把文件上传的安全问题总结一下. 首先看一下DVWA给出的Impossible级别的完整代码: <?php if( iss ...

  7. AutoIt实现Webdriver自动化测试文件上传

    在运用WebDriver进行自动化测试时,由于WebDriver自身的限制,对于上传文件时Windows弹出的文件选择窗口无法控制,通过在网上查找资料锁定使用AutoIt来控制文件上传窗口. Auto ...

  8. Struts的文件上传下载

    Struts的文件上传下载 1.文件上传 Struts2的文件上传也是使用fileUpload的组件,这个组默认是集合在框架里面的.且是使用拦截器:<interceptor name=" ...

  9. .JavaWeb文件上传和FileUpload组件使用

    .JavaWeb文件上传 1.自定义上传 文件上传时的表单设计要符合文件提交的方式: 1.提交方式:post 2.表单中有文件上传的表单项:<input type="file" ...

  10. ASP.NET MVC5+EF6+EasyUI 后台管理系统（32）-swfupload多文件上传[附源码]

    系列目录 文件上传这东西说到底有时候很痛,原来的asp.net服务器控件提供了很简单的上传,但是有回传,还没有进度条提示.这次我们演示利用swfupload多文件上传,项目上文件上传是比不可少的,大家 ...

随机推荐

  1. php解析json字符串变量总是空白null

    通过接口获取的json字符串使用json_decode始终无法正确解析,返回空白. 直接把结果字符串复制出来手动创建一个变量却正常,在前端js也能解析,搞了半天不得其解,最后发现是接口输出的结果包含有 ...

  2. luogu P2746 [USACO5.3]校园网Network of Schools

    题目描述 一些学校连入一个电脑网络.那些学校已订立了协议:每个学校都会给其它的一些学校分发软件(称作“接受学校”).注意即使 B 在 A 学校的分发列表中, A 也不一定在 B 学校的列表中. 你要写 ...

  3. n*n的正方形网格中有多少个长方形

    n*n的正方形网格中有横竖各n+1条直线,其中,任意各取两条都可以组成一个长方形﹙正方形也是长方形﹚.所以长方形个数为C﹙n+2,2﹚×C﹙n+2,2﹚＝﹙n+1﹚²n²/4个.如果正方形不算,则N＝ ...

  4. DLL注入新姿势：反射式DLL注入研究

    在分析koadic渗透利器时,发现它有一个注入模块,其DLL注入实现方式和一般的注入方式不一样.搜索了一下发现是由HarmanySecurity的Stephen Fewer提出的ReflectiveD ...

  5. GDI+ ColorMatrix的完全揭秘

    无论是用何种语言,只要使用过Windows的GDI+的人对ColorMatrix都不陌生,我的BLOG文章中也多次提到过,并在<GDI+ for VCL基础 -- 颜色调整矩阵ColorMatr ...

  6. 电话号码 【trie树】

    电话号码 查看 提交 统计 提问 总时间限制: 1000ms 内存限制: 65536kB 描写叙述 给你一些电话号码,请推断它们是否是一致的,即是否有某个电话是还有一个电话的前缀. 比方: Emerg ...

  7. 赵雅智_Fragment

    当我们须要动态的多界面切换的时候,就须要将UI元素和Activity融合成一个模块. 在2.3中我们一般通过各种Activity中进行跳转来实现多界面的跳转和单个界面动态改变.在4.0或以上系统中就能 ...

  8. 设置安卓开机动画、开机logo

    我们要修改的是system>media文件夹下的bootanimation.zip(手机开机动画)这个文件 先来讲讲这个文件结构:该zip解压后得到两个文件, 第一个目录存放了开机时播放的图片( ...

  9. cocosStudio中使用PageView,ListView和ScrollView

    晚上吃东西好像吃坏肚子了,.但是技术还要继续研究.最近工作中要使用CocosStudio做界面,好吧,不管对他有什么偏见,学习一下吧.这里主要记录一下三个控件的使用和说明.就是ScrollView,L ...

  10. （利用DOM）在新打开的页面点击关闭当前浏览器窗口

    1.在开发过程中我们前端的用户体验中有时候会要求点击一个按钮,关闭当前浏览器窗口,用HTML DOM就可做到 2.注意:本次写法要求在新窗口中关闭.target="_blank" ...