package com.java110.user.cmd.user; 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.dto.user.UserDto; import com.java110.intf.user.IUserInnerServiceSMO; import com.java110.utils.exception.CmdException; import com.java110.utils.util.Assert; import com.java110.utils.util.BeanConvertUtil; import com.java110.utils.util.PinYinUtil; import com.java110.utils.util.StringUtil; import com.java110.vo.api.staff.ApiStaffDataVo; import com.java110.vo.api.staff.ApiStaffVo; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import java.util.ArrayList; import java.util.List; @Java110Cmd(serviceCode = "query.staff.infos") public class QueryStaffInfosCmd extends Cmd { @Autowired private IUserInnerServiceSMO userInnerServiceSMOImpl; @Override public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException { Assert.hasKeyAndValue(reqJson,"page","请求报文中未包含page节点"); Assert.hasKeyAndValue(reqJson,"row","请求报文中未包含rows节点"); if (!reqJson.containsKey("storeId")) { String storeId = context.getReqHeaders().get("store-id"); reqJson.put("storeId", storeId); } Assert.hasKeyAndValue(reqJson,"storeId","请求报文中未包含storeId节点"); } @Override public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException { UserDto userDto = BeanConvertUtil.covertBean(reqJson, UserDto.class); int count = userInnerServiceSMOImpl.getStaffCount(userDto); List staffs = null; if (count > 0) { staffs = BeanConvertUtil.covertBeanList(userInnerServiceSMOImpl.getStaffs(userDto), ApiStaffDataVo.class); refreshInitials(staffs); } else { staffs = new ArrayList<>(); } ApiStaffVo apiStaffVo = new ApiStaffVo(); apiStaffVo.setTotal(count); apiStaffVo.setRecords((int) Math.ceil((double) count / (double) reqJson.getInteger("row"))); apiStaffVo.setStaffs(staffs); ResponseEntity responseEntity = new ResponseEntity(JSONObject.toJSONString(apiStaffVo), HttpStatus.OK); context.setResponseEntity(responseEntity); } /** * 刷入首字母 * @param staffs */ private void refreshInitials(List staffs) { for(ApiStaffDataVo staffDataVo : staffs){ if(StringUtil.isEmpty(staffDataVo.getName())){ continue; } staffDataVo.setInitials(PinYinUtil.getFirstSpell(staffDataVo.getName()).toUpperCase().charAt(0)+""); } } }