text: // Copyright 2022 PingCAP, Ltd.
  //
  // 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.
  
  #pragma once
  #include <memory>
  namespace DB::DM::tests
  {
  struct WorkloadOptions;
  
  class KeyGenerator
  {
  public:
      static std::unique_ptr<KeyGenerator> create(const WorkloadOptions & opts);
  
      KeyGenerator() {}
      virtual ~KeyGenerator() {}
  
      virtual uint64_t get64() = 0;
  };
  } // namespace DB::DM::tests