// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.

#include "util/progress_updater.h"

#include "common/logging.h"

namespace doris {

ProgressUpdater::ProgressUpdater(const std::string& label, int64_t total, int period) :
        _label(label), _logging_level(2), _total(total), _update_period(period),
        _num_complete(0), _last_output_percentage(0) {
}

ProgressUpdater::ProgressUpdater() :
        _logging_level(2),
        _total(0),
        _update_period(0),
        _num_complete(0),
        _last_output_percentage(0) {
}

void ProgressUpdater::update(int64_t delta) {
    DCHECK_GE(delta, 0);

    if (delta == 0) {
        return;
    }

    __sync_fetch_and_add(&_num_complete, delta);

    // Cache some shared variables to avoid locking.  It's possible the progress
    // update is out of order (e.g. prints 1 out of 10 after 2 out of 10)
    double old_percentage = _last_output_percentage;
    int64_t num_complete = _num_complete;

    if (num_complete >= _total) {
        // Always print the final 100% complete
        VLOG(_logging_level) << _label << " 100\% Complete ("
                             << num_complete << " out of " << _total << ")";
        return;
    }

    // Convert to percentage as int
    int new_percentage = (static_cast<double>(num_complete) / _total) * 100;

    if (new_percentage - old_percentage > _update_period) {
        // Only update shared variable if this guy was the latest.
        __sync_val_compare_and_swap(&_last_output_percentage, old_percentage, new_percentage);
        VLOG(_logging_level) << _label << ": " << new_percentage << "\% Complete ("
                             << num_complete << " out of " << _total << ")";
    }
}
}