Wenjun Ruan
1 year ago
committed by
GitHub
24 changed files with 298 additions and 298 deletions
@ -0,0 +1,90 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration; |
||||
|
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
import lombok.AllArgsConstructor; |
||||
import lombok.Data; |
||||
import lombok.NoArgsConstructor; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.validation.Errors; |
||||
import org.springframework.validation.Validator; |
||||
import org.springframework.validation.annotation.Validated; |
||||
|
||||
@Slf4j |
||||
@Data |
||||
@Validated |
||||
@Configuration |
||||
@ConfigurationProperties(value = "api") |
||||
public class ApiConfig implements Validator { |
||||
|
||||
private boolean auditEnable = false; |
||||
|
||||
private TrafficConfiguration trafficControl = new TrafficConfiguration(); |
||||
|
||||
private PythonGatewayConfiguration pythonGateway = new PythonGatewayConfiguration(); |
||||
|
||||
@Override |
||||
public boolean supports(Class<?> clazz) { |
||||
return ApiConfig.class.isAssignableFrom(clazz); |
||||
} |
||||
|
||||
@Override |
||||
public void validate(Object target, Errors errors) { |
||||
printConfig(); |
||||
} |
||||
|
||||
private void printConfig() { |
||||
log.info("API config: auditEnable -> {} ", auditEnable); |
||||
log.info("API config: trafficControl -> {} ", trafficControl); |
||||
log.info("API config: pythonGateway -> {} ", pythonGateway); |
||||
} |
||||
|
||||
@Data |
||||
@NoArgsConstructor |
||||
@AllArgsConstructor |
||||
public static class TrafficConfiguration { |
||||
|
||||
private boolean globalSwitch = false; |
||||
private Integer maxGlobalQpsRate = 300; |
||||
private boolean tenantSwitch = false; |
||||
private Integer defaultTenantQpsRate = 10; |
||||
private Map<String, Integer> customizeTenantQpsRate = new HashMap<>(); |
||||
} |
||||
|
||||
@Data |
||||
@NoArgsConstructor |
||||
@AllArgsConstructor |
||||
public static class PythonGatewayConfiguration { |
||||
|
||||
private boolean enabled = true; |
||||
private String gatewayServerAddress = "0.0.0.0"; |
||||
private int gatewayServerPort = 25333; |
||||
private String pythonAddress = "127.0.0.1"; |
||||
private int pythonPort = 25334; |
||||
private int connectTimeout = 0; |
||||
private int readTimeout = 0; |
||||
private String authToken = "jwUDzpLsNKEFER4*a8gruBH_GsAurNxU7A@Xc"; |
||||
} |
||||
|
||||
} |
||||
@ -1,38 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration; |
||||
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
import org.springframework.boot.context.properties.EnableConfigurationProperties; |
||||
import org.springframework.stereotype.Component; |
||||
|
||||
@Component |
||||
@EnableConfigurationProperties |
||||
@ConfigurationProperties(value = "audit", ignoreUnknownFields = false) |
||||
public class AuditConfiguration { |
||||
|
||||
private boolean enabled; |
||||
|
||||
public boolean getEnabled() { |
||||
return enabled; |
||||
} |
||||
|
||||
public void setEnabled(boolean enabled) { |
||||
this.enabled = enabled; |
||||
} |
||||
} |
||||
@ -1,38 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration; |
||||
|
||||
import lombok.Data; |
||||
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
import org.springframework.context.annotation.Configuration; |
||||
|
||||
@Data |
||||
@Configuration |
||||
@ConfigurationProperties(value = "python-gateway") |
||||
public class PythonGatewayConfiguration { |
||||
|
||||
private boolean enabled; |
||||
private String gatewayServerAddress; |
||||
private int gatewayServerPort; |
||||
private String pythonAddress; |
||||
private int pythonPort; |
||||
private int connectTimeout; |
||||
private int readTimeout; |
||||
private String authToken; |
||||
} |
||||
@ -1,38 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration; |
||||
|
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
import lombok.Data; |
||||
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
import org.springframework.context.annotation.Configuration; |
||||
|
||||
@Data |
||||
@Configuration |
||||
@ConfigurationProperties(prefix = "traffic.control") |
||||
public class TrafficConfiguration { |
||||
|
||||
private boolean globalSwitch; |
||||
private Integer maxGlobalQpsRate = 300; |
||||
private boolean tenantSwitch; |
||||
private Integer defaultTenantQpsRate = 10; |
||||
private Map<String, Integer> customizeTenantQpsRate = new HashMap<>(); |
||||
} |
||||
@ -1,37 +0,0 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You 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 org.apache.dolphinscheduler.api.configuration; |
||||
|
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.junit.jupiter.api.Test; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.boot.test.context.SpringBootTest; |
||||
import org.springframework.test.context.ActiveProfiles; |
||||
|
||||
@ActiveProfiles("audit") |
||||
@SpringBootTest(classes = AuditConfiguration.class) |
||||
public class AuditConfigurationTest { |
||||
|
||||
@Autowired |
||||
private AuditConfiguration auditConfiguration; |
||||
|
||||
@Test |
||||
public void isAuditGlobalControlSwitch() { |
||||
Assertions.assertTrue(auditConfiguration.getEnabled()); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue