1/*- 2 * See the file LICENSE for redistribution information. 3 * 4 * Copyright (c) 2002-2009 Oracle. All rights reserved. 5 * 6 * $Id$ 7 */ 8 9package com.sleepycat.collections.test; 10 11import com.sleepycat.bind.RecordNumberBinding; 12import com.sleepycat.collections.PrimaryKeyAssigner; 13import com.sleepycat.db.DatabaseEntry; 14 15/** 16 * @author Mark Hayes 17 */ 18class TestKeyAssigner implements PrimaryKeyAssigner { 19 20 private byte next = 1; 21 private final boolean isRecNum; 22 23 TestKeyAssigner(boolean isRecNum) { 24 25 this.isRecNum = isRecNum; 26 } 27 28 public void assignKey(DatabaseEntry keyData) { 29 if (isRecNum) { 30 RecordNumberBinding.recordNumberToEntry(next, keyData); 31 } else { 32 keyData.setData(new byte[] { next }, 0, 1); 33 } 34 next += 1; 35 } 36 37 void reset() { 38 39 next = 1; 40 } 41} 42