Saiku更改源代码实现默认查询一天的数据（十）

Saiku通过更改源代码实现默认查询前一天数据

saiku在本地进行的编译的方式前面已有教程说明，接下来就是更改原代码了 （从网上学得教程，目前只了解到获取最新一天的数据信息）

参考博客地址： https://blog.csdn.net/zstao168/article/details/52818569

1. 主要更改的类信息：在saiku-web工程中的Query2Resource.java中，具体类路径为：  /saiku-web/src/main/java/org/saiku/web/rest/resources/Query2Resource.java

在该类的 public QueryResult execute(ThinQuery tq  ) 方法中添加  tq = this.restrictQueryByDate(tq);  (execute方法大概在该类的 185-220行左右)

 /**

   *

   * Execute a Saiku Query

   * @summary Execute Query

   * @param tq Thin Query model

   * @return A query result set.

   */

    @POST

    @Consumes({"application/json" })

    @Path("/execute")

    public QueryResult execute(ThinQuery tq) {

    	//add code to limit query date

    	tq = this.restrictQueryByDate(tq);

        try {

            if (thinQueryService.isMdxDrillthrough(tq)) {

                Long start = (new Date()).getTime();

                ResultSet rs = thinQueryService.drillthrough(tq);

                QueryResult rsc = RestUtil.convert(rs);

                rsc.setQuery(tq);

                Long runtime = (new Date()).getTime()- start;

                rsc.setRuntime(runtime.intValue());

                return rsc;

            }

            QueryResult qr = RestUtil.convert(thinQueryService.execute(tq));

            ThinQuery tqAfter = thinQueryService.getContext(tq.getName()).getOlapQuery();

            qr.setQuery(tqAfter);

            return qr;

        }

        catch (Exception e) {

            log.error("Cannot execute query (" + tq + ")",e);

            String error = ExceptionUtils.getRootCauseMessage(e);

            return new QueryResult(error);

        }

    }

　

在Query2Resource.java类中添加　ThinQuery  restrictQueryByDate（ThinQuery tq） 方法，内容如下

  // add code for query limit

  private ThinQuery restrictQueryByDate(ThinQuery tq) {

		ThinQueryModel queryModel = tq.getQueryModel();

	  	Map<AxisLocation, ThinAxis> axesMap = queryModel.getAxes();

	  	NamedList<ThinHierarchy> namedList = new NamedListImpl<ThinHierarchy>();

	  	ThinAxis filterAxis = axesMap.get(AxisLocation.FILTER);

	  	List<ThinHierarchy> filterHie = filterAxis.getHierarchies();

	  	namedList = this.resetThinHierachy(filterHie);

	  	//将修改后的Row重新set到queryModel

	  	if(namedList.size() > 0) {

			ThinAxis newFilterAxis = new ThinAxis(

					AxisLocation.FILTER,

					namedList,

					filterAxis.isNonEmpty(),

					filterAxis.getAggregators()

					);

			axesMap.put(AxisLocation.FILTER,newFilterAxis);

	  	}

	  	//将修改后的Row重新set到queryModel

	  	if(namedList.size() == 0) {

	  		ThinAxis rowAxis = axesMap.get(AxisLocation.ROWS);

	  	  	List<ThinHierarchy> rowHie = rowAxis.getHierarchies();

	  	  	namedList = this.resetThinHierachy(rowHie);

	  		if(namedList.size() > 0) {

				ThinAxis newRowsAxis = new ThinAxis(

						AxisLocation.ROWS,

						namedList,

						rowAxis.isNonEmpty(),

						rowAxis.getAggregators()

						);

				axesMap.put(AxisLocation.ROWS,newRowsAxis);

	  		}

	  	}

	  	//if columns contained date member

	  	//if contained and have not set the limit ,then add limit to one day ,if do not hava then add this column and limit to one day;

	  	if(namedList.size() == 0) {

//	  		namedList.clear();

	    	ThinAxis colAxis = axesMap.get(AxisLocation.COLUMNS);

	    	List<ThinHierarchy> colHie = colAxis.getHierarchies();

	    	namedList = this.resetThinHierachy(colHie);

	    	if(namedList.size() == 0) {

	    		//if list is empty,then column don't hava date ,then add colHie to list and enfore to add date member;

	    		namedList.addAll(colHie);

	    		SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");

				String yesterday = format.format(new Date(new Date().getTime() - 24 * 60 * 60 * 1000));

				String newDateMdx = "[SaikuUseDate].[SaikuUseDate].["+yesterday+"]";

				ThinMember thinMember = new ThinMember(yesterday,newDateMdx,yesterday);

	    		ThinHierarchy thinHie = new ThinHierarchy();

	    		thinHie.setName("[SaikuUseDate].[SaikuUseDate]");

	    		List<ThinMember> thinMemberList = new ArrayList<ThinMember>();

				thinMemberList.add(thinMember);

				ThinSelection selection = new ThinSelection();

				selection.setMembers(thinMemberList);

				selection.setType(ThinSelection.Type.INCLUSION);

				ThinLevel thinLevel = new ThinLevel(yesterday,yesterday,selection,null);

		//  			thinLevel.setSelection(selection);

				Map<String,ThinLevel> mapLevel = new LinkedHashMap<String,ThinLevel>();

				mapLevel.put("SaikuUseDate", thinLevel);

				thinHie.setLevels(mapLevel);

				namedList.add(thinHie);

	    		ThinAxis newColAxis = new ThinAxis(

	    				AxisLocation.COLUMNS,

	    				namedList,

	    				colAxis.isNonEmpty(),

	    				colAxis.getAggregators()

	    				);

	    		axesMap.put(AxisLocation.COLUMNS,newColAxis);

	    	}

	  	}

	  	return tq;

	  }

	  private NamedList<ThinHierarchy> resetThinHierachy(List<ThinHierarchy> hieList) {

		  NamedList<ThinHierarchy> namedList = new NamedListImpl<ThinHierarchy>();

		  boolean flag = false;

		  for(ThinHierarchy hie : hieList) {

		  		if(hie.getName().equals("[SaikuUseDate].[SaikuUseDate]")) {

		    		if(hie.getLevels().get("SaikuUseDate").getSelection() == null) {

		    			SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");

		    			String yesterday = format.format(new Date(new Date().getTime() - 24 * 60 * 60 * 1000));

		    			String newDateMdx = "[SaikuUseDate].[SaikuUseDate].["+yesterday+"]";

		    			ThinMember thinMember = new ThinMember(null,newDateMdx,yesterday);

		    			List<ThinMember> thinMemberList = new ArrayList<ThinMember>();

		    			thinMemberList.add(thinMember);

		    			ThinSelection selection = new ThinSelection();

		    			selection.setMembers(thinMemberList);

		    			hie.getLevels().get("SaikuUseDate").setSelection(selection);

		    		}

		    		flag = true;

		  		}

		  		namedList.add(hie);

		  	}

		  if(flag) return namedList;

		  namedList.clear();

		  return namedList;

	  }

　

ps:  如果　hie.getLevels().get("SaikuUseDate") 中得到的ThinLevel对象没有setSeletion(selection)方法，需要手工在ThinLevel中添加set方法

还有一点需要注意，就是加上这些代码之后，saiku的schame文件中cube需要有 SaikuUseDate 日期字段。

最后将整个项目打包编译即可，将新的saiku-server启动，然后配置一些数据信息，便能获取最近一天的数据信息了。