ContentObjectConverter.java

/*
 *
 * This is a simple Content Management System (CMS)
 * Copyright (C) 2010  Imran M Yousuf (imyousuf@smartitengineering.com)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.smartitengineering.cms.spi.impl.content;

import com.google.inject.Inject;
import com.smartitengineering.cms.api.content.ContentId;
import com.smartitengineering.cms.api.factory.SmartContentAPI;
import com.smartitengineering.cms.api.factory.content.WriteableContent;
import com.smartitengineering.cms.api.type.ContentStatus;
import com.smartitengineering.cms.api.type.ContentType;
import com.smartitengineering.cms.api.type.ContentTypeId;
import com.smartitengineering.cms.api.type.MutableContentStatus;
import com.smartitengineering.cms.spi.SmartContentSPI;
import com.smartitengineering.cms.spi.content.PersistableContent;
import com.smartitengineering.cms.spi.impl.hbase.Utils;
import com.smartitengineering.cms.spi.impl.type.PersistentContentType;
import com.smartitengineering.dao.impl.hbase.spi.ExecutorService;
import com.smartitengineering.dao.impl.hbase.spi.SchemaInfoProvider;
import com.smartitengineering.dao.impl.hbase.spi.impl.AbstractObjectRowConverter;
import java.io.IOException;
import org.apache.hadoop.hbase.client.Delete;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.util.Bytes;

/**
 *
 * @author imyousuf
 */
public class ContentObjectConverter extends AbstractObjectRowConverter<PersistentContent, ContentId> {

  public final static byte[] FAMILY_SELF = Bytes.toBytes("self");
  public final static byte[] CELL_PARENT_ID = Bytes.toBytes("parent");
  public final static byte[] CELL_CONTENT_TYPE_ID = Bytes.toBytes("contentType");
  public final static byte[] CELL_STATUS = Bytes.toBytes("status");
  public final static byte[] CELL_CREATED = Bytes.toBytes("created");
  public final static byte[] CELL_LAST_MODIFIED = Bytes.toBytes("lastModified");
  public final static byte[] CELL_ENTITY_TAG = Bytes.toBytes("entityTag");
  public final static byte[] CELL_PRIVATE = Bytes.toBytes("isPrivate");
  @Inject
  private SchemaInfoProvider<PersistentContentType, ContentTypeId> contentTypeSchemaProvider;

  @Override
  protected String[] getTablesToAttainLock() {
    return new String[]{getInfoProvider().getMainTableName()};
  }

  @Override
  protected void getPutForTable(PersistentContent instance, ExecutorService service, Put put) {
    WriteableContent content = instance.getMutableContent();
    putSelfData(content, put);
  }

  @Override
  protected void getDeleteForTable(PersistentContent instance, ExecutorService service, Delete put) {
    //Should delete all row at once
  }

  @Override
  public PersistentContent rowsToObject(Result startRow, ExecutorService executorService) {
    PersistableContent content = SmartContentSPI.getInstance().getPersistableDomainFactory().createPersistableContent(
        false);
    try {
      content.setContentId(getInfoProvider().getIdFromRowId(startRow.getRow()));
    }
    catch (Exception ex) {
      logger.error("Could not parse content Id", ex);
      throw new RuntimeException(ex);
    }
    final byte[] parentId = startRow.getValue(FAMILY_SELF, CELL_PARENT_ID);
    try {
      if (parentId != null) {
        content.setParentId(getInfoProvider().getIdFromRowId(parentId));
      }
    }
    catch (Exception ex) {
      logger.error("Could not parse parent content Id", ex);
      throw new RuntimeException(ex);
    }
    content.setCreationDate(Utils.toDate(startRow.getValue(FAMILY_SELF, CELL_CREATED)));
    content.setLastModifiedDate(Utils.toDate(startRow.getValue(FAMILY_SELF, CELL_LAST_MODIFIED)));
    content.setEntityTagValue(Bytes.toString(startRow.getValue(FAMILY_SELF, CELL_ENTITY_TAG)));
    final byte[] isPrivate = startRow.getValue(FAMILY_SELF, CELL_PRIVATE);
    if (isPrivate != null) {
      content.setPrivate(Bytes.toBoolean(isPrivate));
    }
    else {
      content.setPrivate(false);
    }
    ContentType type = null;
    try {
      ContentTypeId typeId =
                    contentTypeSchemaProvider.getIdFromRowId(startRow.getValue(FAMILY_SELF, CELL_CONTENT_TYPE_ID));
      type = SmartContentAPI.getInstance().getContentTypeLoader().loadContentType(typeId);
      if (type != null) {
        content.setContentDefinition(type);
      }
      else {
        logger.error(new StringBuilder("Could not retrieve content type id ").append(typeId).append(
            " for content ").append(content.getContentId()).toString());
        throw new RuntimeException("Could not find content type with id " + typeId);
      }
      byte[] status = startRow.getValue(FAMILY_SELF, CELL_STATUS);
      final String statusString = Bytes.toString(status);
      final ContentStatus definedStatus = type.getStatuses().get(statusString);
      if (definedStatus != null) {
        content.setStatus(definedStatus);
      }
      else {
        MutableContentStatus contentStatus = SmartContentAPI.getInstance().getContentTypeLoader().
            createMutableContentStatus();
        contentStatus.setName(statusString);
        contentStatus.setContentTypeID(typeId);
        contentStatus.setId(0);
        content.setStatus(contentStatus);
      }
    }
    catch (RuntimeException ex) {
      logger.error("Could not parse content type id", ex);
      throw ex;
    }
    catch (Exception ex) {
      logger.error("Could not parse content type id", ex);
      throw new RuntimeException(ex);
    }
    PersistentContent persistentContent = new PersistentContent();
    persistentContent.setMutableContent(content);
    return persistentContent;
  }

  protected void putSelfData(WriteableContent content, Put put) {
    try {
      put.add(FAMILY_SELF, CELL_CONTENT_TYPE_ID,
              contentTypeSchemaProvider.getRowIdFromId(content.getContentDefinition().getContentTypeID()));
    }
    catch (IOException ex) {
      logger.error("Could put content type ID", ex);
      throw new RuntimeException(ex);
    }
    put.add(FAMILY_SELF, CELL_CREATED, Utils.toBytes(content.getCreationDate()));
    put.add(FAMILY_SELF, CELL_LAST_MODIFIED, Utils.toBytes(content.getLastModifiedDate()));
    put.add(FAMILY_SELF, CELL_ENTITY_TAG, Bytes.toBytes(content.getEntityTagValue()));
    put.add(FAMILY_SELF, CELL_PRIVATE, Bytes.toBytes(content.isPrivate()));
    if (content.getStatus() != null) {
      put.add(FAMILY_SELF, CELL_STATUS, Bytes.toBytes(content.getStatus().getName()));
    }
    if (content.getParentId() != null) {
      try {
        put.add(FAMILY_SELF, CELL_PARENT_ID, getInfoProvider().getRowIdFromId(content.getParentId()));
      }
      catch (IOException ex) {
        logger.error("Could put parent ID", ex);
        throw new RuntimeException(ex);
      }
    }
  }
}