com.serphacker.serposcope.db.google.GoogleTargetDB Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
There is a newer version: 2.10.0
/* 
 * Serposcope - SEO rank checker https://serposcope.serphacker.com/
 * 
 * Copyright (c) 2016 SERP Hacker
 * @author Pierre Nogues 
 * @license https://opensource.org/licenses/MIT MIT License
 */
package com.serphacker.serposcope.db.google;

import com.google.inject.Singleton;
import com.querydsl.core.Tuple;
import com.querydsl.core.types.Expression;
import com.querydsl.core.types.dsl.Expressions;
import com.querydsl.sql.SQLQuery;
import com.querydsl.sql.dml.SQLDeleteClause;
import com.querydsl.sql.dml.SQLInsertClause;
import com.querydsl.sql.dml.SQLUpdateClause;
import com.serphacker.serposcope.db.AbstractDB;
import com.serphacker.serposcope.models.google.GoogleTarget;
import com.serphacker.serposcope.models.google.GoogleTarget.PatternType;
import com.serphacker.serposcope.querybuilder.QGoogleTarget;
import java.sql.Connection;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

@Singleton
public class GoogleTargetDB extends AbstractDB {

    QGoogleTarget t_target = QGoogleTarget.googleTarget;

    public int insert(Collection targets){
        int inserted = 0;
        
        try(Connection con = ds.getConnection()){
            
            for (GoogleTarget target : targets) {
                
                Integer key = new SQLInsertClause(con, dbTplConf, t_target)
                    .set(t_target.groupId, target.getGroupId())
                    .set(t_target.name,target.getName())
                    .set(t_target.patternType, (byte)target.getType().ordinal())
                    .set(t_target.pattern, target.getPattern())
                    .executeWithKey(t_target.id);
                
                if(key != null){
                    target.setId(key);
                    inserted++;
                }
            }
            
        } catch(Exception ex){
            LOG.error("SQL error", ex);
        }
        
        return inserted;
    }
    
    public boolean rename(GoogleTarget target){
        long inserted = 0;
        
        try(Connection con = ds.getConnection()){
            
            inserted = new SQLUpdateClause(con, dbTplConf, t_target)
                .set(t_target.name, target.getName())
                .where(t_target.id.eq(target.getId()))
                .execute();
            
        } catch(Exception ex){
            LOG.error("SQL error", ex);
        }
        
        return inserted != 0;
    }    
    
    public boolean delete(int targetId){
        boolean deleted = false;
        try(Connection con = ds.getConnection()){
            deleted = new SQLDeleteClause(con, dbTplConf, t_target)
                .where(t_target.id.eq(targetId))
                .execute() == 1;
        }catch(Exception ex){
            LOG.error("SQLError", ex);
        }
        return deleted;
    }
    
    public void wipe(){
        try(Connection con = ds.getConnection()){
            new SQLDeleteClause(con, dbTplConf, t_target).execute();
        }catch(Exception ex){
            LOG.error("SQLError", ex);
        }
    }    
    
    /**
     * list all target
     */
    public List list(){
        return list(null);
    }
    
    public boolean hasTarget(){
        Integer hasOne=null;
        
        try(Connection con = ds.getConnection()){
            
            hasOne = new SQLQuery(con, dbTplConf)
                .select(Expressions.ONE)
                .from(t_target)
                .limit(1)
                .fetchFirst();
        } catch(Exception ex){
            LOG.error("SQL error", ex);
        }        
        
        return hasOne != null;
    }
    
    /***
     * list target by group
     */
    public List list(Collection groups){
        List targets = new ArrayList<>();
        
        try(Connection con = ds.getConnection()){
            
            SQLQuery query = new SQLQuery(con, dbTplConf)
                .select(t_target.all())
                .from(t_target);
            
            if(groups != null){
                query.where(t_target.groupId.in(groups));
            }
            
            List tuples = query.fetch();
            
            if(tuples != null){
                for (Tuple tuple : tuples) {
                    targets.add(fromTuple(tuple));
                }
            }
            
        } catch(Exception ex){
            LOG.error("SQL error", ex);
        }
        
        return targets;
    }
    
    public GoogleTarget get(int targetId){
        GoogleTarget target = null;
        
        try(Connection con = ds.getConnection()){
            
            Tuple tuple = new SQLQuery(con, dbTplConf)
                .select(t_target.all())
                .from(t_target)
                .where(t_target.id.eq(targetId))
                .fetchOne();
            
            target = fromTuple(tuple);
            
        } catch(Exception ex){
            LOG.error("SQL error", ex);
        }
        
        return target;
    }    
    
    GoogleTarget fromTuple(Tuple tuple) throws Exception{
        return new GoogleTarget(
            tuple.get(t_target.id),
            tuple.get(t_target.groupId),
            tuple.get(t_target.name),
            tuple.get(t_target.patternType) == null ? PatternType.REGEX : GoogleTarget.PatternType.values()[tuple.get(t_target.patternType)],
            tuple.get(t_target.pattern)
        );
    }
    
}