mirror of https://github.com/apache/cassandra
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from cqlshlib.displaying import MAGENTA
|
|
from datetime import datetime
|
|
import time
|
|
from cassandra.query import QueryTrace
|
|
|
|
|
|
def print_trace_session(shell, session, session_id):
|
|
trace = QueryTrace(session_id, session)
|
|
rows = fetch_trace_session(trace)
|
|
if not rows:
|
|
shell.printerr("Session %s wasn't found." % session_id)
|
|
return
|
|
names = ['activity', 'timestamp', 'source', 'source_elapsed']
|
|
|
|
formatted_names = map(shell.myformat_colname, names)
|
|
formatted_values = [map(shell.myformat_value, row) for row in rows]
|
|
|
|
shell.writeresult('')
|
|
shell.writeresult('Tracing session: ', color=MAGENTA, newline=False)
|
|
shell.writeresult(session_id)
|
|
shell.writeresult('')
|
|
shell.print_formatted_result(formatted_names, formatted_values)
|
|
shell.writeresult('')
|
|
|
|
|
|
def fetch_trace_session(trace):
|
|
trace.populate()
|
|
if not trace.events:
|
|
return []
|
|
|
|
rows = [[trace.request_type, str(datetime_from_utc_to_local(trace.started_at)), trace.coordinator, 0]]
|
|
|
|
# append main rows (from events table).
|
|
for event in trace.events:
|
|
rows.append(["%s [%s]" % (event.description, event.thread_name),
|
|
str(datetime_from_utc_to_local(event.datetime)),
|
|
event.source,
|
|
event.source_elapsed.microseconds])
|
|
# append footer row (from sessions table).
|
|
if trace.duration:
|
|
finished_at = (datetime_from_utc_to_local(trace.started_at) + trace.duration)
|
|
else:
|
|
finished_at = trace.duration = "--"
|
|
|
|
rows.append(['Request complete', str(finished_at), trace.coordinator, trace.duration.microseconds])
|
|
|
|
return rows
|
|
|
|
|
|
def datetime_from_utc_to_local(utc_datetime):
|
|
now_timestamp = time.time()
|
|
offset = datetime.fromtimestamp(now_timestamp) - datetime.utcfromtimestamp(now_timestamp)
|
|
return utc_datetime + offset
|
|
|