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