mirror of
https://github.com/openfaas/faas.git
synced 2025-06-09 08:46:48 +00:00
updated hubstats to accept repo for pull count
Signed-off-by: Burton Rheutan <rheutan7@gmail.com>
This commit is contained in:
parent
d1fb0084b4
commit
26c8671865
@ -10,18 +10,22 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type dockerHubStatsType struct {
|
type dockerHubOrgStatsType struct {
|
||||||
Count int `json:"count"`
|
Count int `json:"count"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type dockerHubRepoStatsType struct {
|
||||||
|
PullCount int `json:"pull_count"`
|
||||||
|
}
|
||||||
|
|
||||||
func sanitizeInput(input string) string {
|
func sanitizeInput(input string) string {
|
||||||
parts := strings.Split(input, "\n")
|
parts := strings.Split(input, "\n")
|
||||||
return strings.Trim(parts[0], " ")
|
return strings.Trim(parts[0], " ")
|
||||||
}
|
}
|
||||||
|
|
||||||
func requestHubStats(org string) dockerHubStatsType {
|
func requestStats(repo string) []byte {
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
res, err := client.Get("https://hub.docker.com/v2/repositories/" + org)
|
res, err := client.Get("https://hub.docker.com/v2/repositories/" + repo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalln("Unable to reach Docker Hub server.")
|
log.Fatalln("Unable to reach Docker Hub server.")
|
||||||
}
|
}
|
||||||
@ -31,9 +35,19 @@ func requestHubStats(org string) dockerHubStatsType {
|
|||||||
log.Fatalln("Unable to parse response from server.")
|
log.Fatalln("Unable to parse response from server.")
|
||||||
}
|
}
|
||||||
|
|
||||||
dockerHubStats := dockerHubStatsType{}
|
return body
|
||||||
json.Unmarshal(body, &dockerHubStats)
|
}
|
||||||
return dockerHubStats
|
|
||||||
|
func parseOrgStats(response []byte) dockerHubOrgStatsType {
|
||||||
|
dockerHubOrgStats := dockerHubOrgStatsType{}
|
||||||
|
json.Unmarshal(response, &dockerHubOrgStats)
|
||||||
|
return dockerHubOrgStats
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseRepoStats(response []byte) dockerHubRepoStatsType {
|
||||||
|
dockerHubRepoStats := dockerHubRepoStatsType{}
|
||||||
|
json.Unmarshal(response, &dockerHubRepoStats)
|
||||||
|
return dockerHubRepoStats
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
@ -41,13 +55,19 @@ func main() {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("Unable to read standard input:", err)
|
log.Fatal("Unable to read standard input:", err)
|
||||||
}
|
}
|
||||||
org := string(input)
|
request := string(input)
|
||||||
if len(input) == 0 {
|
if len(input) == 0 {
|
||||||
log.Fatalln("A username or organisation is required.")
|
log.Fatalln("A username/organisation or repository is required.")
|
||||||
}
|
}
|
||||||
|
|
||||||
org = sanitizeInput(org)
|
request = sanitizeInput(request)
|
||||||
dockerHubStats := requestHubStats(org)
|
response := requestStats(request)
|
||||||
|
|
||||||
fmt.Printf("The organisation or user %s has %d repositories on the Docker hub.\n", org, dockerHubStats.Count)
|
if strings.Contains(request, "/") {
|
||||||
|
dockerHubRepoStats := parseRepoStats(response)
|
||||||
|
fmt.Printf("Repo: %s has been pulled %d times from the Docker Hub", request, dockerHubRepoStats.PullCount)
|
||||||
|
} else {
|
||||||
|
dockerHubOrgStats := parseOrgStats(response)
|
||||||
|
fmt.Printf("The organisation or user %s has %d repositories on the Docker hub.\n", request, dockerHubOrgStats.Count)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user