From 093b561221a11da52299b6286d781b8bc583feaa Mon Sep 17 00:00:00 2001 From: gacerioni Date: Wed, 31 Jul 2024 17:43:14 -0300 Subject: [PATCH] 0.0.4 --- app.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/app.py b/app.py index 36ea55b..5238641 100644 --- a/app.py +++ b/app.py @@ -10,10 +10,7 @@ # Carregar variáveis de ambiente load_dotenv() -# Constantes -REDIS_URL = os.getenv('GABS_REDIS_URL', "redis://localhost:6379") -OPENAI_API_KEY = os.getenv('GABS_OPENAI_API_KEY', 'sk_nadanadanada') - +# Constants REDIS_DATABASE_HOST = os.getenv('REDIS_DATABASE_HOST', '34.198.15.233') REDIS_DATABASE_PORT = os.getenv('REDIS_DATABASE_PORT', 19419) REDIS_DATABASE_PASSWORD = os.getenv('REDIS_DATABASE_PASSWORD', 'secret42') @@ -25,12 +22,11 @@ ORACLE_DATABASE_USER = os.getenv('ORACLE_DATABASE_USER', 'c##dbzuser') ORACLE_DATABASE_PASSWORD = os.getenv('ORACLE_DATABASE_PASSWORD', 'dbz') - app = Flask(__name__) # Initialize Oracle client with the updated library directory -#cx_Oracle.init_oracle_client(lib_dir="/root/something/instantclient_19_24") -cx_Oracle.init_oracle_client(lib_dir="/Users/gabriel.cerioni/instantclient") +cx_Oracle.init_oracle_client(lib_dir="/root/something/instantclient_19_24") +#cx_Oracle.init_oracle_client(lib_dir="/Users/gabriel.cerioni/instantclient") # Oracle connection setup oracle_dsn = cx_Oracle.makedsn(ORACLE_DATABASE_HOST, ORACLE_DATABASE_PORT, service_name=ORACLE_DATABASE_SERVICE_NAME) @@ -42,6 +38,7 @@ redis_search_client = Client(REDIS_INDEX_NAME, conn=redis_client) index_prefix = REDIS_INDEX_PREFIX + def create_redis_index(): """ Creates the index in Redis if it doesn't exist. @@ -54,8 +51,10 @@ def create_redis_index(): definition = IndexDefinition(prefix=[index_prefix], index_type=IndexType.JSON) redis_search_client.create_index(schema, definition=definition) + create_redis_index() + @app.route('/query', methods=['GET']) def query(): search_term = request.args.get('search_term', '').upper() # Convert search term to upper case @@ -73,7 +72,9 @@ def query(): oracle_latency_ms = (time.time() - oracle_start_time) * 1000 # Convert to milliseconds # Redis search remains unchanged - redis_query = Query(f'@NM_LOGR:{search_term}').return_fields("$.ID_ESTABELECIMENTO_SAUDE_PK", "$.NM_RAZ_SOC", "$.NM_FANTS", "$.NM_LOGR", "$.NM_NUMERO", "$.NM_BAIRRO", "$.CD_CEP") + redis_query = Query(f'@NM_LOGR:{search_term}').return_fields("$.ID_ESTABELECIMENTO_SAUDE_PK", "$.NM_RAZ_SOC", + "$.NM_FANTS", "$.NM_LOGR", "$.NM_NUMERO", + "$.NM_BAIRRO", "$.CD_CEP") redis_start_time = time.time() redis_results = redis_search_client.search(redis_query) redis_latency_ms = (time.time() - redis_start_time) * 1000 # Convert to milliseconds