base-gin/utils/base/env.go

package base

import "os"

// env list
const (
	ENV_DEV     = "dev"
	ENV_TEST    = "test"
	ENV_PRODUCT = "product"

	LOCATION_SAAS     = "saas"
	LOCATION_LOCAL    = "local"
	LOCATION_FANGZHEN = "fangzhen"
)

// GetSrgEnv for API
func GetJHYEnv() (env string) {
	env = os.Getenv("JHY_ENV")
	if env != "" {
		return env
	}
	return ENV_DEV
}

// GetSrgLocation for API
func GetJHYLocation() (location string) {
	location = os.Getenv("JHY_LOCATION")
	if location != "" {
		return location
	}
	return LOCATION_LOCAL
}

// IsCliENV for API
func IsCliENV() bool {
	if GetJHYEnv() != ENV_PRODUCT {
		return true
	}

	cron := os.Getenv("JHY_CLI")
	return cron == "true"
}

// IsPprofEnv for API
func IsPprofEnv() bool {
	if GetJHYEnv() == ENV_DEV {
		return true
	}
	if os.Getenv("JHY_PPROF") == "true" {
		return true
	}
	return false
}

// GetServiceName for API
func GetServiceName() (env string) {
	env = os.Getenv("JHY_SERVICE")
	if env != "" {
		return env
	}
	return "unknown"
}

// GetHostName for API
func GetHostName() string {
	hostname, err := os.Hostname()
	// not has hostname? how
	if err != nil {
		// todo set machine id
		return ""
	}
	return hostname
}