├── config ├── upload.go ├── http.go ├── paths.go └── index.go ├── Makefile ├── main.go ├── config.yaml ├── http └── http.go ├── .gitignore ├── README.md └── LICENSE /config/upload.go: -------------------------------------------------------------------------------- 1 | package config 2 | 3 | import "github.com/axetroy/gin-uploader" 4 | 5 | var Upload uploader.TConfig 6 | 7 | func InitUpload() { 8 | Upload = Config.Upload 9 | } 10 | -------------------------------------------------------------------------------- /config/http.go: -------------------------------------------------------------------------------- 1 | package config 2 | 3 | type HttpConfig struct { 4 | Host string 5 | Port string 6 | } 7 | 8 | var Http HttpConfig 9 | 10 | func InitHttp() { 11 | Http = Config.Http 12 | } 13 | -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- 1 | # use the vendor/ subdir which holds the vendored apache thrift go library, version 2 | # the vendored thrift is commit fa0796d33208eadafb6f42964c8ef29d7751bfc2 on 1.0.0-dev, 3 | # last commit there is Fri Oct 16 21:33:39 2015 +0200, from https://github.com/apache/thrift 4 | all: 5 | go build -v -o ./bin/server 6 | 7 | build: 8 | make all 9 | echo "Build Success!" 10 | 11 | clean: 12 | rm -f ./bin/* -------------------------------------------------------------------------------- /config/paths.go: -------------------------------------------------------------------------------- 1 | package config 2 | 3 | import "os" 4 | 5 | type GlobalPaths struct { 6 | root string 7 | cwd string 8 | } 9 | 10 | var Paths GlobalPaths 11 | 12 | /** 13 | get current work dir 14 | */ 15 | func getCwd() (pwd string, err error) { 16 | pwd, err = os.Getwd() 17 | if err != nil { 18 | os.Exit(1) 19 | } 20 | return 21 | } 22 | 23 | func InitPaths() { 24 | var ( 25 | cwd, _ = getCwd() 26 | ) 27 | 28 | Paths = GlobalPaths{ 29 | cwd: cwd, 30 | } 31 | } 32 | -------------------------------------------------------------------------------- /main.go: -------------------------------------------------------------------------------- 1 | package main 2 | 3 | import ( 4 | "github.com/axetroy/go-upload/config" 5 | "github.com/axetroy/go-upload/http" 6 | "os" 7 | "os/signal" 8 | "syscall" 9 | "fmt" 10 | ) 11 | 12 | func main() { 13 | 14 | var ( 15 | err error 16 | ) 17 | 18 | sigc := make(chan os.Signal, 1) 19 | signal.Notify(sigc, 20 | syscall.SIGHUP, 21 | syscall.SIGINT, 22 | syscall.SIGTERM, 23 | syscall.SIGQUIT) 24 | go func() { 25 | s := <-sigc 26 | fmt.Print("signal: ", s) 27 | os.Exit(1) 28 | }() 29 | 30 | // init anything 31 | if err = config.Init(); err != nil { 32 | panic(err) 33 | return 34 | } 35 | 36 | http.RunServer() 37 | } 38 | -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- 1 | # config.yaml 2 | 3 | # 在GO中，所有属性绑定到结构体，都是小写 4 | 5 | # HTTP模块设置 6 | http: 7 | host: 0.0.0.0 # 监听地址 8 | port: 9044 # 监听端口 9 | 10 | # 上传模块的相关设置 11 | 12 | upload: 13 | urlprefix: /api/v1 14 | path: uploads # 文件上传的根目录 15 | # 普通文件上传 16 | file: 17 | path: files # 文件上传的目录 18 | maxsize: 10485760 # 上传文件的最大大小 19 | allowtype: # 允许上传的文件后缀名 20 | - .log 21 | - .txt 22 | - .text 23 | - .md 24 | # 图片上传 25 | image: 26 | path: image # 图片上传的目录 27 | maxsize: 10485760 # 上传图片的最大大小 28 | thumbnail: 29 | path: thumbnail # 缩略图存放的目录 30 | maxwidth: 300 # 缩略图最大宽度 31 | maxheight: 300 # 缩略图最大高度 32 | -------------------------------------------------------------------------------- /http/http.go: -------------------------------------------------------------------------------- 1 | package http 2 | 3 | import ( 4 | "github.com/gin-gonic/gin" 5 | "github.com/axetroy/go-upload/config" 6 | "github.com/axetroy/gin-uploader" 7 | "github.com/itsjamie/gin-cors" 8 | "time" 9 | ) 10 | 11 | var ( 12 | Router *gin.Engine 13 | ) 14 | 15 | func RunServer() (err error) { 16 | gin.SetMode(config.Config.Mode) 17 | Router = gin.Default() 18 | Router.Use(gin.Logger()) 19 | Router.Use(gin.Recovery()) 20 | 21 | Router.Use(cors.Middleware(cors.Config{ 22 | Origins: "*", 23 | Methods: "GET, PUT, POST, DELETE", 24 | RequestHeaders: "Origin, X-Requested-With, X_Requested_With, Content-Type, Accept, Authentication, Authorization, X-Server, X-Password-Pay", 25 | ExposedHeaders: "", 26 | MaxAge: 60 * time.Second, 27 | Credentials: true, // cookies 28 | ValidateHeaders: false, 29 | })) 30 | 31 | if loader, err := uploader.New(Router, config.Config.Upload); err != nil { 32 | return err 33 | } else { 34 | loader.Resolve() 35 | } 36 | 37 | return Router.Run(config.Http.Host + ":" + config.Http.Port) 38 | } 39 | -------------------------------------------------------------------------------- /config/index.go: -------------------------------------------------------------------------------- 1 | package config 2 | 3 | import ( 4 | "path/filepath" 5 | "io/ioutil" 6 | "gopkg.in/yaml.v2" 7 | "os" 8 | "github.com/gin-gonic/gin" 9 | "github.com/axetroy/gin-uploader" 10 | ) 11 | 12 | type ConfigType struct { 13 | Http HttpConfig 14 | Upload uploader.TConfig 15 | Env string 16 | Mode string 17 | } 18 | 19 | var Config ConfigType 20 | 21 | func Init() (err error) { 22 | 23 | var ( 24 | configfile string 25 | yamlFile []byte 26 | ) 27 | 28 | configfile, err = filepath.Abs("./config.yaml") 29 | 30 | if err != nil { 31 | return 32 | } 33 | 34 | yamlFile, err = ioutil.ReadFile(configfile) 35 | 36 | if err != nil { 37 | return 38 | } 39 | 40 | err = yaml.Unmarshal(yamlFile, &Config) 41 | 42 | if err != nil { 43 | return err 44 | } 45 | 46 | Config.Env = os.Getenv("GO_ENV") 47 | 48 | if Config.Env == gin.ReleaseMode || Config.Env == "production" || Config.Env == "publish" { 49 | Config.Mode = gin.ReleaseMode 50 | } else if Config.Env == gin.TestMode { 51 | Config.Mode = gin.TestMode 52 | } else { 53 | Config.Mode = gin.DebugMode 54 | } 55 | 56 | InitPaths() 57 | InitHttp() 58 | InitUpload() 59 | 60 | return 61 | } 62 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | # Created by .ignore support plugin (hsz.mobi) 2 | ### JetBrains template 3 | # Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm 4 | # Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839 5 | 6 | # User-specific stuff: 7 | .idea/**/workspace.xml 8 | .idea/**/tasks.xml 9 | .idea/dictionaries 10 | 11 | # Sensitive or high-churn files: 12 | .idea/**/dataSources/ 13 | .idea/**/dataSources.ids 14 | .idea/**/dataSources.xml 15 | .idea/**/dataSources.local.xml 16 | .idea/**/sqlDataSources.xml 17 | .idea/**/dynamic.xml 18 | .idea/**/uiDesigner.xml 19 | 20 | # Gradle: 21 | .idea/**/gradle.xml 22 | .idea/**/libraries 23 | 24 | # CMake 25 | cmake-build-debug/ 26 | 27 | # Mongo Explorer plugin: 28 | .idea/**/mongoSettings.xml 29 | 30 | ## File-based project format: 31 | *.iws 32 | 33 | ## Plugin-specific files: 34 | 35 | # IntelliJ 36 | out/ 37 | 38 | # mpeltonen/sbt-idea plugin 39 | .idea_modules/ 40 | 41 | # JIRA plugin 42 | atlassian-ide-plugin.xml 43 | 44 | # Cursive Clojure plugin 45 | .idea/replstate.xml 46 | 47 | # Crashlytics plugin (for Android Studio and IntelliJ) 48 | com_crashlytics_export_strings.xml 49 | crashlytics.properties 50 | crashlytics-build.properties 51 | fabric.properties 52 | ### Go template 53 | # Binaries for programs and plugins 54 | *.exe 55 | *.dll 56 | *.so 57 | *.dylib 58 | 59 | # Test binary, build with `go test -c` 60 | *.test 61 | 62 | # Output of the go coverage tool, specifically when used with LiteIDE 63 | *.out 64 | 65 | # Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736 66 | .glide/ 67 | 68 | # thrift 69 | vendor 70 | .idea 71 | bin/* 72 | uploads/* -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | ### go文件上传模块 2 | 3 | 基于[github.com/axetroy/gin-uploader](https://github.com/axetroy/gin-uploader) 4 | 5 | - [x] 文件上传 6 | - [x] 限制上传文件的后缀名 7 | - [x] 限制上传文件的大小 8 | - [x] Hash去重，防止重复上传 9 | - [x] 图片自动生成缩略图 10 | - [x] 全由配置 11 | 12 | ### 如何运行 13 | 14 | ```bash 15 | go get -v -u github.com/axetroy/go-upload 16 | cd $GOPATH/src/github.com/axetroy/go-upload 17 | make build 18 | ./bin/server 19 | # 或者以生产环境运行 20 | GO_ENV=production ./bin/server 21 | ``` 22 | 23 | ### API 24 | 25 | ```bash 26 | 27 | # 上传相关 28 | [POST] /upload/image # 图片上传 29 | [POST] /upload/file # 其他文件上传 30 | [GET] /upload/example # 上传demo，仅在开发模式下 31 | # 下载相关 32 | [GET] /download/image/origin/:filename # 获取上传的原始图片 33 | [GET] /download/image/thumbnail/:filename # 获取上传的缩略图片 34 | [GET] /download/file/download/:filename # 下载文件 35 | [GET]   /download/file/raw/:filename            # 获取文件 36 | ``` 37 | 38 | ### 配置文件 39 | 40 | 配置文件与二进制同一目录 41 | 42 | ```yaml 43 | # config.yaml 44 | 45 | # 在GO中，所有属性绑定到结构体，都是小写 46 | 47 | # HTTP模块设置 48 | http: 49 | host: localhost # 监听地址 50 | port: 9044 # 监听端口 51 | 52 | # 上传模块的相关设置 53 | 54 | upload: 55 | path: uploads # 文件上传的根目录 56 | # 普通文件上传 57 | file: 58 | path: files # 文件上传的目录 59 | maxsize: 10485760 # 上传文件的最大大小 60 | allowtype: # 允许上传的文件后缀名 61 | - .log 62 | - .txt 63 | - .text 64 | - .md 65 | # 图片上传 66 | image: 67 | path: image # 图片上传的目录 68 | maxsize: 10485760 # 上传图片的最大大小 69 | thumbnail: 70 | path: thumbnail # 缩略图存放的目录 71 | maxwidth: 300 # 缩略图最大宽度 72 | maxheight: 300 # 缩略图最大高度 73 | ``` 74 | 75 | 76 | ## License 77 | 78 | [![FOSSA Status](https://app.fossa.io/api/projects/git%2Bgithub.com%2Faxetroy%2Fgo-upload.svg?type=large)](https://app.fossa.io/projects/git%2Bgithub.com%2Faxetroy%2Fgo-upload?ref=badge_large) 79 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, 10 | and distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by 13 | the copyright owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all 16 | other entities that control, are controlled by, or are under common 17 | control with that entity. For the purposes of this definition, 18 | "control" means (i) the power, direct or indirect, to cause the 19 | direction or management of such entity, whether by contract or 20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the 21 | outstanding shares, or (iii) beneficial ownership of such entity. 22 | 23 | "You" (or "Your") shall mean an individual or Legal Entity 24 | exercising permissions granted by this License. 25 | 26 | "Source" form shall mean the preferred form for making modifications, 27 | including but not limited to software source code, documentation 28 | source, and configuration files. 29 | 30 | "Object" form shall mean any form resulting from mechanical 31 | transformation or translation of a Source form, including but 32 | not limited to compiled object code, generated documentation, 33 | and conversions to other media types. 34 | 35 | "Work" shall mean the work of authorship, whether in Source or 36 | Object form, made available under the License, as indicated by a 37 | copyright notice that is included in or attached to the work 38 | (an example is provided in the Appendix below). 39 | 40 | "Derivative Works" shall mean any work, whether in Source or Object 41 | form, that is based on (or derived from) the Work and for which the 42 | editorial revisions, annotations, elaborations, or other modifications 43 | represent, as a whole, an original work of authorship. For the purposes 44 | of this License, Derivative Works shall not include works that remain 45 | separable from, or merely link (or bind by name) to the interfaces of, 46 | the Work and Derivative Works thereof. 47 | 48 | "Contribution" shall mean any work of authorship, including 49 | the original version of the Work and any modifications or additions 50 | to that Work or Derivative Works thereof, that is intentionally 51 | submitted to Licensor for inclusion in the Work by the copyright owner 52 | or by an individual or Legal Entity authorized to submit on behalf of 53 | the copyright owner. For the purposes of this definition, "submitted" 54 | means any form of electronic, verbal, or written communication sent 55 | to the Licensor or its representatives, including but not limited to 56 | communication on electronic mailing lists, source code control systems, 57 | and issue tracking systems that are managed by, or on behalf of, the 58 | Licensor for the purpose of discussing and improving the Work, but 59 | excluding communication that is conspicuously marked or otherwise 60 | designated in writing by the copyright owner as "Not a Contribution." 61 | 62 | "Contributor" shall mean Licensor and any individual or Legal Entity 63 | on behalf of whom a Contribution has been received by Licensor and 64 | subsequently incorporated within the Work. 65 | 66 | 2. Grant of Copyright License. Subject to the terms and conditions of 67 | this License, each Contributor hereby grants to You a perpetual, 68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 69 | copyright license to reproduce, prepare Derivative Works of, 70 | publicly display, publicly perform, sublicense, and distribute the 71 | Work and such Derivative Works in Source or Object form. 72 | 73 | 3. Grant of Patent License. Subject to the terms and conditions of 74 | this License, each Contributor hereby grants to You a perpetual, 75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 76 | (except as stated in this section) patent license to make, have made, 77 | use, offer to sell, sell, import, and otherwise transfer the Work, 78 | where such license applies only to those patent claims licensable 79 | by such Contributor that are necessarily infringed by their 80 | Contribution(s) alone or by combination of their Contribution(s) 81 | with the Work to which such Contribution(s) was submitted. If You 82 | institute patent litigation against any entity (including a 83 | cross-claim or counterclaim in a lawsuit) alleging that the Work 84 | or a Contribution incorporated within the Work constitutes direct 85 | or contributory patent infringement, then any patent licenses 86 | granted to You under this License for that Work shall terminate 87 | as of the date such litigation is filed. 88 | 89 | 4. Redistribution. You may reproduce and distribute copies of the 90 | Work or Derivative Works thereof in any medium, with or without 91 | modifications, and in Source or Object form, provided that You 92 | meet the following conditions: 93 | 94 | (a) You must give any other recipients of the Work or 95 | Derivative Works a copy of this License; and 96 | 97 | (b) You must cause any modified files to carry prominent notices 98 | stating that You changed the files; and 99 | 100 | (c) You must retain, in the Source form of any Derivative Works 101 | that You distribute, all copyright, patent, trademark, and 102 | attribution notices from the Source form of the Work, 103 | excluding those notices that do not pertain to any part of 104 | the Derivative Works; and 105 | 106 | (d) If the Work includes a "NOTICE" text file as part of its 107 | distribution, then any Derivative Works that You distribute must 108 | include a readable copy of the attribution notices contained 109 | within such NOTICE file, excluding those notices that do not 110 | pertain to any part of the Derivative Works, in at least one 111 | of the following places: within a NOTICE text file distributed 112 | as part of the Derivative Works; within the Source form or 113 | documentation, if provided along with the Derivative Works; or, 114 | within a display generated by the Derivative Works, if and 115 | wherever such third-party notices normally appear. The contents 116 | of the NOTICE file are for informational purposes only and 117 | do not modify the License. You may add Your own attribution 118 | notices within Derivative Works that You distribute, alongside 119 | or as an addendum to the NOTICE text from the Work, provided 120 | that such additional attribution notices cannot be construed 121 | as modifying the License. 122 | 123 | You may add Your own copyright statement to Your modifications and 124 | may provide additional or different license terms and conditions 125 | for use, reproduction, or distribution of Your modifications, or 126 | for any such Derivative Works as a whole, provided Your use, 127 | reproduction, and distribution of the Work otherwise complies with 128 | the conditions stated in this License. 129 | 130 | 5. Submission of Contributions. Unless You explicitly state otherwise, 131 | any Contribution intentionally submitted for inclusion in the Work 132 | by You to the Licensor shall be under the terms and conditions of 133 | this License, without any additional terms or conditions. 134 | Notwithstanding the above, nothing herein shall supersede or modify 135 | the terms of any separate license agreement you may have executed 136 | with Licensor regarding such Contributions. 137 | 138 | 6. Trademarks. This License does not grant permission to use the trade 139 | names, trademarks, service marks, or product names of the Licensor, 140 | except as required for reasonable and customary use in describing the 141 | origin of the Work and reproducing the content of the NOTICE file. 142 | 143 | 7. Disclaimer of Warranty. Unless required by applicable law or 144 | agreed to in writing, Licensor provides the Work (and each 145 | Contributor provides its Contributions) on an "AS IS" BASIS, 146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 147 | implied, including, without limitation, any warranties or conditions 148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A 149 | PARTICULAR PURPOSE. You are solely responsible for determining the 150 | appropriateness of using or redistributing the Work and assume any 151 | risks associated with Your exercise of permissions under this License. 152 | 153 | 8. Limitation of Liability. In no event and under no legal theory, 154 | whether in tort (including negligence), contract, or otherwise, 155 | unless required by applicable law (such as deliberate and grossly 156 | negligent acts) or agreed to in writing, shall any Contributor be 157 | liable to You for damages, including any direct, indirect, special, 158 | incidental, or consequential damages of any character arising as a 159 | result of this License or out of the use or inability to use the 160 | Work (including but not limited to damages for loss of goodwill, 161 | work stoppage, computer failure or malfunction, or any and all 162 | other commercial damages or losses), even if such Contributor 163 | has been advised of the possibility of such damages. 164 | 165 | 9. Accepting Warranty or Additional Liability. While redistributing 166 | the Work or Derivative Works thereof, You may choose to offer, 167 | and charge a fee for, acceptance of support, warranty, indemnity, 168 | or other liability obligations and/or rights consistent with this 169 | License. However, in accepting such obligations, You may act only 170 | on Your own behalf and on Your sole responsibility, not on behalf 171 | of any other Contributor, and only if You agree to indemnify, 172 | defend, and hold each Contributor harmless for any liability 173 | incurred by, or claims asserted against, such Contributor by reason 174 | of your accepting any such warranty or additional liability. 175 | 176 | END OF TERMS AND CONDITIONS 177 | 178 | APPENDIX: How to apply the Apache License to your work. 179 | 180 | To apply the Apache License to your work, attach the following 181 | boilerplate notice, with the fields enclosed by brackets "{}" 182 | replaced with your own identifying information. (Don't include 183 | the brackets!) The text should be enclosed in the appropriate 184 | comment syntax for the file format. We also recommend that a 185 | file or class name and description of purpose be included on the 186 | same "printed page" as the copyright notice for easier 187 | identification within third-party archives. 188 | 189 | Copyright 2017 axetroy 190 | 191 | Licensed under the Apache License, Version 2.0 (the "License"); 192 | you may not use this file except in compliance with the License. 193 | You may obtain a copy of the License at 194 | 195 | http://www.apache.org/licenses/LICENSE-2.0 196 | 197 | Unless required by applicable law or agreed to in writing, software 198 | distributed under the License is distributed on an "AS IS" BASIS, 199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 200 | See the License for the specific language governing permissions and 201 | limitations under the License. --------------------------------------------------------------------------------