package info.tdoc.spring.example.tracer;

import io.opencensus.trace.Tracing;
import io.opencensus.trace.Tracer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.hibernate.resource.jdbc.spi.StatementInspector;
import io.opencensus.trace.AttributeValue;
import org.springframework.stereotype.Component;

@Component
public class SQLTracer implements StatementInspector {
    private static final Logger log = LoggerFactory.getLogger(SQLTracer.class);
    private static final Tracer tracer = Tracing.getTracer();

    public String inspect(String sql) {
        tracer.getCurrentSpan().putAttribute("sql.query", AttributeValue.stringAttributeValue(sql));
        return sql;
    }
}