diff --git a/internal/model/query/query.go b/internal/model/query/query.go index d4c06b0..625f279 100644 --- a/internal/model/query/query.go +++ b/internal/model/query/query.go @@ -114,7 +114,8 @@ type ModelQuery struct { } type RepositoryReq struct { - Id int64 `json:"id"` + Id int64 `json:"id"` + Token string `json:"token"` } type TagQuery struct { diff --git a/internal/service/repository_service.go b/internal/service/repository_service.go index d8e96b2..103c7c2 100644 --- a/internal/service/repository_service.go +++ b/internal/service/repository_service.go @@ -257,6 +257,14 @@ func (s *RepositoryService) MountRepository(repoReq *query.RepositoryReq) error if err != nil { return err } + + authHeaders := make(map[string]string) + if repoReq.Token != "" { + authHeaders["Authorization"] = fmt.Sprintf("Bearer %s", repoReq.Token) + } else { + authHeaders = s.hfTokenDao.GetHeaders() + } + _, err = util.PostForDomain(speedDomain, "/api/cacheJob/create", "application/json", b, s.hfTokenDao.GetHeaders()) if err != nil { return err