bdl requests to pass internationalized information (#3362)

Author: kakj-go

conf/core-services/i18n/i18n.yaml

@@ -14,6 +14,9 @@ zh:
  Mem: ' 内存 '
  CPUQuotaIsLessThanRequest: '调整后的%s环境的 CPU 配额（%.3f 核）不得小于当前使用量（%.3f 核），请重新设置'
  MemQuotaIsLessThanRequest: '调整后的%s环境的内存配额（%.3f G）不得小于当前使用量（%.3f G），请重新设置'
+ DeleteProjectErrorApplicationExist: '删除项目失败(项目中还存在应用)'
+ FailedGetProject: "获取项目失败"
+ FailedDeleteProject: "删除项目失败"
 en:
  AvailableIsLessThanQuota: The actual available resource on this workspace in the cluster is less than the quota. Please ask the Ops to allocate project resources reasonably
  NoResourceForTheWorkspace: No allocatable resources on this workspace in the cluster, please check the workspace labels for the nodes
@@ -30,3 +33,6 @@ en:
  Mem: ' Mem '
  CPUQuotaIsLessThanRequest: "The env %s's new CPU quota (%.3f Core) cannot be less than the actual request (%.3f Core), please reset it"
  MemQuotaIsLessThanRequest: "The env %s's new Mem quota (%.3f G) cannot be less than the actual request (%.3f G), please reset it"
+ DeleteProjectErrorApplicationExist: 'failed to delete project(there exists applications)'
+ FailedGetProject: "failed to get project"
+ FailedDeleteProject: "failed to delete project"

go.mod

@@ -52,6 +52,7 @@ require (
 	github.com/ghodss/yaml v1.0.1-0.20190212211648-25d852aebe32
 	github.com/gin-gonic/gin v1.7.0
 	github.com/go-echarts/go-echarts/v2 v2.2.4
+	github.com/go-eden/routine v0.0.2
 	github.com/go-openapi/loads v0.19.4
 	github.com/go-openapi/spec v0.19.8
 	github.com/go-openapi/strfmt v0.19.5

go.sum

@@ -569,8 +569,9 @@ github.com/go-bindata/go-bindata/v3 v3.1.3/go.mod h1:1/zrpXsLD8YDIbhZRqXzm1Ghc7N
 github.com/go-chi/chi v4.0.2+incompatible/go.mod h1:eB3wogJHnLi3x/kFX2A+IbTBlXxmMeXJVKy9tTv1XzQ=
 github.com/go-echarts/go-echarts/v2 v2.2.4 h1:SKJpdyNIyD65XjbUZjzg6SwccTNXEgmh+PlaO23g2H0=
 github.com/go-echarts/go-echarts/v2 v2.2.4/go.mod h1:6TOomEztzGDVDkOSCFBq3ed7xOYfbOqhaBzD0YV771A=
-github.com/go-eden/routine v0.0.1 h1:EFzGtHcxmv2gAcdoG7vMESe+26nuIl0/vTMtc030m/4=
 github.com/go-eden/routine v0.0.1/go.mod h1:R+sklWRKa0GPYXW5Opyr1nSb9KbwpEF9AXBARb7r1XI=
+github.com/go-eden/routine v0.0.2 h1:jcRSRZEKX0R1OqYqfbBRXxpaDxANUJn2m7liPe7KzQI=
+github.com/go-eden/routine v0.0.2/go.mod h1:R+sklWRKa0GPYXW5Opyr1nSb9KbwpEF9AXBARb7r1XI=
 github.com/go-errors/errors v1.0.1 h1:LUHzmkK3GUKUrL/1gfBUxAHzcev3apQlezX/+O7ma6w=
 github.com/go-errors/errors v1.0.1/go.mod h1:f4zRHt4oKfwPJE5k8C9vpYG+aDHdBFUsgrm6/TyX73Q=
 github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=

modules/core-services/endpoints/project.go

@@ -245,7 +245,7 @@ func (e *Endpoints) DeleteProject(ctx context.Context, r *http.Request, vars map
 	orgIDStr := r.Header.Get(httputil.OrgHeader)
 	orgID, err := strutil.Atoi64(orgIDStr)
 	if err != nil {
-		return apierrors.ErrUpdateProject.InvalidParameter(err).ToResp(), nil
+		return apierrors.ErrDeleteProject.InvalidParameter(err).ToResp(), nil
 	}
 
 	// 获取当前用户

modules/core-services/services/project/project.go

@@ -37,6 +37,7 @@ import (
 	"github.com/erda-project/erda/modules/core-services/types"
 	"github.com/erda-project/erda/pkg/crypto/uuid"
 	"github.com/erda-project/erda/pkg/filehelper"
+	local "github.com/erda-project/erda/pkg/i18n"
 	"github.com/erda-project/erda/pkg/numeral"
 	calcu "github.com/erda-project/erda/pkg/resourcecalculator"
 	"github.com/erda-project/erda/pkg/ucauth"
@@ -637,14 +638,15 @@ func (p *Project) DeleteWithEvent(projectID int64) error {
 
 // Delete 删除项目
 func (p *Project) Delete(projectID int64) (*model.Project, error) {
+	langCodes, _ := i18n.ParseLanguageCode(local.GetGoroutineBindLang())
 	// check if application exists
 	if count, err := p.db.GetApplicationCountByProjectID(projectID); err != nil || count > 0 {
-		return nil, errors.Errorf("failed to delete project(there exists applications)")
+		return nil, errors.Errorf(p.trans.Text(langCodes, "DeleteProjectErrorApplicationExist"))
 	}
 
 	project, err := p.db.GetProjectByID(projectID)
 	if err != nil {
-		return nil, errors.Errorf("failed to get project, (%v)", err)
+		return nil, errors.Errorf(p.trans.Text(langCodes, "FailedGetProject")+"(%v)", err)
 	}
 
 	// TODO We need to turn this check on after adding the delete portal to the UI
@@ -658,7 +660,7 @@ func (p *Project) Delete(projectID int64) (*model.Project, error) {
 	// }
 
 	if err = p.db.DeleteProject(projectID); err != nil {
-		return nil, errors.Errorf("failed to delete project, (%v)", err)
+		return nil, errors.Errorf(p.trans.Text(langCodes, "FailedDeleteProject")+"(%v)", err)
 	}
 	_ = p.db.DeleteProjectQutoa(projectID)
 	logrus.Infof("deleted project %d", projectID)

pkg/goroutine_context/context.go

@@ -0,0 +1,77 @@
+// Copyright (c) 2021 Terminus, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package goroutine_context
+
+import (
+	"context"
+	"sync"
+
+	"github.com/go-eden/routine"
+)
+
+const LocaleNameContextKey = "locale_name_context_key"
+
+const bucketsSize = 128
+
+type (
+	contextBucket struct {
+		lock sync.RWMutex
+		data map[int64]context.Context
+	}
+	contextBuckets struct {
+		buckets [bucketsSize]*contextBucket
+	}
+)
+
+var goroutineContext contextBuckets
+
+func init() {
+	for i := range goroutineContext.buckets {
+		goroutineContext.buckets[i] = &contextBucket{
+			data: make(map[int64]context.Context),
+		}
+	}
+}
+
+// GetContext .
+func GetContext() context.Context {
+	goid := routine.Goid()
+	idx := goid % bucketsSize
+	bucket := goroutineContext.buckets[idx]
+	bucket.lock.RLock()
+	ctx := bucket.data[goid]
+	bucket.lock.RUnlock()
+	return ctx
+}
+
+// SetContext .
+func SetContext(ctx context.Context) {
+	goid := routine.Goid()
+	idx := goid % bucketsSize
+	bucket := goroutineContext.buckets[idx]
+	bucket.lock.Lock()
+	defer bucket.lock.Unlock()
+	bucket.data[goid] = ctx
+}
+
+// ClearContext .
+func ClearContext() {
+	goid := routine.Goid()
+	idx := goid % bucketsSize
+	bucket := goroutineContext.buckets[idx]
+	bucket.lock.Lock()
+	defer bucket.lock.Unlock()
+	delete(bucket.data, goid)
+}

pkg/http/httpclient/request.go

@@ -32,6 +32,7 @@ import (
 	"github.com/pkg/errors"
 
 	"github.com/erda-project/erda/pkg/http/customhttp"
+	"github.com/erda-project/erda/pkg/i18n"
 	"github.com/erda-project/erda/pkg/terminal/loading"
 )
 
@@ -105,6 +106,13 @@ func (r *Request) Do() AfterDo {
 	for k, v := range r.header {
 		req.Header.Set(k, v)
 	}
+
+	// get goroutine context il8n header to bdl request
+	localeName := i18n.GetGoroutineBindLang()
+	if localeName != "" {
+		req.Header.Set(i18n.LangHeader, localeName)
+	}
+
 	for _, v := range r.cookie {
 		req.AddCookie(v)
 	}

pkg/http/httpserver/server.go

@@ -32,6 +32,7 @@ import (
 
 	"github.com/erda-project/erda/apistructs"
 	"github.com/erda-project/erda/pkg/crypto/uuid"
+	"github.com/erda-project/erda/pkg/goroutine_context"
 	"github.com/erda-project/erda/pkg/http/httpserver/ierror"
 	"github.com/erda-project/erda/pkg/i18n"
 	"github.com/erda-project/erda/pkg/strutil"
@@ -148,6 +149,11 @@ func (s *Server) internal(handler func(context.Context, *http.Request, map[strin
 			locale = s.localeLoader.Locale(localeName)
 		}
 
+		// set global context bind goroutine id
+		i18n.SetGoroutineBindLang(localeName)
+		// clear all global context
+		defer goroutine_context.ClearContext()
+
 		// Manual decoding url var
 		muxVars := mux.Vars(r)
 		for k, v := range muxVars {

pkg/i18n/helper.go

@@ -14,21 +14,54 @@
 
 package i18n
 
-import "net/http"
+import (
+	"context"
+	"net/http"
+
+	"github.com/erda-project/erda/pkg/goroutine_context"
+)
 
 const ZH = "zh-CN"
 const EN = "en-US"
 
+const LangHeader = "Lang"
+
 // GetLocaleNameByRequest 从request获取语言名称
 func GetLocaleNameByRequest(request *http.Request) string {
 	// 优先querystring 其次header
 	lang := request.URL.Query().Get("lang")
 	if lang != "" {
 		return lang
 	}
-	lang = request.Header.Get("Lang")
+	lang = request.Header.Get(LangHeader)
 	if lang != "" {
 		return lang
 	}
 	return ""
 }
+
+func GetGoroutineBindLang() string {
+	globalContext := goroutine_context.GetContext()
+	if globalContext == nil {
+		return ""
+	}
+	key := globalContext.Value(goroutine_context.LocaleNameContextKey)
+	if key == nil {
+		return ""
+	}
+	localeName, ok := key.(string)
+	if !ok {
+		return ""
+	}
+
+	return localeName
+}
+
+func SetGoroutineBindLang(localeName string) {
+	ctx := goroutine_context.GetContext()
+	if ctx == nil {
+		ctx = context.Background()
+	}
+
+	goroutine_context.SetContext(context.WithValue(ctx, goroutine_context.LocaleNameContextKey, localeName))
+}

pkg/i18n/helper_test.go

@@ -0,0 +1,65 @@
+// Copyright (c) 2021 Terminus, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package i18n
+
+import "testing"
+
+func TestSetGoroutineBindLang(t *testing.T) {
+	type args struct {
+		localeName string
+	}
+	tests := []struct {
+		name string
+		args args
+	}{
+		{
+			name: "test",
+			args: args{
+				localeName: "test",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+		})
+	}
+}
+
+func TestGetGoroutineBindLang(t *testing.T) {
+	tests := []struct {
+		name string
+		want string
+	}{
+		{
+			name: "test",
+			want: "test",
+		},
+		{
+			name: "test",
+			want: "",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.want != "" {
+				SetGoroutineBindLang(tt.want)
+			}
+
+			if got := GetGoroutineBindLang(); got != tt.want {
+				t.Errorf("GetGoroutineBindLang() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}