Blame view

service-job/src/main/java/com/java110/job/cmd/plugin/GetPluginTokenCmd.java 4.6 KB
88e030b7   王彪总   init project
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
  package com.java110.job.cmd.plugin;
  
  import com.alibaba.fastjson.JSONObject;
  import com.java110.core.annotation.Java110Cmd;
  import com.java110.core.client.RestTemplate;
  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.core.factory.AuthenticationFactory;
  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.utils.cache.MappingCache;
  import com.java110.utils.exception.CmdException;
  import com.java110.utils.util.Assert;
  import com.java110.utils.util.DateUtil;
  import com.java110.vo.ResultVo;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.http.HttpEntity;
  import org.springframework.http.HttpHeaders;
  import org.springframework.http.HttpMethod;
  import org.springframework.http.ResponseEntity;
  
  import java.text.ParseException;
  import java.util.List;
  
  /**
   * 获取插件token
   */
  @Java110Cmd(serviceCode = "plugin.getPluginToken")
  public class GetPluginTokenCmd extends Cmd {
  
      public static final String URL_DOMAIN = "URL_DOMAIN"; // 物联网域
  
      @Autowired
      private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl;
  
      @Autowired
      private IStoreUserV1InnerServiceSMO storeUserV1InnerServiceSMOImpl;
  
      @Autowired
      private IStoreV1InnerServiceSMO storeV1InnerServiceSMOImpl;
  
      @Autowired
      private RestTemplate outRestTemplate;
  
  
      @Override
      public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
          Assert.hasKeyAndValue(reqJson, "targetUrl", "未包含targetUrl");
          Assert.hasKeyAndValue(reqJson, "communityId", "未包含communityId");
          String pluginSwitch = MappingCache.getValue("PLUGIN", "PLUGIN_SWITCH");
  
          if (!"ON".equals(pluginSwitch)) {
              throw new CmdException("插件系统未部署");
          }
      }
  
      @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("staffId", userId);
          staff.put("timestamp", DateUtil.getCurrentDate().getTime());
          staff.put("sign", createSign(staff));
  
          HttpHeaders header = new HttpHeaders();
          HttpEntity<String> httpEntity = new HttpEntity<String>(staff.toJSONString(), header);
  
          String pluginUrl = MappingCache.getValue("PLUGIN", "PLUGIN_URL") + "/pluginToken/apply";
  
          ResponseEntity<String> tokenRes = outRestTemplate.exchange(pluginUrl, HttpMethod.POST, httpEntity, String.class);
  
          String body = tokenRes.getBody();
          JSONObject paramOut = JSONObject.parseObject(body);
  
          if (paramOut.getIntValue("code") != ResultVo.CODE_OK) {
              throw new CmdException(paramOut.getString("msg"));
          }
  
          pluginUrl = MappingCache.getValue(URL_DOMAIN, IotConstant.PLUGIN_URL);
          String targetUrl = pluginUrl + reqJson.getString("targetUrl");
          String url = pluginUrl + "/sso.html?token=" + paramOut.getString("data") + "&communityId=" + reqJson.getString("communityId") + "&targetUrl=" + targetUrl;
          paramOut.put("url", url);
          context.setResponseEntity(ResultVo.createResponseEntity(paramOut));
      }
  
      public String createSign(JSONObject staff) {
          String secure = MappingCache.getValue("PLUGIN", "PLUGIN_SECURE");
          String sign = staff.getString("staffId") + staff.getString("timestamp") + secure;
          sign = AuthenticationFactory.md5(sign);
          return sign;
      }
  }