content: /*
   * Copyright 2019 Aiven Oy and jdbc-connector-for-apache-kafka project contributors
   * Copyright 2016 Confluent 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 io.aiven.connect.jdbc.source;
  
  import java.util.concurrent.TimeUnit;
  import java.util.function.Supplier;
  
  import org.apache.kafka.common.errors.TimeoutException;
  import org.apache.kafka.common.utils.Time;
  
  /**
   * A clock that you can manually advance by calling sleep
   */
  public class MockTime implements Time {
  
      private long nanos;
      private long autoTickMs;
  
      public MockTime() {
          this.nanos = System.nanoTime();
      }
  
      public MockTime(final long autoTickMs) {
          this.nanos = System.nanoTime();
          this.autoTickMs = autoTickMs;
      }
  
      @Override
      public long milliseconds() {
          this.sleep(autoTickMs);
          return TimeUnit.MILLISECONDS.convert(this.nanos, TimeUnit.NANOSECONDS);
      }
  
      @Override
      public long hiResClockMs() {
          return TimeUnit.NANOSECONDS.toMillis(nanoseconds());
      }
  
      @Override
      public long nanoseconds() {
          this.sleep(autoTickMs);
          return nanos;
      }
  
      @Override
      public void sleep(final long ms) {
          this.nanos += TimeUnit.NANOSECONDS.convert(ms, TimeUnit.MILLISECONDS);
      }
  
      @Override
      public void waitObject(final Object object, final Supplier<Boolean> condition, final long deadlineMs)
          throws InterruptedException {
          synchronized (object) {
              while (true) {
                  if (condition.get()) {
                      return;
                  }
  
                  final long currentTimeMs = milliseconds();
                  if (currentTimeMs >= deadlineMs) {
                      throw new TimeoutException("Condition not satisfied before deadline");
                  }
  
                  object.wait(deadlineMs - currentTimeMs);
              }
          }
      }
  
  }
  