/**
 * Copyright (C) 2014-2016 LinkedIn Corp. (pinot-core@linkedin.com)
 *
 * 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.linkedin.pinot.common.utils;

import org.joda.time.DateTime;
import org.joda.time.DateTimeZone;


public class LLCSegmentName extends SegmentName implements Comparable {
  private final static String DATE_FORMAT = "yyyyMMdd'T'HHmm'Z'";
  private final String _tableName;
  private final int _partitionId;
  private final int _sequenceNumber;
  private final String _creationTime;
  private final String _segmentName;

  public LLCSegmentName(String segmentName) {
    if (!isLowLevelConsumerSegmentName(segmentName)) {
      throw new RuntimeException(segmentName + " is not a Low level consumer segment name");
    }

    String[] parts = segmentName.split(SEPARATOR);
    _segmentName = segmentName;
    _tableName = parts[0];
    _partitionId = Integer.valueOf(parts[1]);
    _sequenceNumber = Integer.valueOf(parts[2]);
    _creationTime = parts[3];
  }

  public LLCSegmentName(String tableName, int partitionId, int sequenceNumber, long msSinceEpoch) {
    if (!isValidComponentName(tableName)) {
      throw new RuntimeException("Invalid table name " + tableName);
    }
    _tableName = tableName;
    _partitionId = partitionId;
    _sequenceNumber = sequenceNumber;
    // ISO8601 date: 20160120T1234Z
    DateTime dateTime = new DateTime(msSinceEpoch, DateTimeZone.UTC);
    _creationTime = dateTime.toString("yyyyMMdd'T'HHmm'Z'");
    _segmentName = tableName + SEPARATOR + partitionId + SEPARATOR + sequenceNumber + SEPARATOR + _creationTime;
  }

  @Override
  public RealtimeSegmentType getSegmentType() {
    return RealtimeSegmentType.LLC;
  }

  @Override
  public String getTableName() {
    return _tableName;
  }

  @Override
  public int getPartitionId() {
    return _partitionId;
  }

  @Override
  public String getPartitionRange() {
    return Integer.toString(getPartitionId());
  }

  @Override
  public int getSequenceNumber() {
    return _sequenceNumber;
  }

  public String getCreationTime() {
    return _creationTime;
  }

  @Override
  public String getSegmentName() {
    return _segmentName;
  }

  @Override
  public String getSequenceNumberStr() {
    return Integer.toString(_sequenceNumber);
  }

  @Override
  public int compareTo(Object o) {
    LLCSegmentName other = (LLCSegmentName)o;
    if (!this.getTableName().equals(other.getTableName())) {
      throw new RuntimeException("Cannot compare segment names " + this.getSegmentName() + " and " + other.getSegmentName());
    }
    if (this.getPartitionId() > other.getPartitionId()) {
      return 1;
    } else if (this.getPartitionId() < other.getPartitionId()) {
      return -1;
    } else {
      if (this.getSequenceNumber() > other.getSequenceNumber()) {
        return 1;
      } else if (this.getSequenceNumber() < other.getSequenceNumber()) {
        return -1;
      } else {
        if (!this.getCreationTime().equals(other.getCreationTime())) {
          // If sequence number is the same, time cannot be different.
          throw new RuntimeException("Cannot compare segment names " + this.getSegmentName() + " and " + other.getSegmentName());
        }
        return 0;
      }
    }
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }

    LLCSegmentName segName = (LLCSegmentName) o;

    if (_partitionId != segName._partitionId) {
      return false;
    }
    if (_sequenceNumber != segName._sequenceNumber) {
      return false;
    }
    if (_tableName != null ? !_tableName.equals(segName._tableName) : segName._tableName != null) {
      return false;
    }
    if (_creationTime != null ? !_creationTime.equals(segName._creationTime) : segName._creationTime != null) {
      return false;
    }
    return !(_segmentName != null ? !_segmentName.equals(segName._segmentName) : segName._segmentName != null);
  }

  @Override
  public int hashCode() {
    int result = _tableName != null ? _tableName.hashCode() : 0;
    result = 31 * result + _partitionId;
    result = 31 * result + _sequenceNumber;
    result = 31 * result + (_creationTime != null ? _creationTime.hashCode() : 0);
    result = 31 * result + (_segmentName != null ? _segmentName.hashCode() : 0);
    return result;
  }

  @Override
  public String toString() {
    return getSegmentName();
  }
}