QueryLocationTracksCmd.java 2.96 KB
/*
 * Copyright 2017-2020 吴学文 and java110 team.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.java110.user.cmd.property;

import com.alibaba.fastjson.JSONObject;
import com.java110.core.annotation.Java110Cmd;
import com.java110.core.context.ICmdDataFlowContext;
import com.java110.core.event.cmd.Cmd;
import com.java110.core.event.cmd.CmdEvent;
import com.java110.user.dao.property.ILocationTrackV1ServiceDao;
import com.java110.utils.util.Assert;
import com.java110.vo.ResultVo;
import org.springframework.beans.factory.annotation.Autowired;

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

@Java110Cmd(serviceCode = "property.queryLocationTracks")
public class QueryLocationTracksCmd extends Cmd {

    @Autowired
    private ILocationTrackV1ServiceDao locationTrackV1ServiceDao;

    @Override
    public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) {
        Assert.hasKeyAndValue(reqJson, "userId", "未包含用户ID");
    }

    @Override
    public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) {
        Map params = buildQueryParams(reqJson);
        List<Map> list = locationTrackV1ServiceDao.queryLocationTracks(params);
        int count = locationTrackV1ServiceDao.queryLocationTracksCount(params);
        context.setResponseEntity(ResultVo.createResponseEntity(count, count, list));
    }

    private Map buildQueryParams(JSONObject reqJson) {
        Map params = new HashMap<>();
        if (reqJson.containsKey("userId")) params.put("userId", reqJson.getString("userId"));
        if (reqJson.containsKey("startTime")) {
            String startTime = reqJson.getString("startTime");
            if (startTime != null && startTime.trim().length() == 10) {
                startTime = startTime.trim() + " 00:00:00";
            }
            params.put("startTime", startTime);
        }
        if (reqJson.containsKey("endTime")) {
            String endTime = reqJson.getString("endTime");
            if (endTime != null && endTime.trim().length() == 10) {
                endTime = endTime.trim() + " 23:59:59";
            }
            params.put("endTime", endTime);
        }
        if (reqJson.containsKey("page")) {
            int page = reqJson.getIntValue("page");
            int row = reqJson.containsKey("row") ? reqJson.getIntValue("row") : 20;
            params.put("page", (page - 1) * row);
            params.put("row", row);
        }
        return params;
    }
}