devops-test-app/app/main.go

73 lines
1.4 KiB
Go
Raw Normal View History

2020-07-07 14:12:13 +05:00
package main
import (
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"github.com/logrusorgru/aurora"
)
type appConfig struct {
Port string
ImagePath string
}
var config appConfig
func init() {
config.Port = os.Getenv("PORT")
if config.Port == "" {
config.Port = "3000"
}
config.ImagePath = os.Getenv("IMAGE_PATH")
if config.ImagePath == "" {
config.ImagePath = "tmp-image"
}
}
func uploadFile(w http.ResponseWriter, r *http.Request) {
file, _, err := r.FormFile("image")
if err != nil {
log.Fatalf("Error Retrieving the File %v", err)
}
defer file.Close()
fileBytes, err := ioutil.ReadAll(file)
if err != nil {
log.Fatalf("Error reading file %v", err)
}
ioutil.WriteFile(config.ImagePath, fileBytes, os.FileMode(0600))
log.Println(aurora.Cyan("Successfully Uploaded File"))
fmt.Fprintf(w, "Successfully Uploaded File\n")
}
func image(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, config.ImagePath)
}
func livenessProbe(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "ok")
}
func readinessProbe(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "ok")
}
func main() {
http.HandleFunc("/upload", uploadFile)
http.HandleFunc("/image", image)
http.HandleFunc("/healthz/liveness", livenessProbe)
http.HandleFunc("/healthz/readiness", readinessProbe)
err := http.ListenAndServe(fmt.Sprintf(":%s", config.Port), nil)
if err != nil {
log.Fatal(err)
}
}