add custom port

daniwebdev · daniwebdev · commit da168dc9c12f · 2023-12-17T16:17:01.000+07:00
diff --git a/README.md b/README.md
@@ -81,6 +81,14 @@ Example:
 curl "http://localhost:8080/compressor?url=https://example.com/image.jpg&output=webp&quality=80&resolution=1024x720"
 ```
 
+### Custom Port
+
+By default, the server listens on port `8080`. If you wish to use a custom port, you can specify it during the startup of the server using the `-p` flag. For example:
+
+```bash
+./image-compressor -o ./tmp -p 8888
+```
+
 ## License
 
 This project is licensed under the [MIT License](LICENSE).
diff --git a/image-compressor b/image-compressor
diff --git a/main.go b/main.go
@@ -19,10 +19,14 @@ import (
 	"github.com/nfnt/resize"
 )
 
-var outputDirectory string
+var (
+	outputDirectory string
+	port            int
+)
 
 func init() {
 	flag.StringVar(&outputDirectory, "o", ".", "Output directory for compressed images")
+	flag.IntVar(&port, "p", 8080, "Port for the server to listen on")
 	flag.Parse()
 }
 
@@ -250,6 +254,6 @@ func main() {
 
 	http.Handle("/", r)
 
-	fmt.Printf("Server is listening on :8080. Output directory: %s\n", outputDirectory)
-	http.ListenAndServe(":8080", nil)
+	fmt.Printf("Server is listening on :%d. Output directory: %s\n", port, outputDirectory)
+	http.ListenAndServe(fmt.Sprintf(":%d", port), nil)
 }

Original file line number	Diff line number	Diff line change
`@@ -19,10 +19,14 @@ import (`
`19`	`19`	`"github.com/nfnt/resize"`
`20`	`20`	`)`
`21`	`21`
`22`		`-var outputDirectory string`
	`22`	`+var (`
	`23`	`+ outputDirectory string`
	`24`	`+ port int`
	`25`	`+)`
`23`	`26`
`24`	`27`	`func init() {`
`25`	`28`	`flag.StringVar(&outputDirectory, "o", ".", "Output directory for compressed images")`
	`29`	`+ flag.IntVar(&port, "p", 8080, "Port for the server to listen on")`
`26`	`30`	`flag.Parse()`
`27`	`31`	`}`
`28`	`32`
`@@ -250,6 +254,6 @@ func main() {`
`250`	`254`
`251`	`255`	`http.Handle("/", r)`
`252`	`256`
`253`		`- fmt.Printf("Server is listening on :8080. Output directory: %s\n", outputDirectory)`
`254`		`- http.ListenAndServe(":8080", nil)`
	`257`	`+ fmt.Printf("Server is listening on :%d. Output directory: %s\n", port, outputDirectory)`
	`258`	`+ http.ListenAndServe(fmt.Sprintf(":%d", port), nil)`
`255`	`259`	`}`