Merge pull request #15 from nebula-contrib/add-log

log setting

core/src/main/scala/zio/nebula/GlobalSettings.scala

@@ -0,0 +1,32 @@
+package zio.nebula
+
+import zio._
+
+/**
+ * @author
+ *   梦境迷离
+ * @version 1.0,2023/9/14
+ */
+object GlobalSettings {
+
+  var printStatement: Boolean = false
+  var logLevel: LogLevel      = LogLevel.Warning
+
+  def printLog(stmt: String): ZIO[Any, Nothing, Unit] =
+    ZIO
+      .when(GlobalSettings.printStatement) {
+        if (GlobalSettings.logLevel == LogLevel.Error) {
+          ZIO.logError(s"Nebula client executed statement: $stmt")
+        } else if (GlobalSettings.logLevel == LogLevel.Warning) {
+          ZIO.logWarning(s"Nebula client executed statement: $stmt")
+        } else if (GlobalSettings.logLevel == LogLevel.Info) {
+          ZIO.logInfo(s"Nebula client executed statement: $stmt")
+        } else if (GlobalSettings.logLevel == LogLevel.Debug) {
+          ZIO.logDebug(s"Nebula client executed statement: $stmt")
+        } else if (GlobalSettings.logLevel == LogLevel.Trace) {
+          ZIO.logTrace(s"Nebula client executed statement: $stmt")
+        } else
+          ZIO.logFatal(s"Nebula client executed statement: $stmt").when(GlobalSettings.logLevel == LogLevel.Fatal).unit
+      }
+      .unit
+}

core/src/main/scala/zio/nebula/NebulaSessionClientLive.scala

@@ -12,7 +12,8 @@ import com.vesoft.nebula.client.graph._
 private[nebula] final class NebulaSessionClientLive(underlying: SessionPool) extends NebulaSessionClient {
 
   override def execute(stmt: String): Task[NebulaResultSet] =
-    ZIO.attempt(new NebulaResultSet(underlying.execute(stmt)))
+    GlobalSettings.printLog(stmt) *>
+      ZIO.attempt(new NebulaResultSet(underlying.execute(stmt)))
 
   override def idleSessionNum: Task[Int] = ZIO.attempt(underlying.getIdleSessionNums)