springboot使用Cassandra組合查詢數據

在application.yml中配置cassandra

 data:
   cassandra:
     contact-points: 192.168.0.122
     port: 9042   
     username: XXX
     password: XX
     keyspace-name: XXX

pom中引入Cassandra

<!-- 引入cassandra-->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-cassandra</artifactId>
</dependency>

封裝Cassandra的通用類

package com.wuzheng.vehicle.info.repository;

import org.springframework.data.cassandra.core.query.Query;

import java.io.Serializable;
import java.util.List;

/**
 * @description：基礎 倉儲層
 * @author：twislyn
 * @date：Created in 2020/05/21
 * @modified By：
 * @version: 1.0
 */
public interface BaseRepository<T> {

    /**
     * 按主鍵查詢
     * @param id
     * @return
     */
    T getById(Serializable id);

    /**
     * 按條件查詢對象
     * @param query
     * @return
     */
    T getObj(Query query);

    /**
     * 按條件查詢列表
     * @param query
     * @return
     */
    List<T> listObjs(Query query);

//    /**
//     * 分頁查詢
//     * @param pagination
//     * @param query
//     * @return
//     */
//    Slice<T> pageObjs(Pagination pagination, Query query);
}

實現類:

package com.wuzheng.vehicle.info.repository.impl;

import com.datastax.driver.core.PagingState;

import com.wuzheng.vehicle.info.repository.BaseRepository;
import com.wuzheng.vehicle.info.utils.Pagination;
import org.springframework.data.cassandra.core.CassandraTemplate;
import org.springframework.data.cassandra.core.query.CassandraPageRequest;
import org.springframework.data.cassandra.core.query.Query;
import org.springframework.data.domain.Slice;

import javax.annotation.Resource;
import java.io.Serializable;
import java.lang.reflect.ParameterizedType;
import java.util.List;

/**
 * @description：基礎 倉儲實現層
 * @author：twislyn
 * @date：Created in 2020/05/21
 * @modified By：
 * @version: 1.0
 */
public class BaseRepositoryImpl<T> implements BaseRepository<T> {
    @Resource
    CassandraTemplate cassandraTemplate;

    protected Class<T> entityClass = currentModelClass();

    /**
     * 獲取泛型類型
     *
     * @return
     */
    @SuppressWarnings("unchecked")
    protected Class<T> currentModelClass() {
        return (Class<T>) ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments()[0];
    }

    @Override
    public T getById(Serializable id) {
        return cassandraTemplate.selectOneById(id, entityClass);
    }

    @Override
    public T getObj(Query query) {
        return cassandraTemplate.selectOne(query, entityClass);
    }

    @Override
    public List<T> listObjs(Query query) {
        return cassandraTemplate.select(query, entityClass);
    }

//    @Override
//    public Slice<T> pageObjs(Pagination pagination, Query query) {
//        if (pagination.getPageSize() > 1) {
//            CassandraPageRequest pageRequest = pagination.getPageRequest();
//            Slice<T> slice = cassandraTemplate.slice(query.pageRequest(pageRequest), entityClass);
//            for (int i = 1, len = pagination.getPageSize(); i < len; i++) {
//                PagingState pagingState = ((CassandraPageRequest) slice.getPageable()).getPagingState();
//                if (pagingState == null) {
//                    return slice;
//                }
//                pageRequest = CassandraPageRequest.of(slice.getPageable(), pagingState);
//                slice = cassandraTemplate.slice(query.pageRequest(pageRequest), entityClass);
//            }
//            return slice;
//        }
//        return cassandraTemplate.slice(query.pageRequest(pagination.getPageRequest()), entityClass);
//
//    }
}

具體使用

自定義對應Cassandra的數據庫的實體

package com.wuzheng.vehicle.supervise.entity;

import org.springframework.data.cassandra.core.mapping.Column;
import org.springframework.data.cassandra.core.mapping.PrimaryKey;
import org.springframework.data.cassandra.core.mapping.Table;

import java.util.List;
import java.util.Map;

/**
 * @description：燃油車輛實時數據
 * @author：twislyn
 * @date：Created in 2020/
 * 05/22
 * @modified By：
 * @version: 1.0
 */
@Table("w_iot_fuelvehicle_realtime_base")
public class IotFuelVehicleRealtimeBase {
    @PrimaryKey("rowid")
    private String rowId;
    /**
     * 進氣量
     */
    @Column("airin")
    private Double airIn;
    /**
     * 大氣壓力
     */
    @Column("atmp")
    private Double atmp;
    /**
     * 標定驗證碼
     */
    @Column("cvn")
    private String cvn;
    @Column("dpf")
    private Double dpf;
    /**
     * 發動機冷卻液溫度
     */
    @Column("ect")
    private Integer ect;
    /**
     * 發動機燃料流量
     */
    @Column("eflow")
    private Double eflow;
    /**
     * 發動機轉速
     */
    @Column("espeed")
    private Double espeed;
    @Column("flist")
    private List flist;
    /**
     * 油箱液位
     */
    @Column("flvl")
    private Double flvl;
    /**
     * 故障碼數量
     */
    @Column("fnum")
    private Integer fnum;
    /**
     * 摩擦扭矩（作爲發動機最大基準扭矩的百分比）
     */
    @Column("ftorq")
    private Integer ftorq;
    /**
     * 數據採集時間_登入
     */
    @Column("gatime")
    private String gatime;
    /**
     * IUPR值
     */
    @Column("iupr")
    private String iupr;
    /**
     * 緯度
     */
    @Column("lat")
    private Double lat;
    /**
     * 經度
     */
    @Column("lng")
    private Double lng;
    @Column("locs")
    private Map<String, Double> locs;
    /**
     * 定位狀態
     */
    @Column("locsts")
    private String locsts;
    /**
     * MIL狀態
     */
    @Column("milsts")
    private Integer milsts;
    /**
     * OBD診斷協議
     */
    @Column("obddp")
    private Integer obddp;
    /**
     * 發動機淨輸出扭矩，或發動機實際扭矩/指示扭矩
     */
    @Column("otorq")
    private Integer otorq;
    @Column("rdysts")
    private String rdysts;
    @Column("reiss")
    private Boolean reiss;
    @Column("remrg")
    private Double remrg;
    @Column("rfld")
    private Map rfld;
    @Column("scrdnnox")
    private Double scrdnnox;
    @Column("scrint")
    private Double scrint;
    @Column("scroutt")
    private Double scroutt;
    @Column("scrupnox")
    private Double scrupnox;
    /**
     * 車速
     */
    @Column("speed")
    private Double speed;
    @Column("sstr")
    private String sstr;
    @Column("supsts")
    private String supsts;
    @Column("swareid")
    private String swareid;
    /**
     * 累計里程
     */
    @Column("tmiles")
    private Double tmiles;
    /**
     * vin碼
     */
    @Column("vin")
    private String vin;

    public String getRowId() {
        return rowId;
    }

    public void setRowId(String rowId) {
        this.rowId = rowId;
    }

    public Double getAirIn() {
        return airIn;
    }

    public void setAirIn(Double airIn) {
        this.airIn = airIn;
    }

    public Double getAtmp() {
        return atmp;
    }

    public void setAtmp(Double atmp) {
        this.atmp = atmp;
    }

    public String getCvn() {
        return cvn;
    }

    public void setCvn(String cvn) {
        this.cvn = cvn;
    }

    public Double getDpf() {
        return dpf;
    }

    public void setDpf(Double dpf) {
        this.dpf = dpf;
    }

    public Integer getEct() {
        return ect;
    }

    public void setEct(Integer ect) {
        this.ect = ect;
    }

    public Double getEflow() {
        return eflow;
    }

    public void setEflow(Double eflow) {
        this.eflow = eflow;
    }

    public Double getEspeed() {
        return espeed;
    }

    public void setEspeed(Double espeed) {
        this.espeed = espeed;
    }

    public List getFlist() {
        return flist;
    }

    public void setFlist(List flist) {
        this.flist = flist;
    }

    public Double getFlvl() {
        return flvl;
    }

    public void setFlvl(Double flvl) {
        this.flvl = flvl;
    }

    public Integer getFnum() {
        return fnum;
    }

    public void setFnum(Integer fnum) {
        this.fnum = fnum;
    }

    public Integer getFtorq() {
        return ftorq;
    }

    public void setFtorq(Integer ftorq) {
        this.ftorq = ftorq;
    }

    public String getGatime() {
        return gatime;
    }

    public void setGatime(String gatime) {
        this.gatime = gatime;
    }

    public String getIupr() {
        return iupr;
    }

    public void setIupr(String iupr) {
        this.iupr = iupr;
    }

    public Double getLat() {
        return lat;
    }

    public void setLat(Double lat) {
        this.lat = lat;
    }

    public Double getLng() {
        return lng;
    }

    public void setLng(Double lng) {
        this.lng = lng;
    }

    public Map<String, Double> getLocs() {
        return locs;
    }

    public void setLocs(Map<String, Double> locs) {
        this.locs = locs;
    }

    public String getLocsts() {
        return locsts;
    }

    public void setLocsts(String locsts) {
        this.locsts = locsts;
    }

    public Integer getMilsts() {
        return milsts;
    }

    public void setMilsts(Integer milsts) {
        this.milsts = milsts;
    }

    public Integer getObddp() {
        return obddp;
    }

    public void setObddp(Integer obddp) {
        this.obddp = obddp;
    }

    public Integer getOtorq() {
        return otorq;
    }

    public void setOtorq(Integer otorq) {
        this.otorq = otorq;
    }

    public String getRdysts() {
        return rdysts;
    }

    public void setRdysts(String rdysts) {
        this.rdysts = rdysts;
    }

    public Boolean getReiss() {
        return reiss;
    }

    public void setReiss(Boolean reiss) {
        this.reiss = reiss;
    }

    public Double getRemrg() {
        return remrg;
    }

    public void setRemrg(Double remrg) {
        this.remrg = remrg;
    }

    public Map getRfld() {
        return rfld;
    }

    public void setRfld(Map rfld) {
        this.rfld = rfld;
    }

    public Double getScrdnnox() {
        return scrdnnox;
    }

    public void setScrdnnox(Double scrdnnox) {
        this.scrdnnox = scrdnnox;
    }

    public Double getScrint() {
        return scrint;
    }

    public void setScrint(Double scrint) {
        this.scrint = scrint;
    }

    public Double getScroutt() {
        return scroutt;
    }

    public void setScroutt(Double scroutt) {
        this.scroutt = scroutt;
    }

    public Double getScrupnox() {
        return scrupnox;
    }

    public void setScrupnox(Double scrupnox) {
        this.scrupnox = scrupnox;
    }

    public Double getSpeed() {
        return speed;
    }

    public void setSpeed(Double speed) {
        this.speed = speed;
    }

    public String getSstr() {
        return sstr;
    }

    public void setSstr(String sstr) {
        this.sstr = sstr;
    }

    public String getSupsts() {
        return supsts;
    }

    public void setSupsts(String supsts) {
        this.supsts = supsts;
    }

    public String getSwareid() {
        return swareid;
    }

    public void setSwareid(String swareid) {
        this.swareid = swareid;
    }

    public Double getTmiles() {
        return tmiles;
    }

    public void setTmiles(Double tmiles) {
        this.tmiles = tmiles;
    }

    public String getVin() {
        return vin;
    }

    public void setVin(String vin) {
        this.vin = vin;
    }
}

實現通用接口

package com.wuzheng.vehicle.info.repository;

import com.wuzheng.vehicle.info.entity.ForwardExportMsgTask;
import com.wuzheng.vehicle.info.dto.excel.WIotElecvehicleRealtimeBase;

import java.util.List;

/**
 * 轉發報文數據
 */
public interface IotElecvehicleRealtimeBaseRepository extends BaseRepository<WIotElecvehicleRealtimeBase> {

    /**
     * 根據條件查詢歷史數據
     *
     * @param queryInfo 查詢條件
     * @return
     */
    List<WIotElecvehicleRealtimeBase> list(ForwardExportMsgTask queryInfo);
}

具體查詢邏輯(最重要)

package com.wuzheng.vehicle.info.repository.impl;

import com.wuzheng.vehicle.info.entity.ForwardExportMsgTask;
import com.wuzheng.vehicle.info.dto.excel.WIotElecvehicleRealtimeBase;
import com.wuzheng.vehicle.info.repository.IotElecvehicleRealtimeBaseRepository;
import com.wuzheng.vehicle.info.utils.StringUtils;
import org.springframework.data.cassandra.core.query.Columns;
import org.springframework.data.cassandra.core.query.Query;
import org.springframework.stereotype.Repository;

import java.util.List;

import static org.springframework.data.cassandra.core.query.Criteria.where;

@Repository
public class IotElecvehicleRealtimeBaseRepositoryImpl extends BaseRepositoryImpl<WIotElecvehicleRealtimeBase>
        implements IotElecvehicleRealtimeBaseRepository {

    @Override
    public List<WIotElecvehicleRealtimeBase> list(ForwardExportMsgTask queryInfo) {
        Query query = Query
                .query(where("idcode").like(queryInfo.getTaskPlatIdCode()))
                .and(where("vin").in(queryInfo.getTaskVin()))
                .and(where("msgtype").is(queryInfo.getTaskMsgType() + ""));
        if (StringUtils.isNotBlank(queryInfo.getTaskStartTimeStr())) {
            query = query.and(where("gatime").gt(queryInfo.getTaskStartTimeStr()));
        }
        if (StringUtils.isNotBlank(queryInfo.getTaskEndTimeStr())) {
            query = query.and(where("gatime").lt(queryInfo.getTaskEndTimeStr()));
        }
//        query = query.sort(Sort.by("vin").descending()).withAllowFiltering();
        query = query.withAllowFiltering();
        query = query.columns(Columns.from("vin", "gatime", "sstr", "rectime", "fwdtime", "ressts", "restime", "idcode", "msgtype"));
        System.out.println(query);
        return this.cassandraTemplate.select(query, WIotElecvehicleRealtimeBase.class);
    }


}

使用

---

完