build: upgrade mirrormapping for concurrency

move graphql client to utils module

Author: BoxBoxJason

graphql/main.go

@@ -1,32 +1,72 @@
+/*
+Utility types definitions
+*/
 package utils
 
 import (
+	"bytes"
+	"context"
 	"encoding/json"
 	"fmt"
+	"net/http"
 	"os"
+	"strings"
+	"sync"
 )
 
+// ProjectMirrorOptions defines how the project should be mirrored
+// to the destination GitLab instance
+// - destination_url: the URL of the destination GitLab instance
+// - ci_cd_catalog: whether to add the project to the CI/CD catalog
+// - issues: whether to mirror the issues
 type ProjectMirroringOptions struct {
 	DestinationURL string `json:"destination_url"`
 	CI_CD_Catalog  bool   `json:"ci_cd_catalog"`
 	Issues         bool   `json:"issues"`
 }
 
+// GroupMirrorOptions defines how the group should be mirrored
+// to the destination GitLab instance
+// - destination_url: the URL of the destination GitLab instance
+// - ci_cd_catalog: whether to add the group to the CI/CD catalog
+// - issues: whether to mirror the issues
 type GroupMirroringOptions struct {
 	DestinationURL string `json:"destination_url"`
 	CI_CD_Catalog  bool   `json:"ci_cd_catalog"`
 	Issues         bool   `json:"issues"`
 }
 
+// MirrorMapping defines the mapping of projects and groups
+// to the destination GitLab instance
+// It is used to parse the JSON file that contains the mapping
+// - projects: a map of project names to their mirroring options
+// - groups: a map of group names to their mirroring options
 type MirrorMapping struct {
-	Projects map[string]ProjectMirroringOptions `json:"projects"`
-	Groups   map[string]GroupMirroringOptions   `json:"groups"`
+	Projects   map[string]*ProjectMirroringOptions `json:"projects"`
+	Groups     map[string]*GroupMirroringOptions   `json:"groups"`
+	muProjects sync.Mutex
+	muGroups   sync.Mutex
 }
 
+func (m *MirrorMapping) AddProject(project string, options *ProjectMirroringOptions) {
+	m.muProjects.Lock()
+	defer m.muProjects.Unlock()
+	m.Projects[project] = options
+}
+
+func (m *MirrorMapping) AddGroup(group string, options *GroupMirroringOptions) {
+	m.muGroups.Lock()
+	defer m.muGroups.Unlock()
+	m.Groups[group] = options
+}
+
+// OpenMirrorMapping opens the JSON file that contains the mapping
+// and parses it into a MirrorMapping struct
+// It returns the mapping and an error if any
 func OpenMirrorMapping(path string) (*MirrorMapping, error) {
 	mapping := &MirrorMapping{
-		Projects: make(map[string]ProjectMirroringOptions),
-		Groups:   make(map[string]GroupMirroringOptions),
+		Projects: make(map[string]*ProjectMirroringOptions),
+		Groups:   make(map[string]*GroupMirroringOptions),
 	}
 
 	// Read the file
@@ -80,3 +120,59 @@ func (m *MirrorMapping) check() error {
 
 	return err
 }
+
+// GraphQLClient is a client for sending GraphQL requests to GitLab
+type GraphQLClient struct {
+	token string
+	URL   string
+}
+
+// GraphQLRequest is a struct that represents a GraphQL request
+// It contains the query and the variables
+type GraphQLRequest struct {
+	Query     string `json:"query"`
+	Variables string `json:"variables,omitempty"`
+}
+
+// NewGitlabGraphQLClient creates a new GraphQL client for GitLab
+// It takes the token and the GitLab URL as arguments
+// It returns a pointer to the GraphQLClient struct
+func NewGitlabGraphQLClient(token, gitlabUrl string) *GraphQLClient {
+	return &GraphQLClient{
+		token: token,
+		URL:   strings.TrimSuffix(gitlabUrl, "/") + "/api/graphql",
+	}
+}
+
+// SendRequest sends a GraphQL request to GitLab
+// It takes a GraphQLRequest struct and the HTTP method as arguments
+// It returns the response body as a string and an error if any
+func (g *GraphQLClient) SendRequest(request *GraphQLRequest, method string) (string, error) {
+	requestBody, err := json.Marshal(request)
+	if err != nil {
+		return "", err
+	}
+	LogVerbosef("Sending GraphQL request to %s with body: %s", g.URL, string(requestBody))
+	req, err := http.NewRequestWithContext(context.Background(), method, g.URL, bytes.NewBuffer(requestBody))
+	if err != nil {
+		return "", err
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+g.token)
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("GraphQL request failed with status: %s", resp.Status)
+	}
+	var responseBody bytes.Buffer
+	if _, err := responseBody.ReadFrom(resp.Body); err != nil {
+		return "", err
+	}
+	return responseBody.String(), nil
+}