物联网开发完成单点登录功能

service-job/src/main/java/com/java110/job/cmd/iot/GetIotTokenCmd.java

@@ -0,0 +1,96 @@
+package com.java110.job.cmd.iot;
+
+import com.alibaba.fastjson.JSONObject;
+import com.java110.core.annotation.Java110Cmd;
+import com.java110.core.context.CmdContextUtils;
+import com.java110.core.context.ICmdDataFlowContext;
+import com.java110.core.event.cmd.Cmd;
+import com.java110.core.event.cmd.CmdEvent;
+import com.java110.dto.store.StoreDto;
+import com.java110.dto.store.StoreUserDto;
+import com.java110.dto.user.UserDto;
+import com.java110.intf.store.IStoreUserV1InnerServiceSMO;
+import com.java110.intf.store.IStoreV1InnerServiceSMO;
+import com.java110.intf.user.IUserV1InnerServiceSMO;
+import com.java110.job.adapt.hcIot.IotConstant;
+import com.java110.job.adapt.hcIotNew.http.ISendIot;
+import com.java110.utils.cache.MappingCache;
+import com.java110.utils.exception.CmdException;
+import com.java110.utils.util.Assert;
+import com.java110.vo.ResultVo;
+import org.springframework.beans.factory.annotation.Autowired;
+
+import java.text.ParseException;
+import java.util.List;
+
+/**
+ * 获取IOT的token
+ */
+@Java110Cmd(serviceCode = "iot.getIotToken")
+public class GetIotTokenCmd extends Cmd {
+
+    @Autowired
+    private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl;
+
+    @Autowired
+    private IStoreUserV1InnerServiceSMO storeUserV1InnerServiceSMOImpl;
+
+    @Autowired
+    private IStoreV1InnerServiceSMO storeV1InnerServiceSMOImpl;
+
+    public static final String IOT_DOMAIN = "IOT"; // 物联网域
+
+
+    @Autowired
+    private ISendIot sendIotImpl;
+
+    @Override
+    public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
+
+        Assert.hasKeyAndValue(reqJson, "targetUrl", "未包含小区");
+    }
+
+    @Override
+    public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
+
+        String userId = CmdContextUtils.getUserId(context);
+
+        UserDto userDto = new UserDto();
+        userDto.setUserId(userId);
+        List<UserDto> userDtos = userV1InnerServiceSMOImpl.queryUsers(userDto);
+
+        Assert.listOnlyOne(userDtos,"用户未登录");
+
+        StoreUserDto storeUserDto = new StoreUserDto();
+        storeUserDto.setUserId(userId);
+        List<StoreUserDto> storeUserDtos = storeUserV1InnerServiceSMOImpl.queryStoreUsers(storeUserDto);
+
+        Assert.listOnlyOne(storeUserDtos,"未包含商户");
+
+
+        StoreDto storeDto = new StoreDto();
+        storeDto.setStoreId(storeUserDtos.get(0).getStoreId());
+        List<StoreDto> storeDtos = storeV1InnerServiceSMOImpl.queryStores(storeDto);
+
+        Assert.listOnlyOne(storeDtos,"商户不存在");
+
+        JSONObject staff = new JSONObject();
+        staff.put("tel", userDtos.get(0).getTel());
+        staff.put("storeName", storeDtos.get(0).getStoreName());
+
+        ResultVo resultVo = sendIotImpl.post("/iot/api/sso.getSsoToken", staff);
+
+        if(resultVo.getCode() != ResultVo.CODE_OK){
+            throw new CmdException(resultVo.getMsg());
+        }
+
+        JSONObject paramOut = (JSONObject) resultVo.getData();
+
+        String iotUrl = MappingCache.getValue(IOT_DOMAIN, IotConstant.IOT_URL);
+        String targetUrl = iotUrl+reqJson.getString("targetUrl");
+        String url = iotUrl+"/sso.html?token="+paramOut.getString("token")+"&targetUrl="+targetUrl;
+        paramOut.put("url",url);
+        context.setResponseEntity(ResultVo.createResponseEntity(paramOut));
+
+    }
+}