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