From c014ba93096010691468103e17e1d1f517d4522b Mon Sep 17 00:00:00 2001
From: Ryan King <ryan@theryanking.com>
Date: Tue, 30 Nov 2010 14:31:38 -0800
Subject: [PATCH] add dc and worker ids to ostrich

---
 .../scala/com/twitter/service/snowflake/SnowflakeServer.scala  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/scala/com/twitter/service/snowflake/SnowflakeServer.scala b/src/main/scala/com/twitter/service/snowflake/SnowflakeServer.scala
index 0403e97..c1afb79 100644
--- a/src/main/scala/com/twitter/service/snowflake/SnowflakeServer.scala
+++ b/src/main/scala/com/twitter/service/snowflake/SnowflakeServer.scala
@@ -17,6 +17,7 @@ import org.apache.zookeeper.{KeeperException, CreateMode, Watcher, WatchedEvent}
 import org.apache.zookeeper.KeeperException.NodeExistsException
 import scala.collection.mutable
 import java.net.InetAddress
+import com.twitter.ostrich.Stats
 
 case class Peer(hostname: String, port: Int)
 
@@ -34,6 +35,8 @@ object SnowflakeServer {
     new ZooKeeperClient(zkHostlist)
   }
 
+  Stats.makeGauge("datacenter_id") { datacenterId }
+  Stats.makeGauge("worker_id") { workerId }
   def shutdown(): Unit = {
     if (server != null) {
       log.info("Shutting down.")