fix cli param, exit on file not found

This commit is contained in:
Benny~ 2020-02-23 21:20:10 +08:00
parent a302eeddc0
commit 5ae671f942
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481

View File

@ -27,6 +27,8 @@ type Config struct {
AllowedTypes []string `json:"ALLOWED_TYPES"` AllowedTypes []string `json:"ALLOWED_TYPES"`
} }
var configPath string
func webpEncoder(p1, p2 string, quality float32) { func webpEncoder(p1, p2 string, quality float32) {
var buf bytes.Buffer var buf bytes.Buffer
var img image.Image var img image.Image
@ -47,18 +49,19 @@ func webpEncoder(p1, p2 string, quality float32) {
fmt.Println("Save output.webp ok") fmt.Println("Save output.webp ok")
} }
func init() {
// Config Here
flag.StringVar(&configPath, "config", "config.json", "/path/to/config.json. (Default: ./config.json)")
flag.Parse()
//flag.PrintDefaults()
}
func main() { func main() {
app := fiber.New() app := fiber.New()
app.Banner = false app.Banner = false
app.Server = "WebP Server Go" app.Server = "WebP Server Go"
// Config Here config := load_config(configPath)
var config_path string
flag.StringVar(&config_path, "config", "config.json", "/path/to/config.json. (Default: ./config.json)")
flag.Parse()
flag.PrintDefaults()
config := load_config(config_path)
HOST := config.HOST HOST := config.HOST
PORT := config.PORT PORT := config.PORT
@ -161,7 +164,7 @@ func load_config(path string) Config {
var config Config var config Config
jsonObject, err := os.Open(path) jsonObject, err := os.Open(path)
if err != nil { if err != nil {
fmt.Println(err.Error()) log.Fatal(err)
} }
defer jsonObject.Close() defer jsonObject.Close()
decoder := json.NewDecoder(jsonObject) decoder := json.NewDecoder(jsonObject)