Scala Zookeeper 访问 Demo

灰太狼 2022-03-26 02:22 170阅读 0赞

import scala.collection.mutable.ArrayBuffer
    import scala.collection.JavaConverters._
    
    import com.google.common.base.{ Objects, Strings}
    import org.apache.commons.compress.utils.Charsets
    import org.apache.curator.framework.{ CuratorFramework, CuratorFrameworkFactory}
    import org.apache.curator.retry.ExponentialBackoffRetry
    import org.apache.curator.utils.ZKPaths
    import org.slf4j.LoggerFactory
    
    object ZkTest { 
      private[this] val logger = LoggerFactory.getLogger(classOf[ZkTest])
      private[this] val zkAddress = "localhost:2181"
    
      private[this] var client: CuratorFramework = _
    
      def init(zkAddr: String = zkAddress): Unit = { 
        client = CuratorFrameworkFactory.newClient(
          zkAddr,
          new ExponentialBackoffRetry(1000, 3)
        )
    
        client.start()
      }
    
      def destroy(): Unit = { 
        if (client != null) { 
          client.close()
        }
    
      }
    
      /** * 给定一个ZK的路径，返回路径下所有< 目录, 值 > 的对象数组 * @param path * @return */
      def findProperties(path: String): Array[PropertyItem] = { 
        val properties = new ArrayBuffer[PropertyItem]()
    
        val stat = client.checkExists().forPath(path)
        if (null != stat) { 
          val children = client.getChildren.forPath(path).asScala
          val dataBuilder = client.getData
          if (children != null) for (child <- children) { 
            val propPath = ZKPaths.makePath(path, child)
            val item = PropertyItem(child, new String(dataBuilder.forPath(propPath), Charsets.UTF_8))
            properties += item
          }
        }
    
        properties.toArray
      }
    
      /** * 给定一个ZK路径，返回路径下所有子目录名字的数组 * @param path * @return */
      def listChildren(path: String): Array[String] = { 
        var children = Array.empty[String]
        val stat = client.checkExists.forPath(path)
    
        if (stat != null) { 
          val childrenBuilder = client.getChildren
          children = childrenBuilder.forPath(path).asScala.toArray
        }
        children
      }
    
      /** * 创建ZK节点，返回创建状态 * @param path * @param value * @return */
      def createProperty(path: String, value: String = null): Boolean = { 
        var result = false
    
        try { 
          val stat = client.checkExists.forPath(path)
          if (stat == null) { 
            val data = if (Strings.isNullOrEmpty(value)) { 
              Array.empty[Byte]
            } else { 
              value.getBytes(Charsets.UTF_8)
            }
            val opResult = client.create.creatingParentsIfNeeded.forPath(path, data)
            result = Objects.equal(path, opResult)
          }
        } catch { 
          case e: Exception =>
            logger.error(e.getMessage, e)
        }
        result
      }
    
      /** * 更新节点数据 * @param path * @param value * @return */
      def updateProperty(path: String, value: String): Boolean = { 
        var result = false
        try { 
          val stat = client.checkExists.forPath(path)
          if (stat != null) { 
            val opResult = client.setData().forPath(path, value.getBytes(Charsets.UTF_8))
            result = opResult != null
          }
          else { 
            val opResult = client.create.creatingParentsIfNeeded.forPath(path, value.getBytes(Charsets.UTF_8))
            result = Objects.equal(path, opResult)
          }
        } catch { 
          case e: Exception =>
            logger.error(e.getMessage, e)
        }
        result
      }
    
      /** * 删除节点 * @param path */
      def deleteProperty(path: String): Unit = { 
        try { 
          val stat = client.checkExists.forPath(path)
          if (stat != null) { 
            client.delete.deletingChildrenIfNeeded.forPath(path)
          }
        } catch { 
          case e: Exception =>
            logger.error(e.getMessage, e)
        }
      }
    
      /** * 获取节点数据 * @param path * @return */
      def getValue(path: String): String = { 
        var result: String = null
        try { 
          val stat = client.checkExists.forPath(path)
          if (stat != null) { 
            val data = client.getData.forPath(path)
            result = new String(data)
          } 
        } catch { 
          case e: Exception =>
            logger.error(e.getMessage, e)
        }
        result
      }
    
    }
    
    class ZkTest
    
    case class PropertyItem(name: String, value: String)