31 lines
547 B
Go
31 lines
547 B
Go
package utils
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
func GetPort() string {
|
|
if port := os.Getenv("API_PORT"); port != "" {
|
|
return port
|
|
}
|
|
|
|
return "8080"
|
|
}
|
|
|
|
func GetResponseCode(r *http.Request) int {
|
|
// If the `force-fail` header is set, get and return its value.
|
|
if failCode := r.Header.Get("force-fail"); failCode != "" {
|
|
log.Println("`force-fail` header set...")
|
|
|
|
if code, err := strconv.Atoi(failCode); err == nil {
|
|
log.Printf("Setting the response code to %d...", code)
|
|
|
|
return code
|
|
}
|
|
}
|
|
|
|
return http.StatusOK
|
|
}
|