content: /*
   * Copyright 2012-2024 Aerospike, Inc.
   *
   * Portions may be licensed to Aerospike, Inc. under one or more contributor
   * license agreements WHICH ARE COMPATIBLE WITH THE APACHE LICENSE, VERSION 2.0.
   *
   * 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 com.aerospike.client.query;
  
  import com.aerospike.client.util.RandomShift;
  
  public final class TaskGen {
  	private RandomShift random;
  	private long taskId;
  
  	public TaskGen(Statement statement) {
  		taskId = statement.getTaskId();
  
  		if (taskId == 0) {
  			random = new RandomShift();
  			taskId = random.nextLong();
  		}
  	}
  
  	public long getId() {
  		return taskId;
  	}
  
  	public long nextId() {
  		if (random == null) {
  			random = new RandomShift();
  		}
  		return random.nextLong();
  	}
  }
  