typed register

Author: ryantxu

pkg/github/query_handler.go

@@ -42,29 +42,32 @@ func UnmarshalQuery(b []byte, v interface{}) *backend.DataResponse {
 // GetQueryHandlers creates the QueryTypeMux type for handling queries
 func GetQueryHandlers(s *QueryHandler) *datasource.QueryTypeMux {
 	mux := datasource.NewQueryTypeMux()
+	register := func(qt models.QueryType, handler backend.QueryDataHandlerFunc) {
+		mux.HandleFunc(string(qt), handler)
+	}
 
-	mux.HandleFunc(string(models.QueryTypeCommits), s.HandleCommits)
-	mux.HandleFunc(string(models.QueryTypeIssues), s.HandleIssues)
-	mux.HandleFunc(string(models.QueryTypeContributors), s.HandleContributors)
-	mux.HandleFunc(string(models.QueryTypeLabels), s.HandleLabels)
-	mux.HandleFunc(string(models.QueryTypePullRequests), s.HandlePullRequests)
-	mux.HandleFunc(string(models.QueryTypePullRequestReviews), s.HandlePullRequestReviews)
-	mux.HandleFunc(string(models.QueryTypeReleases), s.HandleReleases)
-	mux.HandleFunc(string(models.QueryTypeTags), s.HandleTags)
-	mux.HandleFunc(string(models.QueryTypePackages), s.HandlePackages)
-	mux.HandleFunc(string(models.QueryTypeMilestones), s.HandleMilestones)
-	mux.HandleFunc(string(models.QueryTypeRepositories), s.HandleRepositories)
-	mux.HandleFunc(string(models.QueryTypeVulnerabilities), s.HandleVulnerabilities)
-	mux.HandleFunc(string(models.QueryTypeProjects), s.HandleProjects)
-	mux.HandleFunc(string(models.QueryTypeStargazers), s.HandleStargazers)
-	mux.HandleFunc(string(models.QueryTypeWorkflows), s.HandleWorkflows)
-	mux.HandleFunc(string(models.QueryTypeWorkflowUsage), s.HandleWorkflowUsage)
-	mux.HandleFunc(string(models.QueryTypeWorkflowRuns), s.HandleWorkflowRuns)
-	mux.HandleFunc(string(models.QueryTypeCodeScanning), s.HandleCodeScanning)
-	mux.HandleFunc(string(models.QueryTypeDeployments), s.HandleDeployments)
-	mux.HandleFunc(string(models.QueryTypeOrganizations), s.HandleOrganizations)
-	mux.HandleFunc(string(models.QueryTypeCommitFiles), s.HandleCommitFiles)
-	mux.HandleFunc(string(models.QueryTypePullRequestFiles), s.HandlePullRequestFiles)
+	register(models.QueryTypeCommits, s.HandleCommits)
+	register(models.QueryTypeIssues, s.HandleIssues)
+	register(models.QueryTypeContributors, s.HandleContributors)
+	register(models.QueryTypeLabels, s.HandleLabels)
+	register(models.QueryTypePullRequests, s.HandlePullRequests)
+	register(models.QueryTypePullRequestReviews, s.HandlePullRequestReviews)
+	register(models.QueryTypeReleases, s.HandleReleases)
+	register(models.QueryTypeTags, s.HandleTags)
+	register(models.QueryTypePackages, s.HandlePackages)
+	register(models.QueryTypeMilestones, s.HandleMilestones)
+	register(models.QueryTypeRepositories, s.HandleRepositories)
+	register(models.QueryTypeVulnerabilities, s.HandleVulnerabilities)
+	register(models.QueryTypeProjects, s.HandleProjects)
+	register(models.QueryTypeStargazers, s.HandleStargazers)
+	register(models.QueryTypeWorkflows, s.HandleWorkflows)
+	register(models.QueryTypeWorkflowUsage, s.HandleWorkflowUsage)
+	register(models.QueryTypeWorkflowRuns, s.HandleWorkflowRuns)
+	register(models.QueryTypeCodeScanning, s.HandleCodeScanning)
+	register(models.QueryTypeDeployments, s.HandleDeployments)
+	register(models.QueryTypeOrganizations, s.HandleOrganizations)
+	register(models.QueryTypeCommitFiles, s.HandleCommitFiles)
+	register(models.QueryTypePullRequestFiles, s.HandlePullRequestFiles)
 
 	return mux
 }