/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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
<<<<<<< Updated upstream
 *
 *     https://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
=======
 * 
 *     https://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 
>>>>>>> Stashed changes
 * limitations under the License.
 */

package org.apache.jdo.tck.pc.fieldtypes;

import java.io.Serializable;
import java.util.Date;

public class SimpleClass implements Serializable, SimpleInterface, Comparable<SimpleClass> {

  private static final long serialVersionUID = 1L;

  private static long counter = new Date().getTime();

  private static long newId() {
    synchronized (SimpleClass.class) {
      return counter++;
    }
  }

  private long id = newId();
  private int intField = 0;
  private String stringField = "Test";

  public SimpleClass() {}

  public SimpleClass(int intField, String stringField) {
    this.intField = intField;
    this.stringField = stringField;
  }

  public void setIntValue(int intField) {
    this.intField = intField;
  }

  public int getIntValue() {
    return intField;
  }

  public void setStringValue(String stringField) {
    this.stringField = stringField;
  }

  public String getStringValue() {
    return stringField;
  }

  public boolean equals(Object obj) {
    if (!(obj instanceof SimpleClass)) return false;

    if (intField == ((SimpleClass) obj).getIntValue()
        && stringField.equals(((SimpleClass) obj).getStringValue())) return true;
    else return false;
  }

  public int hashCode() {
    return intField ^ stringField.hashCode();
  }

  public int compareTo(SimpleClass p) {
    if (intField < p.intField) return -1;
    if (intField > p.intField) return 1;
    return stringField.compareTo(p.stringField);
  }

  public static class Oid implements Serializable, Comparable<Oid> {

    private static final long serialVersionUID = 1L;

    public long id;

    public Oid() {}

    public Oid(String s) {
      id = Integer.parseInt(justTheId(s));
    }

    public String toString() {
      return this.getClass().getName() + ": " + id;
    }

    public int hashCode() {
      return (int) id;
    }

    public boolean equals(Object other) {
      if (other != null && (other instanceof Oid)) {
        Oid k = (Oid) other;
        return k.id == this.id;
      }
      return false;
    }

    protected static String justTheId(String str) {
      return str.substring(str.indexOf(':') + 1);
    }

    public int compareTo(Oid p) {
      return Long.compare(id, p.id);
    }
  }
}
