forked from community/device-mgt-core
osh
2 years ago
commit
36052663ba
@ -0,0 +1,41 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<parent>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>carbon-devicemgt</artifactId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>task-mgt</artifactId>
|
||||
<packaging>pom</packaging>
|
||||
<name>Entgra IoT - Task Management Component</name>
|
||||
<url>http://entgra.io</url>
|
||||
|
||||
<modules>
|
||||
<module>task-manager</module>
|
||||
<module>task-watcher</module>
|
||||
</modules>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,56 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<parent>
|
||||
<artifactId>task-manager</artifactId>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>io.entgra.task.mgt.common</artifactId>
|
||||
<packaging>bundle</packaging>
|
||||
<name>Entgra IoT - Task Management Common</name>
|
||||
<description>Entgra IoT - Task Management Common</description>
|
||||
<url>https://entgra.io</url>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.felix</groupId>
|
||||
<artifactId>maven-bundle-plugin</artifactId>
|
||||
<extensions>true</extensions>
|
||||
<configuration>
|
||||
<instructions>
|
||||
<Bundle-SymbolicName>${project.artifactId}</Bundle-SymbolicName>
|
||||
<Bundle-Name>${project.artifactId}</Bundle-Name>
|
||||
<Bundle-Version>${carbon.device.mgt.version}</Bundle-Version>
|
||||
<Bundle-Description>Task Management Common Bundle</Bundle-Description>
|
||||
<Export-Package>
|
||||
io.entgra.task.mgt.common.*
|
||||
</Export-Package>
|
||||
</instructions>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
</project>
|
||||
@ -0,0 +1,88 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.bean;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
public class DynamicTask {
|
||||
|
||||
private int dynamicTaskId;
|
||||
private String name;
|
||||
private String cronExpression;
|
||||
private boolean isEnabled;
|
||||
private int tenantId;
|
||||
private String taskClassName;
|
||||
private Map<String, String> properties;
|
||||
|
||||
public int getDynamicTaskId() {
|
||||
return dynamicTaskId;
|
||||
}
|
||||
|
||||
public void setDynamicTaskId(int dynamicTaskId) {
|
||||
this.dynamicTaskId = dynamicTaskId;
|
||||
}
|
||||
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
public void setName(String name) {
|
||||
this.name = name;
|
||||
}
|
||||
|
||||
public String getCronExpression() {
|
||||
return cronExpression;
|
||||
}
|
||||
|
||||
public void setCronExpression(String cronExpression) {
|
||||
this.cronExpression = cronExpression;
|
||||
}
|
||||
|
||||
public boolean isEnabled() {
|
||||
return isEnabled;
|
||||
}
|
||||
|
||||
public void setEnabled(boolean enable) {
|
||||
isEnabled = enable;
|
||||
}
|
||||
|
||||
public int getTenantId() {
|
||||
return tenantId;
|
||||
}
|
||||
|
||||
public void setTenantId(int tenantId) {
|
||||
this.tenantId = tenantId;
|
||||
}
|
||||
|
||||
public String getTaskClassName() {
|
||||
return taskClassName;
|
||||
}
|
||||
|
||||
public void setTaskClassName(String taskClassName) {
|
||||
this.taskClassName = taskClassName;
|
||||
}
|
||||
|
||||
public Map<String, String> getProperties() {
|
||||
return properties;
|
||||
}
|
||||
|
||||
public void setProperties(Map<String, String> properties) {
|
||||
this.properties = properties;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,53 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.constant;
|
||||
|
||||
public class TaskMgtConstants {
|
||||
public static final class DataSourceProperties {
|
||||
private DataSourceProperties() {
|
||||
throw new AssertionError();
|
||||
}
|
||||
|
||||
public static final String DB_CHECK_QUERY = "SELECT * FROM DM_DEVICE";
|
||||
public static final String TASK_CONFIG_XML_NAME = "task-mgt-config.xml";
|
||||
}
|
||||
|
||||
public static final class DataBaseTypes {
|
||||
private DataBaseTypes() {
|
||||
throw new AssertionError();
|
||||
}
|
||||
|
||||
public static final String DB_TYPE_MYSQL = "MySQL";
|
||||
public static final String DB_TYPE_ORACLE = "Oracle";
|
||||
public static final String DB_TYPE_MSSQL = "Microsoft SQL Server";
|
||||
public static final String DB_TYPE_DB2 = "DB2";
|
||||
public static final String DB_TYPE_H2 = "H2";
|
||||
public static final String DB_TYPE_POSTGRESQL = "PostgreSQL";
|
||||
}
|
||||
|
||||
public static final class Task {
|
||||
|
||||
public static final String DYNAMIC_TASK_TYPE = "DYNAMIC_TASK";
|
||||
public static final String NAME_SEPARATOR = "_";
|
||||
public static final String PROPERTY_KEY_COLUMN_NAME = "PROPERTY_NAME";
|
||||
public static final String PROPERTY_VALUE_COLUMN_NAME = "PROPERTY_VALUE";
|
||||
public static final String TENANT_ID_PROP = "__TENANT_ID_PROP__";
|
||||
public static final String LOCAL_HASH_INDEX = "__LOCAL_HASH_INDEX__";
|
||||
public static final String LOCAL_TASK_NAME = "__LOCAL_TASK_NAME__";
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
public class IllegalTransactionStateException extends RuntimeException {
|
||||
|
||||
private static final long serialVersionUID = -3151279331929070297L;
|
||||
|
||||
public IllegalTransactionStateException(String msg, Exception nestedEx) {
|
||||
super(msg, nestedEx);
|
||||
}
|
||||
|
||||
public IllegalTransactionStateException(String message, Throwable cause) {
|
||||
super(message, cause);
|
||||
}
|
||||
|
||||
public IllegalTransactionStateException(String msg) {
|
||||
super(msg);
|
||||
}
|
||||
|
||||
public IllegalTransactionStateException() {
|
||||
super();
|
||||
}
|
||||
|
||||
public IllegalTransactionStateException(Throwable cause) {
|
||||
super(cause);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,31 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
public class TaskManagementDAOException extends Exception {
|
||||
|
||||
public TaskManagementDAOException(String msg) {
|
||||
super(msg);
|
||||
}
|
||||
|
||||
public TaskManagementDAOException(String msg, Exception e) {
|
||||
super(msg, e);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
@ -0,0 +1,29 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
public class TaskManagementException extends Exception {
|
||||
|
||||
public TaskManagementException(String msg) {
|
||||
super(msg);
|
||||
}
|
||||
|
||||
public TaskManagementException(String msg, Exception e) {
|
||||
super(msg, e);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,33 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
/**
|
||||
* Represents the exception thrown during validating the request.
|
||||
*/
|
||||
public class TaskNotFoundException extends Exception {
|
||||
|
||||
public TaskNotFoundException(String message) {
|
||||
super(message);
|
||||
}
|
||||
|
||||
public TaskNotFoundException(String message, Exception ex) {
|
||||
super(message, ex);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
public class TransactionManagementException extends Exception {
|
||||
|
||||
private static final long serialVersionUID = -3151279321929070297L;
|
||||
|
||||
public TransactionManagementException(String msg, Exception nestedEx) {
|
||||
super(msg, nestedEx);
|
||||
}
|
||||
|
||||
public TransactionManagementException(String message, Throwable cause) {
|
||||
super(message, cause);
|
||||
}
|
||||
|
||||
public TransactionManagementException(String msg) {
|
||||
super(msg);
|
||||
}
|
||||
|
||||
public TransactionManagementException() {
|
||||
super();
|
||||
}
|
||||
|
||||
public TransactionManagementException(Throwable cause) {
|
||||
super(cause);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,47 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.exception;
|
||||
|
||||
/**
|
||||
* This runtime exception will be thrown if the server has configured with unsupported DB engine.
|
||||
*/
|
||||
public class UnsupportedDatabaseEngineException extends RuntimeException {
|
||||
|
||||
private static final long serialVersionUID = -3151279311929070297L;
|
||||
|
||||
public UnsupportedDatabaseEngineException(String msg, Exception nestedEx) {
|
||||
super(msg, nestedEx);
|
||||
}
|
||||
|
||||
public UnsupportedDatabaseEngineException(String message, Throwable cause) {
|
||||
super(message, cause);
|
||||
}
|
||||
|
||||
public UnsupportedDatabaseEngineException(String msg) {
|
||||
super(msg);
|
||||
}
|
||||
|
||||
public UnsupportedDatabaseEngineException() {
|
||||
super();
|
||||
}
|
||||
|
||||
public UnsupportedDatabaseEngineException(Throwable cause) {
|
||||
super(cause);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,43 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.common.spi;
|
||||
|
||||
import io.entgra.task.mgt.common.exception.TaskNotFoundException;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementException;
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface TaskManagementService {
|
||||
|
||||
void init() throws TaskManagementException;
|
||||
|
||||
void createTask(DynamicTask dynamicTask) throws TaskManagementException;
|
||||
|
||||
void updateTask(int dynamicTaskId, DynamicTask dynamicTask) throws TaskManagementException, TaskNotFoundException;
|
||||
|
||||
void toggleTask(int dynamicTaskId, boolean isEnabled) throws TaskManagementException, TaskNotFoundException;
|
||||
|
||||
void deleteTask(int dynamicTaskId) throws TaskManagementException, TaskNotFoundException;
|
||||
|
||||
List<DynamicTask> getAllDynamicTasks() throws TaskManagementException;
|
||||
|
||||
DynamicTask getDynamicTaskById(int dynamicTaskId) throws TaskManagementException;
|
||||
|
||||
List<DynamicTask> getActiveDynamicTasks() throws TaskManagementException;
|
||||
}
|
||||
@ -0,0 +1,141 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
|
||||
<parent>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>task-manager</artifactId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>io.entgra.task.mgt.core</artifactId>
|
||||
<packaging>bundle</packaging>
|
||||
<name>Entgra IoT - Task manager Core</name>
|
||||
<description>Entgra IoT - Task manager Core</description>
|
||||
<url>http://entgra.io</url>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.felix</groupId>
|
||||
<artifactId>maven-scr-plugin</artifactId>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<groupId>org.apache.felix</groupId>
|
||||
<artifactId>maven-bundle-plugin</artifactId>
|
||||
<extensions>true</extensions>
|
||||
<configuration>
|
||||
<instructions>
|
||||
<Bundle-SymbolicName>${project.artifactId}</Bundle-SymbolicName>
|
||||
<Bundle-Name>${project.artifactId}</Bundle-Name>
|
||||
<Bundle-Version>${carbon.device.mgt.version}</Bundle-Version>
|
||||
<Bundle-Description>Task Management Core Bundle</Bundle-Description>
|
||||
<Private-Package>io.entgra.task.mgt.core.internal</Private-Package>
|
||||
<Import-Package>
|
||||
org.osgi.framework.*;version="${imp.package.version.osgi.framework}",
|
||||
org.osgi.service.*;version="${imp.package.version.osgi.service}",
|
||||
org.apache.commons.logging,
|
||||
org.wso2.carbon.ndatasource.core,
|
||||
org.w3c.dom,
|
||||
javax.xml.bind.annotation,
|
||||
javax.xml.bind,
|
||||
javax.sql,
|
||||
javax.naming,
|
||||
io.entgra.task.mgt.common.*,
|
||||
org.wso2.carbon.utils.*,
|
||||
org.wso2.carbon.ntask.core.*,
|
||||
org.wso2.carbon.ntask.common,
|
||||
org.wso2.carbon.device.mgt.common.*,
|
||||
org.wso2.carbon.context,
|
||||
org.apache.commons.codec.binary;version="${commons-codec.wso2.osgi.version.range}",
|
||||
org.apache.commons.codec.digest;version="${commons-codec.wso2.osgi.version.range}",
|
||||
io.entgra.server.bootup.heartbeat.beacon.dto,
|
||||
io.entgra.server.bootup.heartbeat.beacon.exception,
|
||||
io.entgra.server.bootup.heartbeat.beacon.service,
|
||||
</Import-Package>
|
||||
<Export-Package>
|
||||
!io.entgra.task.mgt.core.internal,
|
||||
io.entgra.task.mgt.core.*
|
||||
</Export-Package>
|
||||
</instructions>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.eclipse.osgi</groupId>
|
||||
<artifactId>org.eclipse.osgi</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.eclipse.osgi</groupId>
|
||||
<artifactId>org.eclipse.osgi.services</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.ndatasource.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>io.entgra.task.mgt.common</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.logging</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.utils</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>commons-codec.wso2</groupId>
|
||||
<artifactId>commons-codec</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>org.wso2.carbon.device.mgt.common</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>io.entgra.server.bootup.heartbeat.beacon</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
|
||||
<!--nTask dependencies-->
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.commons</groupId>
|
||||
<artifactId>org.wso2.carbon.ntask.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,79 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.config;
|
||||
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementException;
|
||||
import io.entgra.task.mgt.core.util.TaskManagementUtil;
|
||||
import org.w3c.dom.Document;
|
||||
import org.wso2.carbon.utils.CarbonUtils;
|
||||
|
||||
import javax.xml.bind.JAXBContext;
|
||||
import javax.xml.bind.JAXBException;
|
||||
import javax.xml.bind.Unmarshaller;
|
||||
import java.io.File;
|
||||
|
||||
/**
|
||||
* Class responsible for the task mgt configuration initialization.
|
||||
*/
|
||||
public class TaskConfigurationManager {
|
||||
|
||||
private TaskManagementConfig taskManagementConfig;
|
||||
private static volatile TaskConfigurationManager taskConfigurationManager;
|
||||
|
||||
private static final String TASK_MGT_CONFIG_PATH =
|
||||
CarbonUtils.getCarbonConfigDirPath() + File.separator +
|
||||
TaskMgtConstants.DataSourceProperties.TASK_CONFIG_XML_NAME;
|
||||
|
||||
public static TaskConfigurationManager getInstance() {
|
||||
if (taskConfigurationManager == null) {
|
||||
synchronized (TaskConfigurationManager.class) {
|
||||
if (taskConfigurationManager == null) {
|
||||
taskConfigurationManager = new TaskConfigurationManager();
|
||||
}
|
||||
}
|
||||
}
|
||||
return taskConfigurationManager;
|
||||
}
|
||||
|
||||
public synchronized void initConfig() throws TaskManagementException {
|
||||
try {
|
||||
File taskMgtConfig = new File(TASK_MGT_CONFIG_PATH);
|
||||
Document doc = TaskManagementUtil.convertToDocument(taskMgtConfig);
|
||||
|
||||
/* Un-marshaling Device Management configuration */
|
||||
JAXBContext cdmContext = JAXBContext.newInstance(TaskManagementConfig.class);
|
||||
Unmarshaller unmarshaller = cdmContext.createUnmarshaller();
|
||||
//unmarshaller.setSchema(getSchema());
|
||||
this.taskManagementConfig = (TaskManagementConfig) unmarshaller.unmarshal(doc);
|
||||
} catch (JAXBException e) {
|
||||
throw new TaskManagementException("Error occurred while initializing Data Source config", e);
|
||||
}
|
||||
}
|
||||
|
||||
public TaskManagementConfig getTaskManagementConfig() throws TaskManagementException {
|
||||
if (taskManagementConfig == null) {
|
||||
initConfig();
|
||||
}
|
||||
return taskManagementConfig;
|
||||
}
|
||||
|
||||
public void setTaskManagementConfig(TaskManagementConfig taskManagementConfig) {
|
||||
this.taskManagementConfig = taskManagementConfig;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.config;
|
||||
|
||||
import javax.xml.bind.annotation.XmlElement;
|
||||
import javax.xml.bind.annotation.XmlRootElement;
|
||||
|
||||
/**
|
||||
* Represents Task Mgt configuration.
|
||||
*/
|
||||
@XmlRootElement(name = "TaskMgtConfiguration")
|
||||
@SuppressWarnings("unused")
|
||||
public final class TaskManagementConfig {
|
||||
|
||||
private TaskManagementConfigRepository taskMgtConfigRepository;
|
||||
private boolean isTaskWatcherEnabled;
|
||||
|
||||
@XmlElement(name = "ManagementRepository", required = true)
|
||||
public TaskManagementConfigRepository getTaskMgtConfigRepository() {
|
||||
return taskMgtConfigRepository;
|
||||
}
|
||||
|
||||
public void setTaskMgtConfigRepository(TaskManagementConfigRepository taskMgtConfigRepository) {
|
||||
this.taskMgtConfigRepository = taskMgtConfigRepository;
|
||||
}
|
||||
|
||||
@XmlElement(name = "TaskWatcherEnable", required = true)
|
||||
public boolean isTaskWatcherEnabled() {
|
||||
return isTaskWatcherEnabled;
|
||||
}
|
||||
|
||||
public void setTaskWatcherEnabled(boolean enabled) {
|
||||
this.isTaskWatcherEnabled = enabled;
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,42 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.config;
|
||||
|
||||
import io.entgra.task.mgt.core.config.datasource.DataSourceConfig;
|
||||
|
||||
import javax.xml.bind.annotation.XmlElement;
|
||||
import javax.xml.bind.annotation.XmlRootElement;
|
||||
|
||||
/**
|
||||
* Class for holding management repository data.
|
||||
*/
|
||||
@XmlRootElement(name = "ManagementRepository")
|
||||
public class TaskManagementConfigRepository {
|
||||
|
||||
private DataSourceConfig dataSourceConfig;
|
||||
|
||||
@XmlElement(name = "DataSourceConfiguration", required = true)
|
||||
public DataSourceConfig getDataSourceConfig() {
|
||||
return dataSourceConfig;
|
||||
}
|
||||
|
||||
public void setDataSourceConfig(DataSourceConfig dataSourceConfig) {
|
||||
this.dataSourceConfig = dataSourceConfig;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.config.datasource;
|
||||
|
||||
import javax.xml.bind.annotation.XmlElement;
|
||||
import javax.xml.bind.annotation.XmlRootElement;
|
||||
|
||||
/**
|
||||
* Class for holding data source configuration in task-mgt-config.xml at parsing with JAXB.
|
||||
*/
|
||||
@XmlRootElement(name = "DataSourceConfiguration")
|
||||
public class DataSourceConfig {
|
||||
|
||||
private JNDILookupDefinition jndiLookupDefinition;
|
||||
|
||||
@XmlElement(name = "JndiLookupDefinition", required = true)
|
||||
public JNDILookupDefinition getJndiLookupDefinition() {
|
||||
return jndiLookupDefinition;
|
||||
}
|
||||
|
||||
public void setJndiLookupDefinition(JNDILookupDefinition jndiLookupDefinition) {
|
||||
this.jndiLookupDefinition = jndiLookupDefinition;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,79 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.config.datasource;
|
||||
|
||||
import javax.xml.bind.annotation.*;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* Class for hold JndiLookupDefinition of task-mgt-config.xml at parsing with JAXB.
|
||||
*/
|
||||
@XmlRootElement(name = "JndiLookupDefinition")
|
||||
public class JNDILookupDefinition {
|
||||
|
||||
private String jndiName;
|
||||
private List<JNDIProperty> jndiProperties;
|
||||
|
||||
@XmlElement(name = "Name", required = false)
|
||||
public String getJndiName() {
|
||||
return jndiName;
|
||||
}
|
||||
|
||||
public void setJndiName(String jndiName) {
|
||||
this.jndiName = jndiName;
|
||||
}
|
||||
|
||||
@XmlElementWrapper(name = "Environment", required = false)
|
||||
@XmlElement(name = "Property", nillable = false)
|
||||
public List<JNDIProperty> getJndiProperties() {
|
||||
return jndiProperties;
|
||||
}
|
||||
|
||||
public void setJndiProperties(List<JNDIProperty> jndiProperties) {
|
||||
this.jndiProperties = jndiProperties;
|
||||
}
|
||||
|
||||
@XmlRootElement(name = "Property")
|
||||
public static class JNDIProperty {
|
||||
|
||||
private String name;
|
||||
|
||||
private String value;
|
||||
|
||||
@XmlAttribute(name = "Name")
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
public void setName(String name) {
|
||||
this.name = name;
|
||||
}
|
||||
|
||||
@XmlValue
|
||||
public String getValue() {
|
||||
return value;
|
||||
}
|
||||
|
||||
public void setValue(String value) {
|
||||
this.value = value;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@ -0,0 +1,42 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao;
|
||||
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementDAOException;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* This class represents the key operations associated with dynamic tasks.
|
||||
*/
|
||||
public interface DynamicTaskDAO {
|
||||
|
||||
int addTask(DynamicTask dynamicTask) throws TaskManagementDAOException;
|
||||
|
||||
boolean updateDynamicTask(DynamicTask dynamicTask) throws TaskManagementDAOException;
|
||||
|
||||
void deleteDynamicTask(int dynamicTaskId) throws TaskManagementDAOException;
|
||||
|
||||
DynamicTask getDynamicTaskById(int dynamicTaskId) throws TaskManagementDAOException;
|
||||
|
||||
List<DynamicTask> getAllDynamicTasks() throws TaskManagementDAOException;
|
||||
|
||||
List<DynamicTask> getActiveDynamicTasks() throws TaskManagementDAOException;
|
||||
|
||||
}
|
||||
@ -0,0 +1,35 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao;
|
||||
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementDAOException;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* This class represents the key operations associated with dynamic task properties.
|
||||
*/
|
||||
public interface DynamicTaskPropDAO {
|
||||
|
||||
void addTaskProperties(int dynamicTaskId, Map<String, String> properties) throws TaskManagementDAOException;
|
||||
|
||||
Map<String, String> getDynamicTaskProps(int dynamicTaskId) throws TaskManagementDAOException;
|
||||
|
||||
void updateDynamicTaskProps(int dynamicTaskId, Map<String, String> properties)
|
||||
throws TaskManagementDAOException;
|
||||
}
|
||||
@ -0,0 +1,206 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao.common;
|
||||
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.IllegalTransactionStateException;
|
||||
import io.entgra.task.mgt.common.exception.TransactionManagementException;
|
||||
import io.entgra.task.mgt.common.exception.UnsupportedDatabaseEngineException;
|
||||
import io.entgra.task.mgt.core.config.datasource.DataSourceConfig;
|
||||
import io.entgra.task.mgt.core.config.datasource.JNDILookupDefinition;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskDAO;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskPropDAO;
|
||||
import io.entgra.task.mgt.core.dao.impl.DynamicTaskDAOImpl;
|
||||
import io.entgra.task.mgt.core.dao.impl.DynamicTaskPropDAOImpl;
|
||||
import io.entgra.task.mgt.core.dao.util.TaskManagementDAOUtil;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
|
||||
import javax.sql.DataSource;
|
||||
import java.sql.Connection;
|
||||
import java.sql.SQLException;
|
||||
import java.util.Hashtable;
|
||||
import java.util.List;
|
||||
|
||||
public class TaskManagementDAOFactory {
|
||||
|
||||
private static DataSource dataSource;
|
||||
private static String databaseEngine;
|
||||
private static final Log log = LogFactory.getLog(TaskManagementDAOFactory.class);
|
||||
private static ThreadLocal<Connection> currentConnection = new ThreadLocal<>();
|
||||
|
||||
|
||||
public static DynamicTaskDAO getDynamicTaskDAO() {
|
||||
if (databaseEngine != null) {
|
||||
switch (databaseEngine) {
|
||||
case TaskMgtConstants.DataBaseTypes.DB_TYPE_H2:
|
||||
case TaskMgtConstants.DataBaseTypes.DB_TYPE_MYSQL:
|
||||
return new DynamicTaskDAOImpl();
|
||||
default:
|
||||
throw new UnsupportedDatabaseEngineException("Unsupported database engine : " + databaseEngine);
|
||||
}
|
||||
}
|
||||
throw new IllegalStateException("Database engine has not initialized properly.");
|
||||
}
|
||||
|
||||
public static DynamicTaskPropDAO getDynamicTaskPropDAO() {
|
||||
if (databaseEngine != null) {
|
||||
switch (databaseEngine) {
|
||||
case TaskMgtConstants.DataBaseTypes.DB_TYPE_H2:
|
||||
case TaskMgtConstants.DataBaseTypes.DB_TYPE_MYSQL:
|
||||
return new DynamicTaskPropDAOImpl();
|
||||
default:
|
||||
throw new UnsupportedDatabaseEngineException("Unsupported database engine : " + databaseEngine);
|
||||
}
|
||||
}
|
||||
throw new IllegalStateException("Database engine has not initialized properly.");
|
||||
}
|
||||
|
||||
public static void init(DataSourceConfig config) {
|
||||
dataSource = resolveDataSource(config);
|
||||
try {
|
||||
databaseEngine = dataSource.getConnection().getMetaData().getDatabaseProductName();
|
||||
} catch (SQLException e) {
|
||||
log.error("Error occurred while retrieving config.datasource connection", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static void init(DataSource dtSource) {
|
||||
dataSource = dtSource;
|
||||
try {
|
||||
databaseEngine = dataSource.getConnection().getMetaData().getDatabaseProductName();
|
||||
} catch (SQLException e) {
|
||||
log.error("Error occurred while retrieving config.datasource connection", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static void beginTransaction() throws TransactionManagementException {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn != null) {
|
||||
throw new IllegalTransactionStateException("A transaction is already active within the context of " +
|
||||
"this particular thread. Therefore, calling 'beginTransaction/openConnection' while another " +
|
||||
"transaction is already active is a sign of improper transaction handling");
|
||||
}
|
||||
try {
|
||||
conn = dataSource.getConnection();
|
||||
conn.setAutoCommit(false);
|
||||
currentConnection.set(conn);
|
||||
} catch (SQLException e) {
|
||||
throw new TransactionManagementException("Error occurred while retrieving config.datasource connection", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static void openConnection() throws SQLException {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn != null) {
|
||||
throw new IllegalTransactionStateException("A transaction is already active within the context of " +
|
||||
"this particular thread. Therefore, calling 'beginTransaction/openConnection' while another " +
|
||||
"transaction is already active is a sign of improper transaction handling");
|
||||
}
|
||||
conn = dataSource.getConnection();
|
||||
currentConnection.set(conn);
|
||||
}
|
||||
|
||||
public static Connection getConnection() throws SQLException {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn == null) {
|
||||
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
||||
"This might have ideally been caused by not properly initiating the transaction via " +
|
||||
"'beginTransaction'/'openConnection' methods");
|
||||
}
|
||||
return conn;
|
||||
}
|
||||
|
||||
public static void commitTransaction() {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn == null) {
|
||||
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
||||
"This might have ideally been caused by not properly initiating the transaction via " +
|
||||
"'beginTransaction'/'openConnection' methods");
|
||||
}
|
||||
try {
|
||||
conn.commit();
|
||||
} catch (SQLException e) {
|
||||
log.error("Error occurred while committing the transaction", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static void rollbackTransaction() {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn == null) {
|
||||
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
||||
"This might have ideally been caused by not properly initiating the transaction via " +
|
||||
"'beginTransaction'/'openConnection' methods");
|
||||
}
|
||||
try {
|
||||
conn.rollback();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while roll-backing the transaction", e);
|
||||
}
|
||||
}
|
||||
|
||||
public static void closeConnection() {
|
||||
Connection conn = currentConnection.get();
|
||||
if (conn == null) {
|
||||
throw new IllegalTransactionStateException("No connection is associated with the current transaction. " +
|
||||
"This might have ideally been caused by not properly initiating the transaction via " +
|
||||
"'beginTransaction'/'openConnection' methods");
|
||||
}
|
||||
try {
|
||||
conn.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while close the connection");
|
||||
}
|
||||
currentConnection.remove();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Resolve data source from the data source definition
|
||||
*
|
||||
* @param config data source configuration
|
||||
* @return data source resolved from the data source definition
|
||||
*/
|
||||
private static DataSource resolveDataSource(DataSourceConfig config) {
|
||||
DataSource dataSource = null;
|
||||
if (config == null) {
|
||||
throw new RuntimeException(
|
||||
"Device Management Repository data source configuration " + "is null and " +
|
||||
"thus, is not initialized");
|
||||
}
|
||||
io.entgra.task.mgt.core.config.datasource.JNDILookupDefinition jndiConfig = config.getJndiLookupDefinition();
|
||||
if (jndiConfig != null) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Initializing Device Management Repository data source using the JNDI " +
|
||||
"Lookup Definition");
|
||||
}
|
||||
List<JNDILookupDefinition.JNDIProperty> jndiPropertyList =
|
||||
jndiConfig.getJndiProperties();
|
||||
if (jndiPropertyList != null) {
|
||||
Hashtable<Object, Object> jndiProperties = new Hashtable<Object, Object>();
|
||||
for (JNDILookupDefinition.JNDIProperty prop : jndiPropertyList) {
|
||||
jndiProperties.put(prop.getName(), prop.getValue());
|
||||
}
|
||||
dataSource = TaskManagementDAOUtil.lookupDataSource(jndiConfig.getJndiName(), jndiProperties);
|
||||
} else {
|
||||
dataSource = TaskManagementDAOUtil.lookupDataSource(jndiConfig.getJndiName(), null);
|
||||
}
|
||||
}
|
||||
return dataSource;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,176 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao.impl;
|
||||
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementDAOException;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskDAO;
|
||||
import io.entgra.task.mgt.core.dao.common.TaskManagementDAOFactory;
|
||||
import io.entgra.task.mgt.core.dao.util.TaskManagementDAOUtil;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
||||
|
||||
import java.sql.*;
|
||||
import java.util.List;
|
||||
|
||||
|
||||
public class DynamicTaskDAOImpl implements DynamicTaskDAO {
|
||||
private static final Log log = LogFactory.getLog(DynamicTaskDAOImpl.class);
|
||||
|
||||
@Override
|
||||
public int addTask(DynamicTask dynamicTask) throws TaskManagementDAOException {
|
||||
PreparedStatement stmt = null;
|
||||
ResultSet rs = null;
|
||||
int taskId = -1;
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
String sql = "INSERT INTO DYNAMIC_TASK(CRON, NAME, IS_ENABLED, TASK_CLASS_NAME, TENANT_ID) " +
|
||||
"VALUES (?, ?, ?, ?, ?)";
|
||||
|
||||
stmt = conn.prepareStatement(sql, new String[]{"DYNAMIC_TASK_ID"});
|
||||
stmt.setString(1, dynamicTask.getCronExpression());
|
||||
stmt.setString(2, dynamicTask.getName());
|
||||
stmt.setBoolean(3, dynamicTask.isEnabled());
|
||||
stmt.setString(4, dynamicTask.getTaskClassName());
|
||||
stmt.setInt(5, PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId());
|
||||
stmt.executeUpdate();
|
||||
|
||||
rs = stmt.getGeneratedKeys();
|
||||
if (rs.next()) {
|
||||
taskId = rs.getInt(1);
|
||||
}
|
||||
return taskId;
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while inserting task '" + dynamicTask.getName() + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOUtil.cleanupResources(stmt, null);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean updateDynamicTask(DynamicTask dynamicTask) throws TaskManagementDAOException {
|
||||
PreparedStatement stmt = null;
|
||||
int rows;
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
String sql = "UPDATE DYNAMIC_TASK SET CRON = ?,IS_ENABLED = ? WHERE DYNAMIC_TASK_ID = ?";
|
||||
stmt = conn.prepareStatement(sql);
|
||||
stmt.setString(1, dynamicTask.getCronExpression());
|
||||
stmt.setBoolean(2, dynamicTask.isEnabled());
|
||||
stmt.setInt(3, dynamicTask.getDynamicTaskId());
|
||||
rows = stmt.executeUpdate();
|
||||
return (rows > 0);
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while updating dynamic task '" + dynamicTask.getDynamicTaskId() + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOUtil.cleanupResources(stmt, null);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void deleteDynamicTask(int dynamicTaskId) throws TaskManagementDAOException {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Request received in DAO Layer to delete dynamic task with the id: " + dynamicTaskId);
|
||||
}
|
||||
String sql = "DELETE FROM DYNAMIC_TASK WHERE DYNAMIC_TASK_ID = ? AND TENANT_ID = ?";
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
try (PreparedStatement stmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {
|
||||
stmt.setInt(1, dynamicTaskId);
|
||||
stmt.setInt(2, PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId());
|
||||
stmt.executeUpdate();
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while executing SQL to delete a dynamic task which has the id " +
|
||||
dynamicTaskId;
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public DynamicTask getDynamicTaskById(int dynamicTaskId) throws TaskManagementDAOException {
|
||||
DynamicTask dynamicTask = null;
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
String sql = "SELECT * FROM DYNAMIC_TASK WHERE DYNAMIC_TASK_ID= ? AND TENANT_ID = ?";
|
||||
|
||||
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||
stmt.setInt(1, dynamicTaskId);
|
||||
stmt.setInt(2, PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId());
|
||||
try (ResultSet rs = stmt.executeQuery()) {
|
||||
if (rs.next()) {
|
||||
dynamicTask = TaskManagementDAOUtil.loadDynamicTask(rs);
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while getting dynamic task data for task identifier '" +
|
||||
dynamicTask + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
}
|
||||
return dynamicTask;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<DynamicTask> getAllDynamicTasks() throws TaskManagementDAOException {
|
||||
List<DynamicTask> dynamicTasks = null;
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
String sql = "SELECT * FROM DYNAMIC_TASK";
|
||||
|
||||
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||
try (ResultSet rs = stmt.executeQuery()) {
|
||||
dynamicTasks = TaskManagementDAOUtil.loadDynamicTasks(rs);
|
||||
}
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while getting all dynamic task data ";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
}
|
||||
return dynamicTasks;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<DynamicTask> getActiveDynamicTasks() throws TaskManagementDAOException {
|
||||
List<DynamicTask> dynamicTasks = null;
|
||||
try {
|
||||
Connection conn = TaskManagementDAOFactory.getConnection();
|
||||
String sql = "SELECT * FROM DYNAMIC_TASK WHERE IS_ENABLED = 'true' ";
|
||||
|
||||
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||
try (ResultSet rs = stmt.executeQuery()) {
|
||||
dynamicTasks = TaskManagementDAOUtil.loadDynamicTasks(rs);
|
||||
}
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while getting all dynamic task data ";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
}
|
||||
return dynamicTasks;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,124 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao.impl;
|
||||
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementDAOException;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskPropDAO;
|
||||
import io.entgra.task.mgt.core.dao.common.TaskManagementDAOFactory;
|
||||
import io.entgra.task.mgt.core.dao.util.TaskManagementDAOUtil;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
public class DynamicTaskPropDAOImpl implements DynamicTaskPropDAO {
|
||||
|
||||
private static final Log log = LogFactory.getLog(DynamicTaskPropDAOImpl.class);
|
||||
|
||||
@Override
|
||||
public void addTaskProperties(int taskId, Map<String, String> properties)
|
||||
throws TaskManagementDAOException {
|
||||
Connection conn = null;
|
||||
PreparedStatement stmt = null;
|
||||
try {
|
||||
conn = TaskManagementDAOFactory.getConnection();
|
||||
stmt = conn.prepareStatement(
|
||||
"INSERT INTO DYNAMIC_TASK_PROPERTIES(DYNAMIC_TASK_ID, PROPERTY_NAME, " +
|
||||
"PROPERTY_VALUE, TENANT_ID) VALUES (?, ?, ?, ?)");
|
||||
for (String propertyKey : properties.keySet()) {
|
||||
stmt.setInt(1, taskId);
|
||||
stmt.setString(2, propertyKey);
|
||||
stmt.setString(3, properties.get(propertyKey));
|
||||
stmt.setInt(4, PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId());
|
||||
stmt.addBatch();
|
||||
}
|
||||
stmt.executeBatch();
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while adding task properties of task '" + taskId + "' to the db.";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOUtil.cleanupResources(stmt, null);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public Map<String, String> getDynamicTaskProps(int dynamicTaskId) throws TaskManagementDAOException {
|
||||
Connection conn = null;
|
||||
PreparedStatement stmt = null;
|
||||
ResultSet resultSet = null;
|
||||
Map<String, String> properties;
|
||||
try {
|
||||
conn = TaskManagementDAOFactory.getConnection();
|
||||
stmt = conn.prepareStatement(
|
||||
"SELECT * FROM DYNAMIC_TASK_PROPERTIES WHERE DYNAMIC_TASK_ID = ?");
|
||||
stmt.setInt(1, dynamicTaskId);
|
||||
resultSet = stmt.executeQuery();
|
||||
properties = new HashMap<>();
|
||||
while (resultSet.next()) {
|
||||
properties.put(resultSet.getString(TaskMgtConstants.Task.PROPERTY_KEY_COLUMN_NAME)
|
||||
, resultSet.getString(TaskMgtConstants.Task.PROPERTY_VALUE_COLUMN_NAME));
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
String msg = "Error occurred while fetching task properties of : '" + dynamicTaskId + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementDAOException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOUtil.cleanupResources(stmt, resultSet);
|
||||
}
|
||||
return properties;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void updateDynamicTaskProps(int dynamicTaskId, Map<String, String> properties)
|
||||
throws TaskManagementDAOException {
|
||||
if (properties.isEmpty()) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Property map of task id :" + dynamicTaskId + " is empty.");
|
||||
}
|
||||
return;
|
||||
}
|
||||
Connection conn;
|
||||
PreparedStatement stmt = null;
|
||||
try {
|
||||
conn = TaskManagementDAOFactory.getConnection();
|
||||
stmt = conn.prepareStatement("UPDATE DYNAMIC_TASK_PROPERTIES SET PROPERTY_VALUE = ? " +
|
||||
"WHERE DYNAMIC_TASK_ID = ? AND PROPERTY_NAME = ?");
|
||||
|
||||
for (Map.Entry<String, String> entry : properties.entrySet()) {
|
||||
stmt.setString(1, entry.getValue());
|
||||
stmt.setInt(2, dynamicTaskId);
|
||||
stmt.setString(3, entry.getKey());
|
||||
stmt.addBatch();
|
||||
}
|
||||
stmt.executeBatch();
|
||||
} catch (SQLException e) {
|
||||
throw new TaskManagementDAOException
|
||||
("Error occurred while updating device properties to database.", e);
|
||||
} finally {
|
||||
TaskManagementDAOUtil.cleanupResources(stmt, null);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,110 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.dao.util;
|
||||
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
|
||||
import javax.naming.InitialContext;
|
||||
import javax.sql.DataSource;
|
||||
import java.sql.Connection;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Hashtable;
|
||||
import java.util.List;
|
||||
|
||||
public class TaskManagementDAOUtil {
|
||||
private static final Log log = LogFactory.getLog(TaskManagementDAOUtil.class);
|
||||
|
||||
public static void cleanupResources(Connection conn, PreparedStatement stmt, ResultSet rs) {
|
||||
if (rs != null) {
|
||||
try {
|
||||
rs.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while closing result set", e);
|
||||
}
|
||||
}
|
||||
if (stmt != null) {
|
||||
try {
|
||||
stmt.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while closing prepared statement", e);
|
||||
}
|
||||
}
|
||||
if (conn != null) {
|
||||
try {
|
||||
conn.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while closing database connection", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static void cleanupResources(PreparedStatement stmt, ResultSet rs) {
|
||||
if (rs != null) {
|
||||
try {
|
||||
rs.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while closing result set", e);
|
||||
}
|
||||
}
|
||||
if (stmt != null) {
|
||||
try {
|
||||
stmt.close();
|
||||
} catch (SQLException e) {
|
||||
log.warn("Error occurred while closing prepared statement", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static DataSource lookupDataSource(String dataSourceName,
|
||||
final Hashtable<Object, Object> jndiProperties) {
|
||||
try {
|
||||
if (jndiProperties == null || jndiProperties.isEmpty()) {
|
||||
return (DataSource) InitialContext.doLookup(dataSourceName);
|
||||
}
|
||||
final InitialContext context = new InitialContext(jndiProperties);
|
||||
return (DataSource) context.lookup(dataSourceName);
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("Error in looking up data source: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
public static DynamicTask loadDynamicTask(ResultSet rs) throws SQLException {
|
||||
DynamicTask dynamicTask = new DynamicTask();
|
||||
dynamicTask.setDynamicTaskId(rs.getInt("DYNAMIC_TASK_ID"));
|
||||
dynamicTask.setName(rs.getString("NAME"));
|
||||
dynamicTask.setCronExpression(rs.getString("CRON"));
|
||||
dynamicTask.setTaskClassName(rs.getString("TASK_CLASS_NAME"));
|
||||
dynamicTask.setEnabled(rs.getBoolean("IS_ENABLED"));
|
||||
dynamicTask.setTenantId(rs.getInt("TENANT_ID"));
|
||||
return dynamicTask;
|
||||
}
|
||||
|
||||
public static List<DynamicTask> loadDynamicTasks(ResultSet rs) throws SQLException {
|
||||
List<DynamicTask> dynamicTasks = new ArrayList<>();
|
||||
while (rs.next()) {
|
||||
dynamicTasks.add(loadDynamicTask(rs));
|
||||
}
|
||||
return dynamicTasks;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,62 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.internal;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService;
|
||||
import io.entgra.task.mgt.common.spi.TaskManagementService;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
|
||||
public class TaskManagerDataHolder {
|
||||
private TaskManagementService taskManagerService;
|
||||
private TaskService nTaskService;
|
||||
|
||||
private HeartBeatManagementService heartBeatService;
|
||||
|
||||
private static final TaskManagerDataHolder thisInstance = new TaskManagerDataHolder();
|
||||
|
||||
private TaskManagerDataHolder() {
|
||||
}
|
||||
|
||||
public static TaskManagerDataHolder getInstance() {
|
||||
return thisInstance;
|
||||
}
|
||||
|
||||
public TaskManagementService getTaskManagementService() {
|
||||
return taskManagerService;
|
||||
}
|
||||
|
||||
public void setTaskManagementService(TaskManagementService taskManagerService) {
|
||||
this.taskManagerService = taskManagerService;
|
||||
}
|
||||
|
||||
public TaskService getnTaskService() {
|
||||
return nTaskService;
|
||||
}
|
||||
|
||||
public void setnTaskService(TaskService nTaskService) {
|
||||
this.nTaskService = nTaskService;
|
||||
}
|
||||
|
||||
public HeartBeatManagementService getHeartBeatService() {
|
||||
return heartBeatService;
|
||||
}
|
||||
|
||||
public void setHeartBeatService(HeartBeatManagementService heartBeatService) {
|
||||
this.heartBeatService = heartBeatService;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,134 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.internal;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService;
|
||||
import io.entgra.task.mgt.core.config.TaskManagementConfig;
|
||||
import io.entgra.task.mgt.core.config.datasource.DataSourceConfig;
|
||||
import io.entgra.task.mgt.common.spi.TaskManagementService;
|
||||
import io.entgra.task.mgt.core.config.TaskConfigurationManager;
|
||||
import io.entgra.task.mgt.core.dao.common.TaskManagementDAOFactory;
|
||||
import io.entgra.task.mgt.core.service.TaskManagementServiceImpl;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.osgi.framework.BundleContext;
|
||||
import org.osgi.service.component.ComponentContext;
|
||||
import org.wso2.carbon.ndatasource.core.DataSourceService;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
|
||||
/**
|
||||
* @scr.component name="io.entgra.task.mgt.service" immediate="true"
|
||||
* @scr.reference name="datasource.service"
|
||||
* interface="org.wso2.carbon.ndatasource.core.DataSourceService"
|
||||
* cardinality="1..1"
|
||||
* policy="dynamic"
|
||||
* bind="setDataSourceService"
|
||||
* unbind="unsetDataSourceService"
|
||||
* @scr.reference name="app.mgt.ntask.component"
|
||||
* interface="org.wso2.carbon.ntask.core.service.TaskService"
|
||||
* cardinality="1..1"
|
||||
* policy="dynamic"
|
||||
* bind="setTaskService"
|
||||
* unbind="unsetTaskService"
|
||||
* @scr.reference name="entgra.heart.beat.service"
|
||||
* interface="io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService"
|
||||
* cardinality="0..1"
|
||||
* policy="dynamic"
|
||||
* bind="setHeartBeatService"
|
||||
* unbind="unsetHeartBeatService"
|
||||
*/
|
||||
public class TaskManagerServiceComponent {
|
||||
|
||||
private static final Log log = LogFactory.getLog(TaskManagerServiceComponent.class);
|
||||
|
||||
protected void activate(ComponentContext ctx) {
|
||||
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Activating Task Manager Service Component");
|
||||
}
|
||||
try {
|
||||
TaskManagementConfig taskManagementConfig = TaskConfigurationManager.getInstance()
|
||||
.getTaskManagementConfig();
|
||||
DataSourceConfig dataSourceConfig = taskManagementConfig.getTaskMgtConfigRepository()
|
||||
.getDataSourceConfig();
|
||||
TaskManagementDAOFactory.init(dataSourceConfig);
|
||||
BundleContext bundleContext = ctx.getBundleContext();
|
||||
TaskManagementService taskManagementService = new TaskManagementServiceImpl();
|
||||
taskManagementService.init();
|
||||
TaskManagerDataHolder.getInstance().setTaskManagementService(taskManagementService);
|
||||
bundleContext.registerService(TaskManagementService.class.getName()
|
||||
, taskManagementService, null);
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task Manager Service Component has been successfully activated");
|
||||
}
|
||||
} catch (Throwable e) {
|
||||
log.error("Error occurred while activating Task Manager Service Component", e);
|
||||
}
|
||||
}
|
||||
|
||||
protected void deactivate(ComponentContext ctx) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("De-activating Task Manager Service Component");
|
||||
}
|
||||
}
|
||||
|
||||
protected void setDataSourceService(DataSourceService dataSourceService) {
|
||||
/* This is to avoid Task Manager Service Component getting initialized before the underlying datasources
|
||||
are registered */
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Data source service set to Task Manager Service Component ");
|
||||
}
|
||||
}
|
||||
|
||||
protected void unsetDataSourceService(DataSourceService dataSourceService) {
|
||||
//do nothing
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
public void setTaskService(TaskService taskService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting the task service to Task Manager Service Component");
|
||||
}
|
||||
TaskManagerDataHolder.getInstance().setnTaskService(taskService);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetTaskService(TaskService taskService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Removing the task service from Task Manager Service Component");
|
||||
}
|
||||
TaskManagerDataHolder.getInstance().setnTaskService(null);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void setHeartBeatService(HeartBeatManagementService heartBeatService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting heart beat service to Task Manager Service Component");
|
||||
}
|
||||
TaskManagerDataHolder.getInstance().setHeartBeatService(heartBeatService);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetHeartBeatService(HeartBeatManagementService heartBeatManagementService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Removing heart beat service from Task Manager Service Component");
|
||||
}
|
||||
TaskManagerDataHolder.getInstance().setHeartBeatService(null);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,397 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (https://entgra.io) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.service;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.exception.HeartBeatManagementException;
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementDAOException;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementException;
|
||||
import io.entgra.task.mgt.common.exception.TaskNotFoundException;
|
||||
import io.entgra.task.mgt.common.exception.TransactionManagementException;
|
||||
import io.entgra.task.mgt.common.spi.TaskManagementService;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskDAO;
|
||||
import io.entgra.task.mgt.core.dao.DynamicTaskPropDAO;
|
||||
import io.entgra.task.mgt.core.dao.common.TaskManagementDAOFactory;
|
||||
import io.entgra.task.mgt.core.internal.TaskManagerDataHolder;
|
||||
import io.entgra.task.mgt.core.util.TaskManagementUtil;
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.ntask.common.TaskException;
|
||||
import org.wso2.carbon.ntask.core.TaskInfo;
|
||||
import org.wso2.carbon.ntask.core.TaskManager;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
public class TaskManagementServiceImpl implements TaskManagementService {
|
||||
private static final Log log = LogFactory.getLog(TaskManagementServiceImpl.class);
|
||||
|
||||
private final DynamicTaskDAO dynamicTaskDAO;
|
||||
|
||||
private final DynamicTaskPropDAO dynamicTaskPropDAO;
|
||||
private TaskManager taskManager;
|
||||
|
||||
public TaskManagementServiceImpl() {
|
||||
this.dynamicTaskDAO = TaskManagementDAOFactory.getDynamicTaskDAO();
|
||||
this.dynamicTaskPropDAO = TaskManagementDAOFactory.getDynamicTaskPropDAO();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void init() throws TaskManagementException {
|
||||
TaskService nTaskService = TaskManagerDataHolder.getInstance().getnTaskService();
|
||||
if (nTaskService == null) {
|
||||
String msg = "Unable to load TaskService, hence unable to schedule the task.";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg);
|
||||
}
|
||||
if (!nTaskService.getRegisteredTaskTypes().contains(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE)) {
|
||||
try {
|
||||
nTaskService.registerTaskType(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
this.taskManager = nTaskService.getTaskManager(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
} catch (TaskException e) {
|
||||
String msg = "Error occurred while registering task type ["
|
||||
+ TaskMgtConstants.Task.DYNAMIC_TASK_TYPE
|
||||
+ "], hence unable to schedule the task.";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void createTask(DynamicTask dynamicTask) throws TaskManagementException {
|
||||
String taskId;
|
||||
try {
|
||||
// add into the dynamic task tables
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
int dynamicTaskId = dynamicTaskDAO.addTask(dynamicTask);
|
||||
|
||||
Map<String, String> taskProperties = dynamicTask.getProperties();
|
||||
dynamicTaskPropDAO.addTaskProperties(dynamicTaskId, taskProperties);
|
||||
|
||||
// add into the ntask core
|
||||
taskId = TaskManagementUtil.generateTaskId(dynamicTaskId);
|
||||
|
||||
try {
|
||||
int serverHashIdx = TaskManagerDataHolder.getInstance().getHeartBeatService()
|
||||
.getServerCtxInfo().getLocalServerHashIdx();
|
||||
taskProperties.put(TaskMgtConstants.Task.LOCAL_HASH_INDEX, String.valueOf(serverHashIdx));
|
||||
taskProperties.put(TaskMgtConstants.Task.LOCAL_TASK_NAME, taskId);
|
||||
} catch (HeartBeatManagementException e) {
|
||||
String msg = "Unexpected exception when getting server hash index.";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
}
|
||||
|
||||
if (!isTaskExists(taskId)) {
|
||||
TaskInfo.TriggerInfo triggerInfo = new TaskInfo.TriggerInfo();
|
||||
triggerInfo.setCronExpression(dynamicTask.getCronExpression());
|
||||
TaskInfo taskInfo = new TaskInfo(taskId, dynamicTask.getTaskClassName(), taskProperties, triggerInfo);
|
||||
taskManager.registerTask(taskInfo);
|
||||
taskManager.scheduleTask(taskId);
|
||||
if (!dynamicTask.isEnabled()) {
|
||||
taskManager.pauseTask(taskId);
|
||||
}
|
||||
} else {
|
||||
String msg = "Task '" + taskId + "' is already exists in the ntask core "
|
||||
+ "Hence not creating another task for the same name.";
|
||||
log.error(msg);
|
||||
}
|
||||
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Failed to add dynamic task " + dynamicTask.getName();
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to add dynamic task";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TaskException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while scheduling task '" + dynamicTask.getName() + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void updateTask(int dynamicTaskId, DynamicTask dynamicTask) throws TaskManagementException
|
||||
, TaskNotFoundException {
|
||||
try {
|
||||
//Update dynamic task table
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
DynamicTask existingTask = dynamicTaskDAO.getDynamicTaskById(dynamicTaskId);
|
||||
|
||||
if (existingTask != null) {
|
||||
existingTask.setEnabled(dynamicTask.isEnabled());
|
||||
existingTask.setCronExpression(dynamicTask.getCronExpression());
|
||||
dynamicTaskDAO.updateDynamicTask(existingTask);
|
||||
if (!dynamicTask.getProperties().isEmpty()) {
|
||||
dynamicTaskPropDAO.updateDynamicTaskProps(dynamicTaskId, dynamicTask.getProperties());
|
||||
}
|
||||
} else {
|
||||
String msg = "Task '" + dynamicTaskId + "' is not exists in the dynamic task table.";
|
||||
log.error(msg);
|
||||
throw new TaskNotFoundException(msg);
|
||||
}
|
||||
|
||||
// Update task in the ntask core
|
||||
String taskId = TaskManagementUtil.generateTaskId(existingTask.getDynamicTaskId());
|
||||
if (isTaskExists(taskId)) {
|
||||
TaskInfo taskInfo = taskManager.getTask(taskId);
|
||||
if (!dynamicTask.getProperties().isEmpty()) {
|
||||
taskInfo.setProperties(dynamicTask.getProperties());
|
||||
}
|
||||
TaskInfo.TriggerInfo triggerInfo;
|
||||
if (taskInfo.getTriggerInfo() == null) {
|
||||
triggerInfo = new TaskInfo.TriggerInfo();
|
||||
} else {
|
||||
triggerInfo = taskInfo.getTriggerInfo();
|
||||
}
|
||||
triggerInfo.setCronExpression(dynamicTask.getCronExpression());
|
||||
taskInfo.setTriggerInfo(triggerInfo);
|
||||
taskManager.registerTask(taskInfo);
|
||||
taskManager.rescheduleTask(taskId);
|
||||
} else {
|
||||
String msg = "Task '" + taskId + "' is not exists in the n task core "
|
||||
+ "Hence cannot update the task.";
|
||||
log.error(msg);
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Failed to update dynamic task " + dynamicTask.getDynamicTaskId();
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to update dynamic task";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TaskException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while updating task '" + dynamicTask.getDynamicTaskId() + "'";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void toggleTask(int dynamicTaskId, boolean isEnabled) throws TaskManagementException
|
||||
, TaskNotFoundException {
|
||||
|
||||
try {
|
||||
//update dynamic task table
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
DynamicTask existingTask = dynamicTaskDAO.getDynamicTaskById(dynamicTaskId);
|
||||
if (existingTask != null) {
|
||||
existingTask.setEnabled(isEnabled);
|
||||
dynamicTaskDAO.updateDynamicTask(existingTask);
|
||||
} else {
|
||||
String msg = "Task '" + dynamicTaskId + "' is not exists.";
|
||||
log.error(msg);
|
||||
throw new TaskNotFoundException(msg);
|
||||
}
|
||||
|
||||
// Update task in the ntask core
|
||||
String taskId = TaskManagementUtil.generateTaskId(existingTask.getDynamicTaskId());
|
||||
if (isTaskExists(taskId)) {
|
||||
if (isEnabled) {
|
||||
taskManager.resumeTask(taskId);
|
||||
} else {
|
||||
taskManager.pauseTask(taskId);
|
||||
}
|
||||
} else {
|
||||
String msg = "Task '" + taskId + "' is not exists in the ntask core "
|
||||
+ "Hence cannot toggle the task in the ntask.";
|
||||
log.error(msg);
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Failed to toggle dynamic task " + dynamicTaskId;
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to toggle dynamic task";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TaskException e) {
|
||||
String msg = "Error occurred while toggling task '" + dynamicTaskId + "' to '" + isEnabled + "'";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deleteTask(int dynamicTaskId) throws TaskManagementException, TaskNotFoundException {
|
||||
// delete task from dynamic task table
|
||||
try {
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
DynamicTask existingTask = dynamicTaskDAO.getDynamicTaskById(dynamicTaskId);
|
||||
if (existingTask != null) {
|
||||
dynamicTaskDAO.deleteDynamicTask(dynamicTaskId);
|
||||
} else {
|
||||
String msg = "Task '" + dynamicTaskId + "' is not exists.";
|
||||
log.error(msg);
|
||||
throw new TaskNotFoundException(msg);
|
||||
}
|
||||
|
||||
String taskId = TaskManagementUtil.generateTaskId(existingTask.getDynamicTaskId());
|
||||
if (isTaskExists(taskId)) {
|
||||
taskManager.deleteTask(taskId);
|
||||
} else {
|
||||
String msg = "Task '" + taskId + "' is not exists in the ntask core "
|
||||
+ "Hence cannot delete from the ntask core.";
|
||||
log.error(msg);
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Failed to update dynamic task " + dynamicTaskId;
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to delete dynamic task";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TaskException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while retrieving task manager to delete task '" + dynamicTaskId + "'";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<DynamicTask> getAllDynamicTasks() throws TaskManagementException {
|
||||
List<DynamicTask> dynamicTasks;
|
||||
try {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Fetching the details of all dynamic tasks");
|
||||
}
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
dynamicTasks = dynamicTaskDAO.getAllDynamicTasks();
|
||||
if (dynamicTasks != null) {
|
||||
for (DynamicTask dynamicTask : dynamicTasks) {
|
||||
dynamicTask.setProperties(dynamicTaskPropDAO
|
||||
.getDynamicTaskProps(dynamicTask.getDynamicTaskId()));
|
||||
}
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while fetching all dynamic tasks";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to get all dynamic tasks";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
return dynamicTasks;
|
||||
}
|
||||
|
||||
@Override
|
||||
public DynamicTask getDynamicTaskById(int dynamicTaskId) throws TaskManagementException {
|
||||
DynamicTask dynamicTask;
|
||||
try {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Fetching the details of dynamic task '" + dynamicTaskId + "'");
|
||||
}
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
dynamicTask = dynamicTaskDAO.getDynamicTaskById(dynamicTaskId);
|
||||
if (dynamicTask != null) {
|
||||
dynamicTask.setProperties(dynamicTaskPropDAO.getDynamicTaskProps(dynamicTask.getDynamicTaskId()));
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while fetching dynamic task '" + dynamicTaskId + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to get dynamic task '" + dynamicTaskId + "'";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
return dynamicTask;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<DynamicTask> getActiveDynamicTasks() throws TaskManagementException {
|
||||
List<DynamicTask> dynamicTasks;
|
||||
try {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Fetching the details of all active dynamic tasks");
|
||||
}
|
||||
TaskManagementDAOFactory.beginTransaction();
|
||||
dynamicTasks = dynamicTaskDAO.getActiveDynamicTasks();
|
||||
if (dynamicTasks != null) {
|
||||
for (DynamicTask dynamicTask : dynamicTasks) {
|
||||
dynamicTask.setProperties(dynamicTaskPropDAO.getDynamicTaskProps(dynamicTask.getDynamicTaskId()));
|
||||
}
|
||||
}
|
||||
TaskManagementDAOFactory.commitTransaction();
|
||||
} catch (TaskManagementDAOException e) {
|
||||
TaskManagementDAOFactory.rollbackTransaction();
|
||||
String msg = "Error occurred while fetching all active dynamic tasks";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} catch (TransactionManagementException e) {
|
||||
String msg = "Failed to start/open transaction to get all active dynamic tasks";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
} finally {
|
||||
TaskManagementDAOFactory.closeConnection();
|
||||
}
|
||||
return dynamicTasks;
|
||||
}
|
||||
|
||||
// check whether task exist in the ntask core
|
||||
private boolean isTaskExists(String taskId) throws TaskManagementException, TaskException {
|
||||
if (StringUtils.isEmpty(taskId)) {
|
||||
String msg = "Task ID must not be null or empty.";
|
||||
log.error(msg);
|
||||
throw new TaskManagementException(msg);
|
||||
}
|
||||
List<TaskInfo> tasks = taskManager.getAllTasks();
|
||||
for (TaskInfo t : tasks) {
|
||||
if (taskId.equals(t.getName())) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,84 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.core.util;
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import io.entgra.server.bootup.heartbeat.beacon.exception.HeartBeatManagementException;
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementException;
|
||||
import io.entgra.task.mgt.core.internal.TaskManagerDataHolder;
|
||||
import org.apache.commons.codec.digest.DigestUtils;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.w3c.dom.Document;
|
||||
|
||||
import javax.xml.XMLConstants;
|
||||
import javax.xml.parsers.DocumentBuilder;
|
||||
import javax.xml.parsers.DocumentBuilderFactory;
|
||||
import java.io.File;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* Provides utility methods required by the task management bundle.
|
||||
*/
|
||||
public class TaskManagementUtil {
|
||||
|
||||
private static final Log log = LogFactory.getLog(TaskManagementUtil.class);
|
||||
|
||||
public static Document convertToDocument(File file) throws TaskManagementException {
|
||||
DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
|
||||
factory.setNamespaceAware(true);
|
||||
try {
|
||||
factory.setFeature(XMLConstants.FEATURE_SECURE_PROCESSING, true);
|
||||
factory.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
|
||||
DocumentBuilder docBuilder = factory.newDocumentBuilder();
|
||||
return docBuilder.parse(file);
|
||||
} catch (Exception e) {
|
||||
throw new TaskManagementException(
|
||||
"Error occurred while parsing file, while converting " +
|
||||
"to a org.w3c.dom.Document : " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
public static String generateTaskId(int dynamicTaskId) throws TaskManagementException {
|
||||
try {
|
||||
int serverHashIdx = TaskManagerDataHolder.getInstance().getHeartBeatService()
|
||||
.getServerCtxInfo().getLocalServerHashIdx();
|
||||
return generateTaskId(dynamicTaskId, serverHashIdx);
|
||||
} catch (HeartBeatManagementException e) {
|
||||
String msg = "Failed to generate task id for a dynamic task " + dynamicTaskId;
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
}
|
||||
}
|
||||
|
||||
public static String generateTaskId(int dynamicTaskId, int serverHashIdx) {
|
||||
return TaskMgtConstants.Task.DYNAMIC_TASK_TYPE + TaskMgtConstants.Task.NAME_SEPARATOR + dynamicTaskId
|
||||
+ TaskMgtConstants.Task.NAME_SEPARATOR + serverHashIdx;
|
||||
}
|
||||
|
||||
public static String generateTaskPropsMD5(Map<String, String> taskProperties) {
|
||||
taskProperties.remove(TaskMgtConstants.Task.TENANT_ID_PROP);
|
||||
taskProperties.remove(TaskMgtConstants.Task.LOCAL_HASH_INDEX);
|
||||
taskProperties.remove(TaskMgtConstants.Task.LOCAL_TASK_NAME);
|
||||
Gson gson = new Gson();
|
||||
String json = gson.toJson(taskProperties);
|
||||
return DigestUtils.md5Hex(json);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,41 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
|
||||
<parent>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>task-mgt</artifactId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>task-manager</artifactId>
|
||||
<name>Entgra IoT - Task Manager Component</name>
|
||||
<packaging>pom</packaging>
|
||||
<url>http://entgra.io</url>
|
||||
|
||||
<modules>
|
||||
<module>io.entgra.task.mgt.core</module>
|
||||
<module>io.entgra.task.mgt.common</module>
|
||||
</modules>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,140 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
|
||||
<parent>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>task-watcher</artifactId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>io.entgra.task.mgt.watcher</artifactId>
|
||||
<packaging>bundle</packaging>
|
||||
<name>Entgra IoT - Task Watcher </name>
|
||||
<description>Entgra IoT - Task Watcher</description>
|
||||
<url>http://entgra.io</url>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.felix</groupId>
|
||||
<artifactId>maven-scr-plugin</artifactId>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<groupId>org.apache.felix</groupId>
|
||||
<artifactId>maven-bundle-plugin</artifactId>
|
||||
<extensions>true</extensions>
|
||||
<configuration>
|
||||
<instructions>
|
||||
<Bundle-SymbolicName>${project.artifactId}</Bundle-SymbolicName>
|
||||
<Bundle-Name>${project.artifactId}</Bundle-Name>
|
||||
<Bundle-Version>${carbon.device.mgt.version}</Bundle-Version>
|
||||
<Bundle-Description>Task Watcher Bundle</Bundle-Description>
|
||||
<Private-Package>io.entgra.task.mgt.watcher.internal</Private-Package>
|
||||
<Import-Package>
|
||||
io.entgra.server.bootup.heartbeat.beacon.*,
|
||||
io.entgra.task.mgt.common.*,
|
||||
io.entgra.task.mgt.core.*,
|
||||
org.apache.commons.logging,
|
||||
org.osgi.framework.*;version="${imp.package.version.osgi.framework}",
|
||||
org.osgi.service.*;version="${imp.package.version.osgi.service}",
|
||||
org.wso2.carbon.context,
|
||||
org.wso2.carbon.core,
|
||||
org.wso2.carbon.device.mgt.common.*,
|
||||
org.wso2.carbon.ntask.*,
|
||||
org.wso2.carbon.user.api,
|
||||
org.wso2.carbon.user.core.*,
|
||||
</Import-Package>
|
||||
<Export-Package>
|
||||
io.entgra.task.mgt.watcher.*
|
||||
</Export-Package>
|
||||
</instructions>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.eclipse.osgi</groupId>
|
||||
<artifactId>org.eclipse.osgi</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.eclipse.osgi</groupId>
|
||||
<artifactId>org.eclipse.osgi.services</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.logging</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>io.entgra.task.mgt.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>io.entgra.task.mgt.common</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>org.wso2.carbon.device.mgt.common</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>org.wso2.carbon.device.mgt.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>io.entgra.server.bootup.heartbeat.beacon</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.user.core</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.user.api</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.wso2.carbon</groupId>
|
||||
<artifactId>org.wso2.carbon.utils</artifactId>
|
||||
<scope>provided</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,260 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.watcher;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.exception.HeartBeatManagementException;
|
||||
import io.entgra.task.mgt.common.bean.DynamicTask;
|
||||
import io.entgra.task.mgt.common.constant.TaskMgtConstants;
|
||||
import io.entgra.task.mgt.common.exception.TaskManagementException;
|
||||
import io.entgra.task.mgt.core.util.TaskManagementUtil;
|
||||
import io.entgra.task.mgt.watcher.internal.TaskWatcherDataHolder;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
||||
import org.wso2.carbon.core.ServerStartupObserver;
|
||||
import org.wso2.carbon.ntask.common.TaskException;
|
||||
import org.wso2.carbon.ntask.core.TaskInfo;
|
||||
import org.wso2.carbon.ntask.core.TaskManager;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
import org.wso2.carbon.user.api.Tenant;
|
||||
import org.wso2.carbon.user.api.UserStoreException;
|
||||
import org.wso2.carbon.user.core.service.RealmService;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.Timer;
|
||||
import java.util.TimerTask;
|
||||
|
||||
public class IoTSStartupHandler implements ServerStartupObserver {
|
||||
private static final Log log = LogFactory.getLog(IoTSStartupHandler.class);
|
||||
|
||||
@Override
|
||||
public void completingServerStartup() {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void completedServerStartup() {
|
||||
Timer timer = new Timer();
|
||||
timer.schedule(new TimerTask() {
|
||||
@Override
|
||||
public void run() {
|
||||
try {
|
||||
compareTasks();
|
||||
} catch (Exception e) {
|
||||
log.error("Error occurred when comparing tasks.", e);
|
||||
}
|
||||
}
|
||||
}, 200000, 300000);
|
||||
}
|
||||
|
||||
private void compareTasks() {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Comparing Tasks from carbon nTask manager and entgra task manager");
|
||||
}
|
||||
TaskService nTaskService = TaskWatcherDataHolder.getInstance().getnTaskService();
|
||||
if (nTaskService == null) {
|
||||
String msg = "Unable to load TaskService from the carbon nTask core";
|
||||
log.error(msg);
|
||||
return;
|
||||
}
|
||||
try {
|
||||
List<DynamicTask> dynamicTasks = TaskWatcherDataHolder.getInstance().getTaskManagementService()
|
||||
.getAllDynamicTasks();
|
||||
|
||||
scheduleMissingTasks(nTaskService, dynamicTasks);
|
||||
deleteObsoleteTasks(nTaskService, dynamicTasks);
|
||||
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task Comparison Completed and all tasks in current node are updated");
|
||||
}
|
||||
} catch (TaskException e) {
|
||||
String msg = "Error occurred while accessing carbon nTask manager.";
|
||||
log.error(msg, e);
|
||||
} catch (TaskManagementException e) {
|
||||
String msg = "Error occurred while retrieving all active tasks from entgra task manager";
|
||||
log.error(msg, e);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static void scheduleMissingTasks(TaskService nTaskService, List<DynamicTask> dynamicTasks)
|
||||
throws TaskException, TaskManagementException {
|
||||
Map<Integer, List<DynamicTask>> tenantedDynamicTasks = new HashMap<>();
|
||||
List<DynamicTask> dts;
|
||||
for (DynamicTask dt : dynamicTasks) {
|
||||
if (tenantedDynamicTasks.containsKey(dt.getTenantId())) {
|
||||
dts = tenantedDynamicTasks.get(dt.getTenantId());
|
||||
} else {
|
||||
dts = new ArrayList<>();
|
||||
}
|
||||
dts.add(dt);
|
||||
tenantedDynamicTasks.put(dt.getTenantId(), dts);
|
||||
}
|
||||
TaskManager taskManager;
|
||||
for (Integer tenantId : tenantedDynamicTasks.keySet()) {
|
||||
if (tenantId == -1) {
|
||||
log.warn("Found " + tenantedDynamicTasks.get(tenantId).size() +
|
||||
" invalid tasks without a valid tenant id.");
|
||||
continue;
|
||||
}
|
||||
PrivilegedCarbonContext.startTenantFlow();
|
||||
PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantId(tenantId, true);
|
||||
if (!nTaskService.getRegisteredTaskTypes().contains(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE)) {
|
||||
nTaskService.registerTaskType(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
}
|
||||
taskManager = nTaskService.getTaskManager(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
List<TaskInfo> tasks = taskManager.getAllTasks();
|
||||
// add or update task into nTask core
|
||||
for (DynamicTask dt : tenantedDynamicTasks.get(tenantId)) {
|
||||
String generatedTaskId = TaskManagementUtil.generateTaskId(dt.getDynamicTaskId());
|
||||
boolean isExist = false;
|
||||
for (TaskInfo taskInfo : tasks) {
|
||||
if (taskInfo.getName().equals(generatedTaskId)) {
|
||||
isExist = true;
|
||||
TaskInfo.TriggerInfo triggerInfo = taskInfo.getTriggerInfo();
|
||||
String dynamicTaskPropMD5 = TaskManagementUtil.generateTaskPropsMD5(dt.getProperties());
|
||||
String existingTaskPropMD5 = TaskManagementUtil.generateTaskPropsMD5(taskInfo.getProperties());
|
||||
if (!triggerInfo.getCronExpression().equals(dt.getCronExpression())
|
||||
|| !dynamicTaskPropMD5.equals(existingTaskPropMD5)) {
|
||||
triggerInfo.setCronExpression(dt.getCronExpression());
|
||||
taskInfo.setTriggerInfo(triggerInfo);
|
||||
taskInfo.setProperties(populateTaskProperties(tenantId, generatedTaskId, dt.getProperties()));
|
||||
taskManager.registerTask(taskInfo);
|
||||
taskManager.rescheduleTask(generatedTaskId);
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task - '" + generatedTaskId + "' updated according to the dynamic task table");
|
||||
}
|
||||
}
|
||||
if (dt.isEnabled()
|
||||
&& taskManager.getTaskState(generatedTaskId) == TaskManager.TaskState.PAUSED) {
|
||||
taskManager.resumeTask(generatedTaskId);
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task - '" + generatedTaskId + "' enabled according to the dynamic task table");
|
||||
}
|
||||
} else if (!dt.isEnabled()
|
||||
&& taskManager.getTaskState(generatedTaskId) != TaskManager.TaskState.PAUSED) {
|
||||
taskManager.pauseTask(generatedTaskId);
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task - '" + generatedTaskId + "' disabled according to the dynamic task table");
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!isExist) {
|
||||
TaskInfo.TriggerInfo triggerInfo = new TaskInfo.TriggerInfo();
|
||||
triggerInfo.setCronExpression(dt.getCronExpression());
|
||||
TaskInfo taskInfo = new TaskInfo(generatedTaskId, dt.getTaskClassName(),
|
||||
populateTaskProperties(tenantId, generatedTaskId, dt.getProperties()), triggerInfo);
|
||||
taskManager.registerTask(taskInfo);
|
||||
taskManager.scheduleTask(generatedTaskId);
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("New task -'" + generatedTaskId + "' created according to the dynamic task table");
|
||||
}
|
||||
}
|
||||
}
|
||||
PrivilegedCarbonContext.endTenantFlow();
|
||||
}
|
||||
}
|
||||
|
||||
private static Map<String, String> populateTaskProperties(int tenantId, String generatedTaskId,
|
||||
Map<String, String> taskProperties)
|
||||
throws TaskManagementException {
|
||||
try {
|
||||
int serverHashIdx = TaskWatcherDataHolder.getInstance().getHeartBeatService()
|
||||
.getServerCtxInfo().getLocalServerHashIdx();
|
||||
taskProperties.put(TaskMgtConstants.Task.LOCAL_HASH_INDEX, String.valueOf(serverHashIdx));
|
||||
taskProperties.put(TaskMgtConstants.Task.LOCAL_TASK_NAME, generatedTaskId);
|
||||
taskProperties.put(TaskMgtConstants.Task.TENANT_ID_PROP, String.valueOf(tenantId));
|
||||
return taskProperties;
|
||||
} catch (HeartBeatManagementException e) {
|
||||
String msg = "Unexpected exception when getting server hash index.";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
}
|
||||
}
|
||||
|
||||
private static void deleteObsoleteTasks(TaskService nTaskService, List<DynamicTask> dynamicTasks)
|
||||
throws TaskManagementException, TaskException {
|
||||
|
||||
List<Tenant> tenants = new ArrayList<>();
|
||||
try {
|
||||
RealmService realmService = TaskWatcherDataHolder.getInstance().getRealmService();
|
||||
Tenant[] tenantArray = realmService.getTenantManager().getAllTenants();
|
||||
if (tenantArray != null && tenantArray.length != 0) {
|
||||
tenants.addAll(Arrays.asList(tenantArray));
|
||||
}
|
||||
Tenant superTenant = new Tenant();
|
||||
superTenant.setId(-1234);
|
||||
tenants.add(superTenant);
|
||||
} catch (UserStoreException e) {
|
||||
String msg = "Unable to load tenants";
|
||||
log.error(msg, e);
|
||||
return;
|
||||
}
|
||||
|
||||
TaskManager taskManager;
|
||||
Set<Integer> hashIds;
|
||||
try {
|
||||
hashIds = TaskWatcherDataHolder.getInstance().getHeartBeatService().getActiveServers().keySet();
|
||||
} catch (HeartBeatManagementException e) {
|
||||
String msg = "Unexpected exception when getting hash indexes of active servers";
|
||||
log.error(msg, e);
|
||||
throw new TaskManagementException(msg, e);
|
||||
}
|
||||
|
||||
for (Tenant tenant : tenants) {
|
||||
PrivilegedCarbonContext.startTenantFlow();
|
||||
PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantId(tenant.getId(), true);
|
||||
if (!nTaskService.getRegisteredTaskTypes().contains(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE)) {
|
||||
nTaskService.registerTaskType(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
}
|
||||
taskManager = nTaskService.getTaskManager(TaskMgtConstants.Task.DYNAMIC_TASK_TYPE);
|
||||
List<TaskInfo> tasks = taskManager.getAllTasks();
|
||||
// Remove deleted items from the nTask core
|
||||
for (TaskInfo taskInfo : tasks) {
|
||||
boolean isExist = false;
|
||||
for (DynamicTask dt : dynamicTasks) {
|
||||
for (int hid : hashIds) {
|
||||
if (tenant.getId() == dt.getTenantId() &&
|
||||
taskInfo.getName().equals(TaskManagementUtil.generateTaskId(dt.getDynamicTaskId(), hid))) {
|
||||
isExist = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (isExist) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!isExist) {
|
||||
taskManager.deleteTask(taskInfo.getName());
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task '" + taskInfo.getName() + "' deleted according to the dynamic task table");
|
||||
}
|
||||
}
|
||||
}
|
||||
PrivilegedCarbonContext.endTenantFlow();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,73 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.watcher.internal;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService;
|
||||
import io.entgra.task.mgt.common.spi.TaskManagementService;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
import org.wso2.carbon.user.core.service.RealmService;
|
||||
|
||||
public class TaskWatcherDataHolder {
|
||||
private TaskManagementService taskManagerService;
|
||||
|
||||
private TaskService nTaskService;
|
||||
|
||||
private HeartBeatManagementService heartBeatService;
|
||||
private RealmService realmService;
|
||||
|
||||
private static final TaskWatcherDataHolder thisInstance = new TaskWatcherDataHolder();
|
||||
|
||||
private TaskWatcherDataHolder() {}
|
||||
|
||||
public static TaskWatcherDataHolder getInstance() {
|
||||
return thisInstance;
|
||||
}
|
||||
|
||||
public TaskManagementService getTaskManagementService() {
|
||||
return taskManagerService;
|
||||
}
|
||||
|
||||
public void setTaskManagementService(TaskManagementService taskManagerService) {
|
||||
this.taskManagerService = taskManagerService;
|
||||
}
|
||||
|
||||
public TaskService getnTaskService() {
|
||||
return nTaskService;
|
||||
}
|
||||
|
||||
public void setnTaskService(TaskService nTaskService) {
|
||||
this.nTaskService = nTaskService;
|
||||
}
|
||||
|
||||
public HeartBeatManagementService getHeartBeatService() {
|
||||
return heartBeatService;
|
||||
}
|
||||
|
||||
public void setHeartBeatService(HeartBeatManagementService heartBeatService) {
|
||||
this.heartBeatService = heartBeatService;
|
||||
}
|
||||
|
||||
public RealmService getRealmService() {
|
||||
return this.realmService;
|
||||
}
|
||||
|
||||
public void setRealmService(RealmService realmService) {
|
||||
this.realmService = realmService;
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,167 @@
|
||||
/*
|
||||
* Copyright (c) 2023, Entgra Pvt Ltd. (http://www.wso2.org) All Rights Reserved.
|
||||
*
|
||||
* Entgra Pvt Ltd. 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 io.entgra.task.mgt.watcher.internal;
|
||||
|
||||
import io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService;
|
||||
import io.entgra.task.mgt.common.spi.TaskManagementService;
|
||||
import io.entgra.task.mgt.core.config.TaskConfigurationManager;
|
||||
import io.entgra.task.mgt.core.config.TaskManagementConfig;
|
||||
import io.entgra.task.mgt.watcher.IoTSStartupHandler;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.osgi.framework.BundleContext;
|
||||
import org.osgi.service.component.ComponentContext;
|
||||
import org.wso2.carbon.core.ServerStartupObserver;
|
||||
import org.wso2.carbon.ntask.core.service.TaskService;
|
||||
import org.wso2.carbon.user.core.service.RealmService;
|
||||
|
||||
/**
|
||||
* @scr.component
|
||||
* name="io.entgra.task.mgt.watcher" immediate="true"
|
||||
* @scr.reference name="app.mgt.ntask.component"
|
||||
* interface="org.wso2.carbon.ntask.core.service.TaskService"
|
||||
* cardinality="1..1"
|
||||
* policy="dynamic"
|
||||
* bind="setTaskService"
|
||||
* unbind="unsetTaskService"
|
||||
* @scr.reference name="io.entgra.task.mgt.service"
|
||||
* interface="io.entgra.task.mgt.common.spi.TaskManagementService"
|
||||
* cardinality="1..1"
|
||||
* policy="dynamic"
|
||||
* bind="setTaskMgtService"
|
||||
* unbind="unsetTaskMgtService"
|
||||
* @scr.reference name="entgra.heart.beat.service"
|
||||
* interface="io.entgra.server.bootup.heartbeat.beacon.service.HeartBeatManagementService"
|
||||
* cardinality="0..1"
|
||||
* policy="dynamic"
|
||||
* bind="setHeartBeatService"
|
||||
* unbind="unsetHeartBeatService"
|
||||
* @scr.reference name="user.realmservice.default"
|
||||
* interface="org.wso2.carbon.user.core.service.RealmService"
|
||||
* cardinality="1..1"
|
||||
* policy="dynamic"
|
||||
* bind="setRealmService"
|
||||
* unbind="unsetRealmService"
|
||||
*/
|
||||
public class TaskWatcherServiceComponent {
|
||||
|
||||
private static final Log log = LogFactory.getLog(TaskWatcherServiceComponent.class);
|
||||
|
||||
protected void activate(ComponentContext ctx) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Activating Task Watcher Service Component");
|
||||
}
|
||||
try {
|
||||
TaskManagementConfig taskManagementConfig = TaskConfigurationManager.getInstance().getTaskManagementConfig();
|
||||
if (taskManagementConfig.isTaskWatcherEnabled()) {
|
||||
BundleContext bundleContext = ctx.getBundleContext();
|
||||
bundleContext.registerService(ServerStartupObserver.class.getName(), new IoTSStartupHandler(), null);
|
||||
} else {
|
||||
String msg = "Task watcher is not enabled in this environment hence wso2 carbon ntask will not " +
|
||||
"update according to the task manager ";
|
||||
log.debug(msg);
|
||||
}
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Task Watcher Service Component has been successfully activated");
|
||||
}
|
||||
} catch (Throwable e) {
|
||||
log.error("Error occurred while activating Task Watcher Service Component", e);
|
||||
}
|
||||
}
|
||||
|
||||
protected void deactivate(ComponentContext ctx) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("De-activating Task Watcher Service Component");
|
||||
}
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
public void setTaskService(TaskService taskService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting the task service to Task Watcher Service Component ");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setnTaskService(taskService);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetTaskService(TaskService taskService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Removing the task service from Task Watcher Service Component ");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setnTaskService(null);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void setTaskMgtService(TaskManagementService taskManagementService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting the task service to Task Watcher Service Component ");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setTaskManagementService(taskManagementService);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetTaskMgtService(TaskManagementService taskManagementService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Removing the task service from Task Watcher Service Component ");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setTaskManagementService(null);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void setHeartBeatService(HeartBeatManagementService heartBeatService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting heart beat service to Task Manager Service Component");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setHeartBeatService(heartBeatService);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetHeartBeatService(HeartBeatManagementService heartBeatManagementService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Removing heart beat service from Task Manager Service Component");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setHeartBeatService(null);
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets Realm Service.
|
||||
*
|
||||
* @param realmService An instance of RealmService
|
||||
*/
|
||||
@SuppressWarnings("unused")
|
||||
protected void setRealmService(RealmService realmService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Setting Realm Service");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setRealmService(realmService);
|
||||
}
|
||||
|
||||
/**
|
||||
* Unsets Realm Service.
|
||||
*
|
||||
* @param realmService An instance of RealmService
|
||||
*/
|
||||
@SuppressWarnings("unused")
|
||||
protected void unsetRealmService(RealmService realmService) {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Unsetting Realm Service");
|
||||
}
|
||||
TaskWatcherDataHolder.getInstance().setRealmService(null);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
~ Copyright (c) 2023, Entgra (Pvt) Ltd. (http://www.entgra.io) All Rights Reserved.
|
||||
~
|
||||
~ Entgra (Pvt) Ltd. 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.
|
||||
-->
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
|
||||
<parent>
|
||||
<groupId>org.wso2.carbon.devicemgt</groupId>
|
||||
<artifactId>task-mgt</artifactId>
|
||||
<version>5.0.20-SNAPSHOT</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<artifactId>task-watcher</artifactId>
|
||||
<name>Entgra IoT - Task Watcher Component</name>
|
||||
<packaging>pom</packaging>
|
||||
<url>http://entgra.io</url>
|
||||
|
||||
<modules>
|
||||
<module>io.entgra.task.mgt.watcher</module>
|
||||
</modules>
|
||||
|
||||
</project>
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue