micromegas 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

micromegas/flightsql/client.py

@@ -238,14 +238,13 @@ class FlightSQLClient:
                 print(row["time"], row["msg"])
 
     def materialize_partitions(
-        self, view_set_name, view_instance_id, begin, end, partition_delta_seconds
+        self, view_set_name, begin, end, partition_delta_seconds
     ):
         sql = """
           SELECT time, msg
-          FROM materialize_partitions('{view_set_name}', '{view_instance_id}', '{begin}', '{end}', {partition_delta_seconds})
+          FROM materialize_partitions('{view_set_name}', '{begin}', '{end}', {partition_delta_seconds})
         """.format(
             view_set_name=view_set_name,
-            view_instance_id=view_instance_id,
             begin=begin.isoformat(),
             end=end.isoformat(),
             partition_delta_seconds=partition_delta_seconds,
@@ -253,3 +252,56 @@ class FlightSQLClient:
         for rb in self.query_stream(sql):
             for index, row in rb.to_pandas().iterrows():
                 print(row["time"], row["msg"])
+
+    def find_process(self, process_id):
+        sql = """
+            SELECT *
+            FROM processes
+            WHERE process_id='{process_id}';
+            """.format(
+            process_id=process_id
+        )
+        return self.query(sql)
+
+    def query_streams(self, begin, end, limit, process_id=None, tag_filter=None):
+        conditions = []
+        if process_id is not None:
+            conditions.append("process_id='{process_id}'".format(process_id=process_id))
+        if tag_filter is not None:
+            conditions.append(
+                "(array_position(tags, '{tag}') is not NULL)".format(tag=tag_filter)
+            )
+        where = ""
+        if len(conditions) > 0:
+            where = "WHERE " + " AND ".join(conditions)
+        sql = """
+            SELECT *
+            FROM streams
+            {where}
+            LIMIT {limit};
+            """.format(
+            where=where, limit=limit
+        )
+        return self.query(sql, begin, end)
+
+    def query_blocks(self, begin, end, limit, stream_id):
+        sql = """
+            SELECT *
+            FROM blocks
+            WHERE stream_id='{stream_id}'
+            LIMIT {limit};
+            """.format(
+                limit=limit,stream_id=stream_id
+        )
+        return self.query(sql, begin, end)
+
+    def query_spans(self, begin, end, limit, stream_id):
+        sql = """
+            SELECT *
+            FROM view_instance('thread_spans', '{stream_id}')
+            LIMIT {limit};
+            """.format(
+                limit=limit,stream_id=stream_id
+        )
+        return self.query(sql, begin, end)
+    

{micromegas-0.3.0.dist-info → micromegas-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: micromegas
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python analytics client for https://github.com/madesroches/micromegas/
 Author: Marc-Antoine Desroches
 Author-email: madesroches@gmail.com
@@ -27,20 +27,22 @@ Python analytics client for https://github.com/madesroches/micromegas/
 
 ## Example usage
 
-Query the 2 most recent log entries from the analytics service
+Query the 2 most recent log entries from the flightsql service
 
 ```python
 import datetime
 import pandas as pd
 import micromegas
+import grpc
 
-BASE_URL = "http://localhost:8082/"
-client = micromegas.client.Client(BASE_URL)
+host_port = "localhost:50051"
+channel_cred = grpc.local_channel_credentials()
+client = micromegas.flightsql.client.FlightSQLClient(host_port, channel_cred)
 sql = """
 SELECT time, process_id, level, target, msg
 FROM log_entries
 WHERE level <= 4
-AND exe LIKE '%analytics%'
+AND exe LIKE '%flight%'
 ORDER BY time DESC
 LIMIT 2
 """

{micromegas-0.3.0.dist-info → micromegas-0.3.2.dist-info}/RECORD

@@ -7,7 +7,7 @@ micromegas/flightsql/Flight_pb2_grpc.py,sha256=mj9Nlla3wpf5xFAGhMZpLDX8vLBClWfZ4
 micromegas/flightsql/__init__.py,sha256=SRq2X59uKG-iuwFnSyT7wVfiTMSSryAAEWnQVjGWOM8,249
 micromegas/flightsql/arrow_flatbuffers.py,sha256=egpmS59sNFwWmtG2wMeE92MMIyGyZbsm9o24mUnY1MQ,100142
 micromegas/flightsql/arrow_ipc_reader.py,sha256=3rxyEgqo5100e0TT9ZKZxNe7lX6Lk0mS6yRoiIJtH6Q,3163
-micromegas/flightsql/client.py,sha256=kCjrq_mybF-SKhT5X-v5ey0AXukxcaLNvt4P4Ur2Wz8,9717
+micromegas/flightsql/client.py,sha256=v4igU4nHZdaqBAw5qDtDNl6UBYiLvVnCJk1qDeO8LjE,11241
 micromegas/flightsql/time.py,sha256=EH3SUEpFvY0lNMj9mOcvfUJuSgrQ3YX4aJnwteK2qhk,582
 micromegas/perfetto.py,sha256=yuIe5iKvca61aWMBQNziSGM-DHcOEsiobtKx2SsNQ3E,7829
 micromegas/request.py,sha256=NV0urom5P3_P2q94gX51hxW_Fnrp_DDRorsP3mUb5NM,941
@@ -217,6 +217,6 @@ micromegas/thirdparty/perfetto/protos/perfetto/trace/translation/translation_tab
 micromegas/thirdparty/perfetto/protos/perfetto/trace/trigger_pb2.py,sha256=We7Yi8o3cEcrSNxY1zLUUO6tEWnD36C2f3O_s8_qv0I,1435
 micromegas/thirdparty/perfetto/protos/perfetto/trace/ui_state_pb2.py,sha256=Af-SXwhroNhRXMrtw6e2eU1liCImMRxSdmkt_AuSHf8,1752
 micromegas/time.py,sha256=eD9fWF2UHxaf-92yd1X2SEgUcpKypqPsvjBosLdpnQA,1026
-micromegas-0.3.0.dist-info/METADATA,sha256=E4E7CwE4gz3c7WqCPZugw47d92FksajbSl2U0CBQfak,30555
-micromegas-0.3.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-micromegas-0.3.0.dist-info/RECORD,,
+micromegas-0.3.2.dist-info/METADATA,sha256=dA9ULkpE-NKW6fQ-yRGNO0sMXEv9AHsIZfxTcubwxaQ,30640
+micromegas-0.3.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+micromegas-0.3.2.dist-info/RECORD,,