Examples with Result com.alibaba.csp.sentinel.dashboard.domain.Result used on opensource projects

Example 1 with Result

use of com.alibaba.csp.sentinel.dashboard.domain.Result in project Sentinel by alibaba.

the class GatewayApiController method addApi.

@PostMapping("/new.json")
@AuthAction(AuthService.PrivilegeType.WRITE_RULE)
public Result<ApiDefinitionEntity> addApi(HttpServletRequest request, @RequestBody AddApiReqVo reqVo) {
    String app = reqVo.getApp();
    if (StringUtil.isBlank(app)) {
        return Result.ofFail(-1, "app can't be null or empty");
    }
    ApiDefinitionEntity entity = new ApiDefinitionEntity();
    entity.setApp(app.trim());
    String ip = reqVo.getIp();
    if (StringUtil.isBlank(ip)) {
        return Result.ofFail(-1, "ip can't be null or empty");
    }
    entity.setIp(ip.trim());
    Integer port = reqVo.getPort();
    if (port == null) {
        return Result.ofFail(-1, "port can't be null");
    }
    entity.setPort(port);
    // API名称
    String apiName = reqVo.getApiName();
    if (StringUtil.isBlank(apiName)) {
        return Result.ofFail(-1, "apiName can't be null or empty");
    }
    entity.setApiName(apiName.trim());
    // 匹配规则列表
    List<ApiPredicateItemVo> predicateItems = reqVo.getPredicateItems();
    if (CollectionUtils.isEmpty(predicateItems)) {
        return Result.ofFail(-1, "predicateItems can't empty");
    }
    List<ApiPredicateItemEntity> predicateItemEntities = new ArrayList<>();
    for (ApiPredicateItemVo predicateItem : predicateItems) {
        ApiPredicateItemEntity predicateItemEntity = new ApiPredicateItemEntity();
        // 匹配模式
        Integer matchStrategy = predicateItem.getMatchStrategy();
        if (!Arrays.asList(URL_MATCH_STRATEGY_EXACT, URL_MATCH_STRATEGY_PREFIX, URL_MATCH_STRATEGY_REGEX).contains(matchStrategy)) {
            return Result.ofFail(-1, "invalid matchStrategy: " + matchStrategy);
        }
        predicateItemEntity.setMatchStrategy(matchStrategy);
        // 匹配串
        String pattern = predicateItem.getPattern();
        if (StringUtil.isBlank(pattern)) {
            return Result.ofFail(-1, "pattern can't be null or empty");
        }
        predicateItemEntity.setPattern(pattern);
        predicateItemEntities.add(predicateItemEntity);
    }
    entity.setPredicateItems(new LinkedHashSet<>(predicateItemEntities));
    // 检查API名称不能重复
    List<ApiDefinitionEntity> allApis = repository.findAllByMachine(MachineInfo.of(app.trim(), ip.trim(), port));
    if (allApis.stream().map(o -> o.getApiName()).anyMatch(o -> o.equals(apiName.trim()))) {
        return Result.ofFail(-1, "apiName exists: " + apiName);
    }
    Date date = new Date();
    entity.setGmtCreate(date);
    entity.setGmtModified(date);
    try {
        entity = repository.save(entity);
    } catch (Throwable throwable) {
        logger.error("add gateway api error:", throwable);
        return Result.ofThrowable(-1, throwable);
    }
    if (!publishApis(app, ip, port)) {
        logger.warn("publish gateway apis fail after add");
    }
    return Result.ofSuccess(entity);
}

Example 2 with Result

use of com.alibaba.csp.sentinel.dashboard.domain.Result in project Sentinel by alibaba.

the class DefaultAuthorizationInterceptor method responseNoPrivilegeMsg.

private void responseNoPrivilegeMsg(HttpServletResponse response, String message) throws IOException {
    Result result = Result.ofFail(-1, message);
    response.addHeader("Content-Type", "application/json;charset=UTF-8");
    response.getOutputStream().write(JSON.toJSONBytes(result));
}

Example 3 with Result

use of com.alibaba.csp.sentinel.dashboard.domain.Result in project Sentinel by alibaba.

the class GatewayApiControllerTest method testQueryApis.

@Test
public void testQueryApis() throws Exception {
    String path = "/gateway/api/list.json";
    List<ApiDefinitionEntity> entities = new ArrayList<>();
    // Mock two entities
    ApiDefinitionEntity entity = new ApiDefinitionEntity();
    entity.setId(1L);
    entity.setApp(TEST_APP);
    entity.setIp(TEST_IP);
    entity.setPort(TEST_PORT);
    entity.setApiName("foo");
    Date date = new Date();
    entity.setGmtCreate(date);
    entity.setGmtModified(date);
    Set<ApiPredicateItemEntity> itemEntities = new LinkedHashSet<>();
    entity.setPredicateItems(itemEntities);
    ApiPredicateItemEntity itemEntity = new ApiPredicateItemEntity();
    itemEntity.setPattern("/aaa");
    itemEntity.setMatchStrategy(URL_MATCH_STRATEGY_EXACT);
    itemEntities.add(itemEntity);
    entities.add(entity);
    ApiDefinitionEntity entity2 = new ApiDefinitionEntity();
    entity2.setId(2L);
    entity2.setApp(TEST_APP);
    entity2.setIp(TEST_IP);
    entity2.setPort(TEST_PORT);
    entity2.setApiName("biz");
    entity.setGmtCreate(date);
    entity.setGmtModified(date);
    Set<ApiPredicateItemEntity> itemEntities2 = new LinkedHashSet<>();
    entity2.setPredicateItems(itemEntities2);
    ApiPredicateItemEntity itemEntity2 = new ApiPredicateItemEntity();
    itemEntity2.setPattern("/bbb");
    itemEntity2.setMatchStrategy(URL_MATCH_STRATEGY_PREFIX);
    itemEntities2.add(itemEntity2);
    entities.add(entity2);
    CompletableFuture<List<ApiDefinitionEntity>> completableFuture = mock(CompletableFuture.class);
    given(completableFuture.get()).willReturn(entities);
    given(sentinelApiClient.fetchApis(TEST_APP, TEST_IP, TEST_PORT)).willReturn(completableFuture);
    MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders.get(path);
    requestBuilder.param("app", TEST_APP);
    requestBuilder.param("ip", TEST_IP);
    requestBuilder.param("port", String.valueOf(TEST_PORT));
    // Do controller logic
    MvcResult mvcResult = mockMvc.perform(requestBuilder).andExpect(MockMvcResultMatchers.status().isOk()).andDo(MockMvcResultHandlers.print()).andReturn();
    // Verify the fetchApis method has been called
    verify(sentinelApiClient).fetchApis(TEST_APP, TEST_IP, TEST_PORT);
    // Verify if two same entities are got
    Result<List<ApiDefinitionEntity>> result = JSONObject.parseObject(mvcResult.getResponse().getContentAsString(), new TypeReference<Result<List<ApiDefinitionEntity>>>() {
    });
    assertTrue(result.isSuccess());
    List<ApiDefinitionEntity> data = result.getData();
    assertEquals(2, data.size());
    assertEquals(entities, data);
    // Verify the entities are add into memory repository
    List<ApiDefinitionEntity> entitiesInMem = repository.findAllByApp(TEST_APP);
    assertEquals(2, entitiesInMem.size());
    assertEquals(entities, entitiesInMem);
}

Example 4 with Result

use of com.alibaba.csp.sentinel.dashboard.domain.Result in project Sentinel by alibaba.

the class GatewayFlowRuleControllerTest method testQueryFlowRules.

@Test
public void testQueryFlowRules() throws Exception {
    String path = "/gateway/flow/list.json";
    List<GatewayFlowRuleEntity> entities = new ArrayList<>();
    // Mock two entities
    GatewayFlowRuleEntity entity = new GatewayFlowRuleEntity();
    entity.setId(1L);
    entity.setApp(TEST_APP);
    entity.setIp(TEST_IP);
    entity.setPort(TEST_PORT);
    entity.setResource("httpbin_route");
    entity.setResourceMode(RESOURCE_MODE_ROUTE_ID);
    entity.setGrade(FLOW_GRADE_QPS);
    entity.setCount(5D);
    entity.setInterval(30L);
    entity.setIntervalUnit(GatewayFlowRuleEntity.INTERVAL_UNIT_SECOND);
    entity.setControlBehavior(CONTROL_BEHAVIOR_DEFAULT);
    entity.setBurst(0);
    entity.setMaxQueueingTimeoutMs(0);
    GatewayParamFlowItemEntity itemEntity = new GatewayParamFlowItemEntity();
    entity.setParamItem(itemEntity);
    itemEntity.setParseStrategy(PARAM_PARSE_STRATEGY_CLIENT_IP);
    entities.add(entity);
    GatewayFlowRuleEntity entity2 = new GatewayFlowRuleEntity();
    entity2.setId(2L);
    entity2.setApp(TEST_APP);
    entity2.setIp(TEST_IP);
    entity2.setPort(TEST_PORT);
    entity2.setResource("some_customized_api");
    entity2.setResourceMode(RESOURCE_MODE_CUSTOM_API_NAME);
    entity2.setCount(30D);
    entity2.setInterval(2L);
    entity2.setIntervalUnit(GatewayFlowRuleEntity.INTERVAL_UNIT_MINUTE);
    entity2.setControlBehavior(CONTROL_BEHAVIOR_DEFAULT);
    entity2.setBurst(0);
    entity2.setMaxQueueingTimeoutMs(0);
    GatewayParamFlowItemEntity itemEntity2 = new GatewayParamFlowItemEntity();
    entity2.setParamItem(itemEntity2);
    itemEntity2.setParseStrategy(PARAM_PARSE_STRATEGY_CLIENT_IP);
    entities.add(entity2);
    CompletableFuture<List<GatewayFlowRuleEntity>> completableFuture = mock(CompletableFuture.class);
    given(completableFuture.get()).willReturn(entities);
    given(sentinelApiClient.fetchGatewayFlowRules(TEST_APP, TEST_IP, TEST_PORT)).willReturn(completableFuture);
    MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders.get(path);
    requestBuilder.param("app", TEST_APP);
    requestBuilder.param("ip", TEST_IP);
    requestBuilder.param("port", String.valueOf(TEST_PORT));
    // Do controller logic
    MvcResult mvcResult = mockMvc.perform(requestBuilder).andExpect(MockMvcResultMatchers.status().isOk()).andDo(MockMvcResultHandlers.print()).andReturn();
    // Verify the fetchGatewayFlowRules method has been called
    verify(sentinelApiClient).fetchGatewayFlowRules(TEST_APP, TEST_IP, TEST_PORT);
    // Verify if two same entities are got
    Result<List<GatewayFlowRuleEntity>> result = JSONObject.parseObject(mvcResult.getResponse().getContentAsString(), new TypeReference<Result<List<GatewayFlowRuleEntity>>>() {
    });
    assertTrue(result.isSuccess());
    List<GatewayFlowRuleEntity> data = result.getData();
    assertEquals(2, data.size());
    assertEquals(entities, data);
    // Verify the entities are add into memory repository
    List<GatewayFlowRuleEntity> entitiesInMem = repository.findAllByApp(TEST_APP);
    assertEquals(2, entitiesInMem.size());
    assertEquals(entities, entitiesInMem);
}

Example 5 with Result

use of com.alibaba.csp.sentinel.dashboard.domain.Result in project pig by pig-mesh.

the class GatewayApiController method addApi.

@PostMapping("/new.json")
@AuthAction(AuthService.PrivilegeType.WRITE_RULE)
public Result<ApiDefinitionEntity> addApi(HttpServletRequest request, @RequestBody AddApiReqVo reqVo) {
    String app = reqVo.getApp();
    if (StringUtil.isBlank(app)) {
        return Result.ofFail(-1, "app can't be null or empty");
    }
    ApiDefinitionEntity entity = new ApiDefinitionEntity();
    entity.setApp(app.trim());
    String ip = reqVo.getIp();
    if (StringUtil.isBlank(ip)) {
        return Result.ofFail(-1, "ip can't be null or empty");
    }
    entity.setIp(ip.trim());
    Integer port = reqVo.getPort();
    if (port == null) {
        return Result.ofFail(-1, "port can't be null");
    }
    entity.setPort(port);
    // API名称
    String apiName = reqVo.getApiName();
    if (StringUtil.isBlank(apiName)) {
        return Result.ofFail(-1, "apiName can't be null or empty");
    }
    entity.setApiName(apiName.trim());
    // 匹配规则列表
    List<ApiPredicateItemVo> predicateItems = reqVo.getPredicateItems();
    if (CollectionUtils.isEmpty(predicateItems)) {
        return Result.ofFail(-1, "predicateItems can't empty");
    }
    List<ApiPredicateItemEntity> predicateItemEntities = new ArrayList<>();
    for (ApiPredicateItemVo predicateItem : predicateItems) {
        ApiPredicateItemEntity predicateItemEntity = new ApiPredicateItemEntity();
        // 匹配模式
        Integer matchStrategy = predicateItem.getMatchStrategy();
        if (!Arrays.asList(URL_MATCH_STRATEGY_EXACT, URL_MATCH_STRATEGY_PREFIX, URL_MATCH_STRATEGY_REGEX).contains(matchStrategy)) {
            return Result.ofFail(-1, "invalid matchStrategy: " + matchStrategy);
        }
        predicateItemEntity.setMatchStrategy(matchStrategy);
        // 匹配串
        String pattern = predicateItem.getPattern();
        if (StringUtil.isBlank(pattern)) {
            return Result.ofFail(-1, "pattern can't be null or empty");
        }
        predicateItemEntity.setPattern(pattern);
        predicateItemEntities.add(predicateItemEntity);
    }
    entity.setPredicateItems(new LinkedHashSet<>(predicateItemEntities));
    // 检查API名称不能重复
    List<ApiDefinitionEntity> allApis = repository.findAllByMachine(MachineInfo.of(app.trim(), ip.trim(), port));
    if (allApis.stream().map(o -> o.getApiName()).anyMatch(o -> o.equals(apiName.trim()))) {
        return Result.ofFail(-1, "apiName exists: " + apiName);
    }
    Date date = new Date();
    entity.setGmtCreate(date);
    entity.setGmtModified(date);
    try {
        entity = repository.save(entity);
    } catch (Throwable throwable) {
        logger.error("add gateway api error:", throwable);
        return Result.ofThrowable(-1, throwable);
    }
    if (!publishApis(app, ip, port)) {
        logger.warn("publish gateway apis fail after add");
    }
    return Result.ofSuccess(entity);
}