Files
opensearch-pyd/samples/bulk/bulk-helpers.py
T
Daniel (dB.) DoubrovkineandGitHub bcfef113c4 Added samples, benchmarks and docs for nox format. (#556)
* Added samples for nox format.

Signed-off-by: dblock <[email protected]>

* Added space after #!/usr/bin/env python.

Signed-off-by: dblock <[email protected]>

* Added benchmarks and docs.

Signed-off-by: dblock <[email protected]>

---------

Signed-off-by: dblock <[email protected]>
2023-10-26 19:55:50 -07:00

57 lines
1.3 KiB
Python
Executable File

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# SPDX-License-Identifier: Apache-2.0
#
# The OpenSearch Contributors require contributions made to
# this file be licensed under the Apache-2.0 license or a
# compatible open source license.
#
# Modifications Copyright OpenSearch Contributors. See
# GitHub history for details.
import os
from opensearchpy import OpenSearch, helpers
# connect to an instance of OpenSearch
host = os.getenv("HOST", default="localhost")
port = int(os.getenv("PORT", 9200))
auth = (os.getenv("USERNAME", "admin"), os.getenv("PASSWORD", "admin"))
client = OpenSearch(
hosts=[{"host": host, "port": port}],
http_auth=auth,
use_ssl=True,
verify_certs=False,
ssl_show_warn=False,
)
# check whether an index exists
index_name = "my-index"
if not client.indices.exists(index_name):
client.indices.create(
index_name,
body={
"mappings": {
"properties": {
"value": {"type": "float"},
}
}
},
)
# index data
data = []
for i in range(100):
data.append({"_index": index_name, "_id": i, "value": i})
rc = helpers.bulk(client, data)
print(f"Bulk-inserted {rc[0]} items.")
# delete index
client.indices.delete(index=index_name)