public List getDocuments()

in java/core/src/main/java/com/aliyun/openservices/tablestore/agent/knowledge/KnowledgeStoreImpl.java [259:282]

23 lines of code
3 McCabe index (conditional complexity)


    public List<Document> getDocuments(List<String> documentIdList, String tenantId) {
        log.info("get documents, documentIdList:{}, tenantId:{}", documentIdList, tenantId);
        String newTenantId = checkEnableMultiTenantId(tenantId);
        List<PrimaryKey> pkList = new ArrayList<>(documentIdList.size());
        for (String docId : documentIdList) {
            PrimaryKeyBuilder primaryKeyBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
            primaryKeyBuilder.addPrimaryKeyColumn(Document.DOCUMENT_DOCUMENT_ID, PrimaryKeyValue.fromString(docId));
            primaryKeyBuilder.addPrimaryKeyColumn(Document.DOCUMENT_TENANT_ID, PrimaryKeyValue.fromString(newTenantId));
            PrimaryKey primaryKey = primaryKeyBuilder.build();
            pkList.add(primaryKey);
        }

        List<Document> documents = TablestoreHelper.batchGetDocuments(client, tableName, pkList, textField, embeddingField);
        if (documents.size() != documentIdList.size()) {
            throw Exceptions.runtime(
                "get documents failed, documentIdList[%s]:%s, tenantId:%s, documents:%s",
                documentIdList.size(),
                documentIdList,
                newTenantId,
                documents.size()
            );
        }
        return documents;
    }