Skip to content

Commit

Permalink
style:fix code style
Browse files Browse the repository at this point in the history
  • Loading branch information
chuntaojun committed Nov 26, 2022
1 parent f520903 commit dd8308e
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 23 deletions.
38 changes: 19 additions & 19 deletions cluster/router/polaris/router.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,24 +53,24 @@ var (
)

func newPolarisRouter() (*polarisRouter, error) {
routerApi, err := remotingpolaris.GetRouterAPI()
routerAPI, err := remotingpolaris.GetRouterAPI()
if err != nil {
return nil, err
}
consumerApi, err := remotingpolaris.GetConsumerAPI()
consumerAPI, err := remotingpolaris.GetConsumerAPI()
if err != nil {
return nil, err
}

return &polarisRouter{
routerApi: routerApi,
consumerApi: consumerApi,
routerAPI: routerAPI,
consumerAPI: consumerAPI,
}, nil
}

type polarisRouter struct {
routerApi polaris.RouterAPI
consumerApi polaris.ConsumerAPI
routerAPI polaris.RouterAPI
consumerAPI polaris.ConsumerAPI

cancel context.CancelFunc

Expand All @@ -97,12 +97,12 @@ func (p *polarisRouter) Route(invokers []protocol.Invoker, url *common.URL,
targetIns := make([]model.Instance, 0, len(invokers))
for i := range invokers {
invoker := invokers[i]
instanceId := invoker.GetURL().GetParam(constant.PolarisInstanceID, "")
if len(instanceId) == 0 {
instanceID := invoker.GetURL().GetParam(constant.PolarisInstanceID, "")
if len(instanceID) == 0 {
continue
}
invokersMap[instanceId] = invoker
if val, ok := instanceMap[instanceId]; ok {
invokersMap[instanceID] = invoker
if val, ok := instanceMap[instanceID]; ok {
targetIns = append(targetIns, val)
}
}
Expand All @@ -116,7 +116,7 @@ func (p *polarisRouter) Route(invokers []protocol.Invoker, url *common.URL,
Namespace: remotingpolaris.GetNamespace(),
}, targetIns)

resp, err := p.routerApi.ProcessRouters(&req)
resp, err := p.routerAPI.ProcessRouters(&req)
if err != nil {
return invokers
}
Expand Down Expand Up @@ -148,7 +148,7 @@ func getService(url *common.URL) string {
}

func (p *polarisRouter) buildRouteRequest(svc string, url *common.URL,
invoaction protocol.Invocation) (polaris.ProcessRoutersRequest, error) {
invocation protocol.Invocation) (polaris.ProcessRoutersRequest, error) {

routeReq := polaris.ProcessRoutersRequest{
ProcessRoutersRequest: model.ProcessRoutersRequest{
Expand All @@ -158,8 +158,8 @@ func (p *polarisRouter) buildRouteRequest(svc string, url *common.URL,
},
}

attachement := invoaction.Attachments()
arguments := invoaction.Arguments()
attachement := invocation.Attachments()
arguments := invocation.Arguments()

labels, err := p.buildTrafficLabels(svc)
if err != nil {
Expand All @@ -169,7 +169,7 @@ func (p *polarisRouter) buildRouteRequest(svc string, url *common.URL,
for i := range labels {
label := labels[i]
if strings.Compare(label, model.LabelKeyPath) == 0 {
routeReq.AddArguments(model.BuildPathArgument(getInvokeMethod(url, invoaction)))
routeReq.AddArguments(model.BuildPathArgument(getInvokeMethod(url, invocation)))
continue
}
if strings.HasPrefix(label, model.LabelKeyHeader) {
Expand All @@ -193,7 +193,7 @@ func (p *polarisRouter) buildTrafficLabels(svc string) ([]string, error) {
req.Namespace = remotingpolaris.GetNamespace()
req.Service = svc
req.SetTimeout(time.Second)
engine := p.routerApi.SDKContext().GetEngine()
engine := p.routerAPI.SDKContext().GetEngine()
resp, err := engine.SyncGetServiceRule(model.EventRouting, req)
if err != nil {
logger.Errorf("[Router][Polaris] ns:%s svc:%s get route rule fail : %+v", req.GetNamespace(), req.GetService(), err)
Expand Down Expand Up @@ -247,7 +247,7 @@ func collectRouteLabels(routings []*v1.Route) []string {
}

func (p *polarisRouter) buildInstanceMap(svc string) map[string]model.Instance {
resp, err := p.consumerApi.GetAllInstances(&polaris.GetAllInstancesRequest{
resp, err := p.consumerAPI.GetAllInstances(&polaris.GetAllInstancesRequest{
GetAllInstancesRequest: model.GetAllInstancesRequest{
Service: svc,
Namespace: remotingpolaris.GetNamespace(),
Expand Down Expand Up @@ -294,14 +294,14 @@ func (p *polarisRouter) Notify(invokers []protocol.Invoker) {
req.Service = service
req.SetTimeout(time.Second)

engine := p.routerApi.SDKContext().GetEngine()
engine := p.routerAPI.SDKContext().GetEngine()
_, err := engine.SyncGetServiceRule(model.EventRouting, req)
if err != nil {
logger.Errorf("[Router][Polaris] ns:%s svc:%s get route rule fail : %+v", req.GetNamespace(), req.GetService(), err)
return
}

_, err = p.consumerApi.GetAllInstances(&polaris.GetAllInstancesRequest{
_, err = p.consumerAPI.GetAllInstances(&polaris.GetAllInstancesRequest{
GetAllInstancesRequest: model.GetAllInstancesRequest{
Service: service,
Namespace: remotingpolaris.GetNamespace(),
Expand Down
8 changes: 4 additions & 4 deletions filter/polaris/limit/limiter.go
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,13 @@ import (
)

type polarisTpsLimiter struct {
limitApi polaris.LimitAPI
limitAPI polaris.LimitAPI
}

func (pl *polarisTpsLimiter) IsAllowable(url *common.URL, invocation protocol.Invocation) bool {
var err error

pl.limitApi, err = remotingpolaris.GetLimiterAPI()
pl.limitAPI, err = remotingpolaris.GetLimiterAPI()
if err != nil {
logger.Error("[TpsLimiter][Polaris] create polaris LimitAPI fail : %+v", err)
return true
Expand All @@ -59,7 +59,7 @@ func (pl *polarisTpsLimiter) IsAllowable(url *common.URL, invocation protocol.In
}
logger.Debugf("[TpsLimiter][Polaris] quota req : %+v", req)

resp, err := pl.limitApi.GetQuota(req)
resp, err := pl.limitAPI.GetQuota(req)
if err != nil {
logger.Error("[TpsLimiter][Polaris] ns:%s svc:%s get quota fail : %+v", remotingpolaris.GetNamespace(), url.Service(), err)
return true
Expand Down Expand Up @@ -121,7 +121,7 @@ func (pl *polarisTpsLimiter) buildQuotaRequest(url *common.URL, invoaction proto
}

func (pl *polarisTpsLimiter) buildArguments(req *model.QuotaRequestImpl) ([]*v1.MatchArgument, bool) {
engine := pl.limitApi.SDKContext().GetEngine()
engine := pl.limitAPI.SDKContext().GetEngine()

getRuleReq := &data.CommonRateLimitRequest{
DstService: model.ServiceKey{
Expand Down

0 comments on commit dd8308e

Please sign in to comment.