fix:修复nacos协议兼容中的问题

apiserver/nacosserver/model/constant.go

@@ -145,5 +145,9 @@ func ToNacosConfigNamespace(ns string) string {
 }
 
 const (
-	ActionGetConfigFile = "NACOS_GET_CONFIG"
+	ActionGetConfigFile         = "NACOS_GET_CONFIG"
+	ActionPublishConfigFile     = "NACOS_PUBLISH_CONFIG"
+	ActionGrpcGetConfigFile     = "NACOS_GRPC_GET_CONFIG"
+	ActionGrpcPublishConfigFile = "NACOS_GRPC_PUBLISH_CONFIG"
+	ActionGrpcPushConfigFile    = "NACOS_GRPC_PUSH_CONFIG"
 )

apiserver/nacosserver/v1/config/config_file.go

@@ -40,7 +40,13 @@ import (
 )
 
 func (n *ConfigServer) handlePublishConfig(ctx context.Context, req *model.ConfigFile) (bool, error) {
-	resp := n.configSvr.UpsertAndReleaseConfigFileFromClient(ctx, req.ToSpecConfigFile())
+	var resp *config_manage.ConfigResponse
+	if req.CasMd5 != "" {
+		resp = n.configSvr.CasUpsertAndReleaseConfigFileFromClient(ctx, req.ToSpecConfigFile())
+	} else {
+		resp = n.configSvr.UpsertAndReleaseConfigFileFromClient(ctx, req.ToSpecConfigFile())
+	}
+
 	if resp.GetCode().GetValue() == uint32(apimodel.Code_ExecuteSuccess) {
 		return true, nil
 	}

apiserver/nacosserver/v1/discover/instance.go

@@ -25,9 +25,11 @@ import (
 
 	apimodel "github.com/polarismesh/specification/source/go/api/v1/model"
 	apiservice "github.com/polarismesh/specification/source/go/api/v1/service_manage"
+	"google.golang.org/protobuf/types/known/wrapperspb"
 
 	"github.com/polarismesh/polaris/apiserver/nacosserver/core"
 	"github.com/polarismesh/polaris/apiserver/nacosserver/model"
+	commonmodel "github.com/polarismesh/polaris/common/model"
 	"github.com/polarismesh/polaris/common/utils"
 )
 
@@ -45,6 +47,24 @@ func (n *DiscoverServer) handleRegister(ctx context.Context, namespace, serviceN
 
 func (n *DiscoverServer) handleUpdate(ctx context.Context, namespace, serviceName string, ins *model.Instance) error {
 	specIns := model.PrepareSpecInstance(namespace, serviceName, ins)
+	if specIns.Id == nil || specIns.GetId().GetValue() == "" {
+		insId, errRsp := utils.CheckInstanceTetrad(specIns)
+		if errRsp != nil {
+			return &model.NacosError{
+				ErrCode: int32(model.ExceptionCode_ServerError),
+				ErrMsg:  errRsp.GetInfo().GetValue(),
+			}
+		}
+		specIns.Id = wrapperspb.String(insId)
+	}
+	saveIns, err := n.discoverSvr.Cache().GetStore().GetInstance(specIns.GetId().GetValue())
+	if err != nil {
+		return &model.NacosError{
+			ErrCode: int32(model.ExceptionCode_ServerError),
+			ErrMsg:  err.Error(),
+		}
+	}
+	specIns = mergeUpdateInstanceInfo(specIns, saveIns)
 	resp := n.discoverSvr.UpdateInstance(ctx, specIns)
 	if apimodel.Code(resp.GetCode().GetValue()) != apimodel.Code_ExecuteSuccess {
 		return &model.NacosError{
@@ -148,3 +168,7 @@ func (n *DiscoverServer) handleQueryInstances(ctx context.Context, params map[st
 	result.Namespace = model.ToNacosNamespace(namespace)
 	return result, nil
 }
+
+func mergeUpdateInstanceInfo(req *apiservice.Instance, saveVal *commonmodel.Instance) *apiservice.Instance {
+	return req
+}

apiserver/nacosserver/v2/access.go

@@ -54,6 +54,7 @@ var (
 )
 
 func (h *NacosV2Server) Request(ctx context.Context, payload *nacospb.Payload) (*nacospb.Payload, error) {
+	ctx = h.ConvertContext(ctx)
 	h.connectionManager.RefreshClient(ctx)
 	ctx = injectPayloadHeader(ctx, payload)
 	handle, val, err := h.UnmarshalPayload(payload)
@@ -64,20 +65,8 @@ func (h *NacosV2Server) Request(ctx context.Context, payload *nacospb.Payload) (
 	if !ok {
 		return nil, ErrorInvalidRequestBodyType
 	}
-
-	if _, ok := debugLevel[msg.GetRequestType()]; !ok {
-		nacoslog.Info("[NACOS-V2] handler client request", zap.String("conn-id", remote.ValueConnID(ctx)),
-			utils.ZapRequestID(msg.GetRequestId()),
-			zap.String("type", msg.GetRequestType()),
-		)
-	} else {
-		if nacoslog.DebugEnabled() {
-			nacoslog.Debug("[NACOS-V2] handler client request", zap.String("conn-id", remote.ValueConnID(ctx)),
-				utils.ZapRequestID(msg.GetRequestId()),
-				zap.String("type", msg.GetRequestType()),
-			)
-		}
-	}
+	nacoslog.Debug("[NACOS-V2] handler client request", zap.String("conn-id", remote.ValueConnID(ctx)),
+		utils.ZapRequestID(msg.GetRequestId()), zap.String("type", msg.GetRequestType()))
 	connMeta := remote.ValueConnMeta(ctx)
 
 	startTime := time.Now()

apiserver/nacosserver/v2/config/config_file.go

@@ -48,7 +48,25 @@ func (h *ConfigServer) handlePublishConfigRequest(ctx context.Context, req nacos
 		return nil, remote.ErrorInvalidRequestBodyType
 	}
 
-	resp := h.configSvr.CasUpsertAndReleaseConfigFileFromClient(ctx, configReq.ToSpec())
+	var resp *config_manage.ConfigResponse
+	startTime := commontime.CurrentMillisecond()
+	defer func() {
+		plugin.GetStatis().ReportDiscoverCall(metrics.ClientDiscoverMetric{
+			Action:    nacosmodel.ActionGrpcPublishConfigFile,
+			ClientIP:  meta.ConnectionID,
+			Namespace: configReq.Tenant,
+			Resource:  metrics.ResourceOfConfigFile(configReq.Group, configReq.DataId),
+			Timestamp: startTime,
+			CostTime:  commontime.CurrentMillisecond() - startTime,
+			Success:   resp.GetCode().GetValue() == uint32(apimodel.Code_ExecuteSuccess),
+		})
+	}()
+
+	if configReq.CasMd5 != "" {
+		resp = h.configSvr.CasUpsertAndReleaseConfigFileFromClient(ctx, configReq.ToSpec())
+	} else {
+		resp = h.configSvr.UpsertAndReleaseConfigFileFromClient(ctx, configReq.ToSpec())
+	}
 	if resp.GetCode().GetValue() != uint32(apimodel.Code_ExecuteSuccess) {
 		nacoslog.Error("[NACOS-V2][Config] publish config file fail", zap.String("tenant", configReq.Tenant),
 			utils.ZapGroup(configReq.Group), utils.ZapFileName(configReq.DataId),
@@ -83,8 +101,8 @@ func (h *ConfigServer) handleGetConfigRequest(ctx context.Context, req nacospb.B
 	startTime := commontime.CurrentMillisecond()
 	defer func() {
 		plugin.GetStatis().ReportDiscoverCall(metrics.ClientDiscoverMetric{
-			Action:    nacosmodel.ActionGetConfigFile,
-			ClientIP:  utils.ParseClientAddress(ctx),
+			Action:    nacosmodel.ActionGrpcGetConfigFile,
+			ClientIP:  meta.ConnectionID,
 			Namespace: configReq.Tenant,
 			Resource:  metrics.ResourceOfConfigFile(configReq.Group, configReq.DataId),
 			Timestamp: startTime,

apiserver/nacosserver/v2/config/watch.go

@@ -28,9 +28,12 @@ import (
 	nacospb "github.com/polarismesh/polaris/apiserver/nacosserver/v2/pb"
 	"github.com/polarismesh/polaris/apiserver/nacosserver/v2/remote"
 	"github.com/polarismesh/polaris/common/eventhub"
+	"github.com/polarismesh/polaris/common/metrics"
 	"github.com/polarismesh/polaris/common/model"
+	commontime "github.com/polarismesh/polaris/common/time"
 	"github.com/polarismesh/polaris/common/utils"
 	"github.com/polarismesh/polaris/config"
+	"github.com/polarismesh/polaris/plugin"
 )
 
 type ConnectionClientManager struct {
@@ -99,6 +102,9 @@ func (c *StreamWatchContext) ClientID() string {
 
 // ShouldNotify .
 func (c *StreamWatchContext) ShouldNotify(event *model.SimpleConfigFileRelease) bool {
+	if event.ReleaseType == model.ReleaseTypeGray && !c.betaMatcher(c.ClientLabels(), event) {
+		return false
+	}
 	key := event.FileKey()
 	watchFile, ok := c.watchConfigFiles.Load(key)
 	if !ok {
@@ -142,6 +148,21 @@ func (c *StreamWatchContext) Reply(event *apiconfig.ConfigClientResponse) {
 	notifyRequest.Group = viewConfig.GetGroup().GetValue()
 	notifyRequest.DataId = viewConfig.GetFileName().GetValue()
 
+	success := false
+	startTime := commontime.CurrentMillisecond()
+	defer func() {
+		plugin.GetStatis().ReportDiscoverCall(metrics.ClientDiscoverMetric{
+			Action:    nacosmodel.ActionGrpcPushConfigFile,
+			ClientIP:  c.ClientID(),
+			Namespace: notifyRequest.Tenant,
+			Resource:  metrics.ResourceOfConfigFile(notifyRequest.Group, notifyRequest.DataId),
+			Timestamp: startTime,
+			CostTime:  commontime.CurrentMillisecond() - startTime,
+			Revision:  viewConfig.GetMd5().GetValue(),
+			Success:   success,
+		})
+	}()
+
 	remoteClient, ok := c.connMgr.GetClient(c.clientId)
 	if !ok {
 		nacoslog.Error("[NACOS-V2][Config][Push] send ConfigChangeNotifyRequest not found remoteClient",
@@ -164,4 +185,5 @@ func (c *StreamWatchContext) Reply(event *apiconfig.ConfigClientResponse) {
 		nacoslog.Error("[NACOS-V2][Config][Push] send ConfigChangeNotifyRequest fail",
 			zap.String("clientId", c.ClientID()), zap.Error(err))
 	}
+	success = true
 }

auth/testexport.go

@@ -35,3 +35,8 @@ func TestInitialize(ctx context.Context, authOpt *Config, storage store.Store,
 	strategyMgn = strategySvr
 	return userSvr, strategySvr, nil
 }
+
+func TestClean() {
+	userMgn = nil
+	strategyMgn = nil
+}

cache/api/types.go

@@ -382,7 +382,7 @@ type (
 	// RateLimitCache rateLimit的cache接口
 	RateLimitCache interface {
 		Cache
-		// GetRateLimit 根据serviceID进行迭代回调
+		// IteratorRateLimit 遍历所有的限流规则
 		IteratorRateLimit(rateLimitIterProc RateLimitIterProc)
 		// GetRateLimitRules 根据serviceID获取限流数据
 		GetRateLimitRules(serviceKey model.ServiceKey) ([]*model.RateLimit, string)

cache/config/config_file.go

@@ -181,10 +181,6 @@ func (fc *fileCache) setReleases(releases []*model.ConfigFileRelease) (map[strin
 		}
 
 		if item.Active {
-			configLog.Info("[Config][Release][Cache] notify config release change",
-				zap.String("namespace", item.Namespace), zap.String("group", item.Group), zap.String("release", item.Name),
-				zap.String("file", item.FileName), zap.Uint64("version", item.Version), zap.Bool("valid", item.Valid),
-				zap.String("type", string(item.ReleaseType)))
 			fc.sendEvent(item)
 		}
 	}

common/metrics/types.go

@@ -141,8 +141,12 @@ type ClientDiscoverMetric struct {
 }
 
 func (c ClientDiscoverMetric) String() string {
-	return fmt.Sprintf("%s|%s|%s|%s|%s|%s|%d|%+v", c.ClientIP, c.Action, c.Namespace, c.Resource,
-		c.Revision, time.Unix(c.Timestamp, 0).Format("2006-01-02 15:04:05"), c.CostTime, c.Success)
+	revision := c.Revision
+	if revision == "" {
+		revision = "-"
+	}
+	return fmt.Sprintf("%s|%s|%s|%s|%s|%s|%dms|%+v", c.ClientIP, c.Action, c.Namespace, c.Resource,
+		revision, time.Unix(c.Timestamp/1000, 0).Format(time.DateTime), c.CostTime, c.Success)
 }
 
 type ConfigMetricType string

config/config_file.go

@@ -58,6 +58,7 @@ func (s *Server) CreateConfigFile(ctx context.Context, req *apiconfig.ConfigFile
 		log.Error("[Config][File] create config file commit tx.", utils.RequestID(ctx), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
+	s.RecordHistory(ctx, configFileRecordEntry(ctx, req, model.OCreate))
 	resp.ConfigFile = req
 	return resp
 }
@@ -88,7 +89,6 @@ func (s *Server) handleCreateConfigFile(ctx context.Context, tx store.Tx,
 			utils.ZapFileName(req.GetName().GetValue()), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
-	s.RecordHistory(ctx, configFileRecordEntry(ctx, req, model.OCreate))
 	return api.NewConfigResponse(apimodel.Code_ExecuteSuccess)
 }
 

config/config_file_release.go

@@ -136,6 +136,7 @@ func (s *Server) handlePublishConfigFile(ctx context.Context, tx store.Tx,
 		req.Name = utils.NewStringValue(fmt.Sprintf("%s-%d-%d", fileName, time.Now().Unix(), s.nextSequence()))
 	}
 
+	fileRelease.Name = req.GetName().GetValue()
 	fileRelease.Format = toPublishFile.Format
 	fileRelease.Metadata = toPublishFile.Metadata
 	fileRelease.Comment = req.GetComment().GetValue()
@@ -591,17 +592,25 @@ func (s *Server) CasUpsertAndReleaseConfigFile(ctx context.Context,
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
 
+	historyRecords := []func(){}
+
 	var upsertResp *apiconfig.ConfigResponse
 	if saveFile == nil {
 		if req.GetMd5().GetValue() != "" {
 			return api.NewConfigResponse(apimodel.Code_DataConflict)
 		}
 		upsertResp = s.handleCreateConfigFile(ctx, tx, upsertFileReq)
+		historyRecords = append(historyRecords, func() {
+			s.RecordHistory(ctx, configFileRecordEntry(ctx, upsertFileReq, model.OCreate))
+		})
 	} else {
 		if req.GetMd5().GetValue() != CalMd5(saveFile.Content) {
 			return api.NewConfigResponse(apimodel.Code_DataConflict)
 		}
 		upsertResp = s.handleUpdateConfigFile(ctx, tx, upsertFileReq)
+		historyRecords = append(historyRecords, func() {
+			s.RecordHistory(ctx, configFileRecordEntry(ctx, upsertFileReq, model.OUpdate))
+		})
 	}
 	if upsertResp.GetCode().GetValue() != uint32(apimodel.Code_ExecuteSuccess) {
 		return upsertResp
@@ -625,6 +634,9 @@ func (s *Server) CasUpsertAndReleaseConfigFile(ctx context.Context,
 		log.Error("[Config][File] upsert config file when commit tx.", utils.RequestID(ctx), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
+	for i := range historyRecords {
+		historyRecords[i]()
+	}
 	s.recordReleaseHistory(ctx, data, utils.ReleaseTypeNormal, utils.ReleaseStatusSuccess, "")
 	return releaseResp
 }
@@ -667,9 +679,18 @@ func (s *Server) UpsertAndReleaseConfigFile(ctx context.Context,
 	defer func() {
 		_ = tx.Rollback()
 	}()
+
+	historyRecords := []func(){}
 	upsertResp := s.handleCreateConfigFile(ctx, tx, upsertFileReq)
 	if upsertResp.GetCode().GetValue() == uint32(apimodel.Code_ExistedResource) {
 		upsertResp = s.handleUpdateConfigFile(ctx, tx, upsertFileReq)
+		historyRecords = append(historyRecords, func() {
+			s.RecordHistory(ctx, configFileRecordEntry(ctx, upsertFileReq, model.OUpdate))
+		})
+	} else {
+		historyRecords = append(historyRecords, func() {
+			s.RecordHistory(ctx, configFileRecordEntry(ctx, upsertFileReq, model.OCreate))
+		})
 	}
 	if upsertResp.GetCode().GetValue() != uint32(apimodel.Code_ExecuteSuccess) {
 		return upsertResp
@@ -693,6 +714,9 @@ func (s *Server) UpsertAndReleaseConfigFile(ctx context.Context,
 		log.Error("[Config][File] upsert config file when commit tx.", utils.RequestID(ctx), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
+	for i := range historyRecords {
+		historyRecords[i]()
+	}
 	s.recordReleaseHistory(ctx, data, utils.ReleaseTypeNormal, utils.ReleaseStatusSuccess, "")
 	return releaseResp
 }