chore: refactor crud repository methods (#38144)

## Description
As part of transaction support in PG, we are moving from using the jpa
methods for database operations. This PR is refactoring the code to use
custom repository class for DatasourceStorageStructureRepository from
the default CrudRepository.

## Automation

/ok-to-test tags="@tag.Datasource"

### 🔍 Cypress test results
<!-- This is an auto-generated comment: Cypress test results  -->
> [!TIP]
> 🟢 🟢 🟢 All cypress tests have passed! 🎉 🎉 🎉
> Workflow run:
<https://github.com/appsmithorg/appsmith/actions/runs/12310505479>
> Commit: f2268094b095c52557664da223936306d1bf5d48
> <a
href="https://internal.appsmith.com/app/cypress-dashboard/rundetails-65890b3c81d7400d08fa9ee5?branch=master&workflowId=12310505479&attempt=2"
target="_blank">Cypress dashboard</a>.
> Tags: `@tag.Datasource`
> Spec:
> <hr>Fri, 13 Dec 2024 07:34:38 UTC
<!-- end of auto-generated comment: Cypress test results  -->


## Communication
Should the DevRel and Marketing teams inform users about this change?
- [ ] Yes
- [ ] No


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Introduced a method to retrieve `DatasourceStorageStructure` based on
datasource and environment identifiers.
  
- **Bug Fixes**
- Simplified the structure of the
`DatasourceStorageStructureRepositoryCE` interface by removing an
outdated method.

- **Documentation**
- Added `@Repository` annotation to the
`DatasourceStorageStructureRepositoryCE` interface to indicate its role
as a Spring Data repository.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
Anagh Hegde 2024-12-13 14:40:02 +05:30 committed by GitHub
parent 88ec7c7c50
commit 5235304687
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 16 additions and 5 deletions

View File

@ -1,9 +1,12 @@
package com.appsmith.server.repositories.ce;
import com.appsmith.external.models.DatasourceStorageStructure;
import com.appsmith.external.models.DatasourceStructure;
import reactor.core.publisher.Mono;
public interface CustomDatasourceStorageStructureRepositoryCE {
Mono<Integer> updateStructure(String datasourceId, String environmentId, DatasourceStructure structure);
Mono<DatasourceStorageStructure> findByDatasourceIdAndEnvironmentId(String datasourceId, String environmentId);
}

View File

@ -3,6 +3,7 @@ package com.appsmith.server.repositories.ce;
import com.appsmith.external.models.DatasourceStorageStructure;
import com.appsmith.external.models.DatasourceStructure;
import com.appsmith.server.helpers.ce.bridge.Bridge;
import com.appsmith.server.helpers.ce.bridge.BridgeQuery;
import com.appsmith.server.repositories.BaseAppsmithRepositoryImpl;
import org.springframework.stereotype.Component;
import reactor.core.publisher.Mono;
@ -19,4 +20,13 @@ public class CustomDatasourceStorageStructureRepositoryCEImpl
.equal(DatasourceStorageStructure.Fields.environmentId, environmentId))
.updateFirst(Bridge.update().set(DatasourceStorageStructure.Fields.structure, structure));
}
@Override
public Mono<DatasourceStorageStructure> findByDatasourceIdAndEnvironmentId(
String datasourceId, String environmentId) {
final BridgeQuery<DatasourceStorageStructure> q = Bridge.<DatasourceStorageStructure>equal(
DatasourceStorageStructure.Fields.datasourceId, datasourceId)
.equal(DatasourceStorageStructure.Fields.environmentId, environmentId);
return queryBuilder().criteria(q).one();
}
}

View File

@ -3,10 +3,8 @@ package com.appsmith.server.repositories.ce;
import com.appsmith.external.models.DatasourceStorageStructure;
import com.appsmith.server.repositories.BaseRepository;
import com.appsmith.server.repositories.CustomDatasourceStorageStructureRepository;
import reactor.core.publisher.Mono;
import org.springframework.stereotype.Repository;
@Repository
public interface DatasourceStorageStructureRepositoryCE
extends BaseRepository<DatasourceStorageStructure, String>, CustomDatasourceStorageStructureRepository {
Mono<DatasourceStorageStructure> findByDatasourceIdAndEnvironmentId(String datasourceId, String environmentId);
}
extends BaseRepository<DatasourceStorageStructure, String>, CustomDatasourceStorageStructureRepository {}