Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
   * Copyright (c) 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014 David Berkman
   * 
   * This file is part of the SmallMind Code Project.
   * 
   * The SmallMind Code Project is free software, you can redistribute
   * it and/or modify it under the terms of GNU Affero General Public
   * License as published by the Free Software Foundation, either version 3
   * of the License, or (at your option) any later version.
  * 
  * The SmallMind Code Project 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 the GNU Affero General Public
  * License, along with the SmallMind Code Project. If not, see
  * <http://www.gnu.org/licenses/>.
  * 
  * Additional permission under the GNU Affero GPL version 3 section 7
  * ------------------------------------------------------------------
  * If you modify this Program, or any covered work, by linking or
  * combining it with other code, such other code is not for that reason
  * alone subject to any of the requirements of the GNU Affero GPL
  * version 3.
  */
 package org.smallmind.persistence.nosql.hector;
 
 import java.util.List;
 
 public abstract class HectorDao<W extends Serializable & Comparable<W>, I extends Serializable & Comparable<I>, D extends Durable<I>> extends NoSqlDao<W, I, D> {
 
 
   public HectorDao (String metricSourceKeyspace keyspaceWideVectoredDao<W, I, D> wideVectoredDaoboolean cacheEnabled) {
 
     super(metricSourcewideVectoredDaocacheEnabled);
 
      = new ThriftColumnFamilyTemplate<CompositeComposite>(keyspacegetManagedClass().getSimpleName(), CompositeSerializer.get(), CompositeSerializer.get());
   }
 
   public abstract I createId ();
 
   @Override
   public void remove (String contextClass<? extends Durable<W>> parentClass, W parentIdClass<D> durableClass) {
 
     WideVectoredDao<W, I, D> wideVectoredDao;
 
     .deleteRow(new Composite(contextparentClass.getSimpleName(), HectorType.getTranslator(getParentIdClass(), "parentId").toHectorValue(parentId)));
 
     if (isCacheEnabled() && ((wideVectoredDao = getWideVectoredDao()) != null)) {
       wideVectoredDao.delete(contextparentClassparentIddurableClass);
     }
   }
 
   @Override
   public List<D> get (String contextClass<? extends Durable<W>> parentClass, W parentIdClass<D> durableClass) {
 
     List<D> durables;
     WideVectoredDao<W, I, D> wideVectoredDao;
     ColumnFamilyResult<CompositeCompositehectorResult;
 
     if (isCacheEnabled() && ((wideVectoredDao = getWideVectoredDao()) != null)) {
       if ((durables = wideVectoredDao.get(contextparentClassparentIddurableClass)) != null) {
 
         return durables;
       }
     }
 
     if ((hectorResult = .queryColumns(new Composite(contextparentClass.getSimpleName(), HectorType.getTranslator(getParentIdClass(), "parentId").toHectorValue(parentId)))).hasResults()) {
 
       HashMap<NaturalKey, D> naturalMap = new HashMap<NaturalKey, D>();
 
       for (Composite columnName : hectorResult.getColumnNames()) {
 
         D durable;
         NaturalKey naturalKey;
         Field[] naturalKeyFields = NaturalKey.getNaturalKeyFields(durableClass);
         Field nonKeyField;
        Object[] naturalKeyValues = new Object[naturalKeyFields.length];
        String nonKeyFieldName;
        int naturalKeyIndex;
        if ((nonKeyField = DurableFields.getField(durableClassnonKeyFieldName = columnName.get(naturalKeyFields.length, StringSerializer.get()))) == null) {
          throw new PersistenceException("Unknown field(%s) in return values"nonKeyFieldName);
        }
        naturalKeyIndex = 0;
        for (Field naturalKeyField : naturalKeyFields) {
          naturalKeyValues[naturalKeyIndex] = HectorType.getTranslator(naturalKeyField.getType(), naturalKeyField.getName()).toEntityValue(naturalKeyField.getType(), naturalKeyIndex++, columnName);
        }
        if ((durable = naturalMap.get(naturalKey = new NaturalKey<D>(durableClassnaturalKeyValues))) == null) {
          try {
            naturalMap.put(naturalKeydurable = durableClass.newInstance());
            naturalKeyIndex = 0;
            for (Field naturalKeyField : naturalKeyFields) {
              naturalKeyField.set(durablenaturalKeyValues[naturalKeyIndex++]);
            }
          }
          catch (Exception exception) {
            throw new PersistenceException(exception);
          }
        }
        try {
          if (nonKeyFieldName.equals("id")) {
            nonKeyField.set(durable, HectorType.getTranslator(getIdClass(), "id").toEntityValue(getIdClass(), columnNamehectorResult));
          }
          else {
            Class<?> nonKeyType = nonKeyField.getType();
            nonKeyField.set(durable, HectorType.getTranslator(nonKeyTypenonKeyFieldName).toEntityValue(nonKeyTypecolumnNamehectorResult));
          }
        }
        catch (IllegalAccessException illegalAccessException) {
          throw new PersistenceException(illegalAccessException);
        }
      }
      durables = new LinkedList<D>(naturalMap.values());
      if (isCacheEnabled() && ((wideVectoredDao = getWideVectoredDao()) != null)) {
        wideVectoredDao.persist(contextparentClassparentIddurableClassdurables);
      }
      return durables;
    }
    return Collections.emptyList();
  }
  public List<D> persist (String contextClass<? extends Durable<W>> parentClass, W parentIdClass<D> durableClassList<D> durables) {
    if ((durables != null) && (!durables.isEmpty())) {
      ColumnFamilyUpdater<CompositeCompositeupdater = .createUpdater(new Composite(contextparentClass.getSimpleName(), HectorType.getTranslator(getParentIdClass(), "parentId").toHectorValue(parentId)));
      WideVectoredDao<W, I, D> wideVectoredDao;
      try {
        for (Field nonKeyField : NaturalKey.getNonKeyFields(durableClass)) {
          for (D durable : durables) {
            I id;
            if ((id = durable.getId()) == null) {
              durable.setId(id = createId());
            }
            Composite nonKeyComposite = new Composite();
            for (Field naturalKeyField : NaturalKey.getNaturalKeyFields(durableClass)) {
              nonKeyComposite.add(HectorType.getTranslator(naturalKeyField.getType(), naturalKeyField.getName()).toHectorValue(naturalKeyField.get(durable)));
            }
            nonKeyComposite.add(nonKeyField.getName());
            if (nonKeyField.getName().equals("id")) {
              HectorTranslator idTranslator = HectorType.getTranslator(getIdClass(), "id");
              updater.setValue(nonKeyCompositeidTranslator.toHectorValue(id), idTranslator.getSerializer());
            }
            else {
              Object nonKeyValue = nonKeyField.get(durable);
              if (nonKeyValue == null) {
                updater.deleteColumn(nonKeyComposite);
              }
              else {
                HectorTranslator nonKeyTranslator = HectorType.getTranslator(nonKeyField.getType(), nonKeyField.getName());
                updater.setValue(nonKeyCompositenonKeyTranslator.toHectorValue(nonKeyValue), nonKeyTranslator.getSerializer());
              }
            }
          }
        }
      }
      catch (IllegalAccessException illegalAccessException) {
        throw new PersistenceException(illegalAccessException);
      }
      if (isCacheEnabled() && ((wideVectoredDao = getWideVectoredDao()) != null)) {
        List<D> cachedInstance;
        if ((cachedInstance = wideVectoredDao.get(contextparentClassparentIddurableClass)) != null) {
          for (D durable : durables) {
            int cachedIndex;
            if ((cachedIndex = cachedInstance.indexOf(durable)) >= 0) {
              cachedInstance.set(cachedIndexdurable);
            }
            else {
              cachedInstance.add(durable);
            }
          }
          wideVectoredDao.persist(contextparentClassparentIddurableClasscachedInstance);
        }
      }
      .update(updater);
    }
    return durables;
  }
  public void delete (String contextClass<? extends Durable<W>> parentClass, W parentIdClass<D> durableClassList<D> durables) {
    if ((durables != null) && (!durables.isEmpty())) {
      ColumnFamilyUpdater<CompositeCompositeupdater = .createUpdater(new Composite(contextparentClass.getSimpleName(), HectorType.getTranslator(getParentIdClass(), "parentId").toHectorValue(parentId)));
      WideVectoredDao<W, I, D> wideVectoredDao;
      try {
        for (Field nonKeyField : NaturalKey.getNonKeyFields(durableClass)) {
          for (D durable : durables) {
            Composite nonKeyComposite = new Composite();
            for (Field naturalKeyField : NaturalKey.getNaturalKeyFields(durableClass)) {
              nonKeyComposite.add(HectorType.getTranslator(naturalKeyField.getType(), naturalKeyField.getName()).toHectorValue(naturalKeyField.get(durable)));
            }
            nonKeyComposite.add(nonKeyField.getName());
            updater.deleteColumn(nonKeyComposite);
          }
        }
      }
      catch (IllegalAccessException illegalAccessException) {
        throw new PersistenceException(illegalAccessException);
      }
      if (isCacheEnabled() && ((wideVectoredDao = getWideVectoredDao()) != null)) {
        List<D> cachedInstance;
        if ((cachedInstance = wideVectoredDao.get(contextparentClassparentIddurableClass)) != null) {
          for (D durable : durables) {
            cachedInstance.remove(durable);
          }
          wideVectoredDao.persist(contextparentClassparentIddurableClasscachedInstance);
        }
      }
      .update(updater);
    }
  }
New to GrepCode? Check out our FAQ X