/**
 * Tencent is pleased to support the open source community by making polaris-go available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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 flow

import (
	"time"

	"gitee.com/meng_mengs_boys/polaris-go/pkg/flow/data"
	"gitee.com/meng_mengs_boys/polaris-go/pkg/model"
)

// AsyncGetQuota 异步获取配额信息
func (e *Engine) AsyncGetQuota(request *model.QuotaRequestImpl) (*model.QuotaFutureImpl, error) {
	commonRequest := data.PoolGetCommonRateLimitRequest()
	commonRequest.InitByGetQuotaRequest(request, e.configuration)
	startTime := model.CurrentMillisecond()
	future, err := e.flowQuotaAssistant.GetQuota(commonRequest)
	consumeTime := model.CurrentMillisecond() - startTime
	if err != nil {
		(&commonRequest.CallResult).SetFail(model.GetErrorCodeFromError(err), time.Duration(consumeTime)*time.Millisecond)
	} else {
		(&commonRequest.CallResult).SetDelay(time.Duration(consumeTime) * time.Millisecond)
	}
	e.syncRateLimitReportAndFinalize(commonRequest, future.GetImmediately())
	return future, err
}