Skip to content

Commit

Permalink
[#11878] Account Request Indexing Script (#13076)
Browse files Browse the repository at this point in the history
* add script

* order commit

* update logic

* convert to batch

* remove print
  • Loading branch information
domoberzin committed Apr 22, 2024
1 parent 10e96cb commit 76db4cc
Showing 1 changed file with 81 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
package teammates.client.scripts;

import java.util.List;

import jakarta.persistence.TypedQuery;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.CriteriaQuery;
import jakarta.persistence.criteria.Root;

import teammates.client.util.BackDoor;
import teammates.client.util.ClientProperties;
import teammates.common.datatransfer.SqlDataBundle;
import teammates.common.exception.HttpRequestFailedException;
import teammates.common.util.HibernateUtil;
import teammates.storage.sqlentity.AccountRequest;

/**
* Script to trigger indexing for all account requests.
*/
public class UpdateAccountRequestIndexing {
public static void main(String[] args) {
UpdateAccountRequestIndexing updater = new UpdateAccountRequestIndexing();
updater.initSession();
updater.updateAccountRequestIndexing();
}

/**
* Initializes the Hibernate session.
*/
public void initSession() {
String connectionUrl = ClientProperties.SCRIPT_API_URL;
String username = ClientProperties.SCRIPT_API_NAME;
String password = ClientProperties.SCRIPT_API_PASSWORD;

HibernateUtil.buildSessionFactory(connectionUrl, username, password);
}

/**
* Updates the indexing for all account requests using the backdoor and batch processing.
*/
public void updateAccountRequestIndexing() {
int batchSize = 100;
int firstResult = 0;
HibernateUtil.beginTransaction();
CriteriaBuilder cb = HibernateUtil.getCriteriaBuilder();
CriteriaQuery<AccountRequest> cr = cb.createQuery(AccountRequest.class);
Root<AccountRequest> root = cr.from(AccountRequest.class);
cr.select(root);
TypedQuery<AccountRequest> query = HibernateUtil.createQuery(cr)
.setMaxResults(batchSize);

List<AccountRequest> accountRequests;

do {
query.setFirstResult(firstResult);
accountRequests = query.getResultList();
SqlDataBundle dataBundle = new SqlDataBundle();

for (AccountRequest accountRequest : accountRequests) {
dataBundle.accountRequests.put(accountRequest.getId().toString(), accountRequest);
}

insertDocs(dataBundle);
firstResult += batchSize;
} while (!accountRequests.isEmpty());

HibernateUtil.commitTransaction();
}

/**
* Inserts the document.
*/
public void insertDocs(SqlDataBundle dataBundle) {
try {
BackDoor.getInstance().putSqlDocuments(dataBundle);
} catch (HttpRequestFailedException e) {
System.out.println("Error occurred while inserting documents: " + e.getMessage());
e.printStackTrace();
}
}
}

0 comments on commit 76db4cc

Please sign in to comment.