From 84810162f5e4f12e553349ff931592047dfab7bb Mon Sep 17 00:00:00 2001
From: Harshavardhana <harsha@minio.io>
Date: Wed, 1 Jul 2015 15:27:03 -0700
Subject: [PATCH] Add simple Version and GetSysInfo services

---
 pkg/server/router.go      |  2 +
 pkg/server/rpc/methods.go | 78 +++++++++++++++++++++++++++++++++++++--
 2 files changed, 77 insertions(+), 3 deletions(-)

diff --git a/pkg/server/router.go b/pkg/server/router.go
index 7c786f550..e9fcaaed1 100644
--- a/pkg/server/router.go
+++ b/pkg/server/router.go
@@ -112,6 +112,8 @@ func RPCHandler() http.Handler {
 	s := rpc.NewServer()
 	s.RegisterJSONCodec()
 	s.RegisterService(new(rpc.HelloService), "")
+	s.RegisterService(new(rpc.VersionService), "")
+	s.RegisterService(new(rpc.GetSysInfoService), "")
 	// add more services here
 	return registerRPC(router.NewRouter(), s)
 }
diff --git a/pkg/server/rpc/methods.go b/pkg/server/rpc/methods.go
index 4c654264b..ab7e9a590 100644
--- a/pkg/server/rpc/methods.go
+++ b/pkg/server/rpc/methods.go
@@ -16,7 +16,12 @@
 
 package rpc
 
-import "net/http"
+import (
+	"net/http"
+	"os"
+	"runtime"
+	"time"
+)
 
 // HelloArgs - hello args
 type HelloArgs struct {
@@ -28,11 +33,78 @@ type HelloReply struct {
 	Message string
 }
 
-// HelloService -
+// HelloService - hello service
 type HelloService struct{}
 
-// Say -
+// Say method
 func (h *HelloService) Say(r *http.Request, args *HelloArgs, reply *HelloReply) error {
 	reply.Message = "Hello, " + args.Who + "!"
 	return nil
 }
+
+// Args basic json RPC params
+type Args struct {
+	Request string
+}
+
+// VersionReply version reply
+type VersionReply struct {
+	Version   string `json:"version"`
+	BuildDate string `json:"build-date"`
+}
+
+// VersionService -
+type VersionService struct{}
+
+func getVersion() string {
+	return "0.0.1"
+}
+func getBuildDate() string {
+	return time.Now().UTC().Format(http.TimeFormat)
+}
+
+func setVersionReply(reply *VersionReply) {
+	reply.Version = getVersion()
+	reply.BuildDate = getBuildDate()
+	return
+}
+
+// Get method
+func (v *VersionService) Get(r *http.Request, args *Args, reply *VersionReply) error {
+	setVersionReply(reply)
+	return nil
+}
+
+// GetSysInfoService -
+type GetSysInfoService struct{}
+
+// GetSysInfoReply -
+type GetSysInfoReply struct {
+	Hostname  string           `json:"hostname"`
+	SysARCH   string           `json:"sys.arch"`
+	SysOS     string           `json:"sys.os"`
+	SysCPUS   int              `json:"sys.ncpus"`
+	Routines  int              `json:"goroutines"`
+	GOVersion string           `json:"goversion"`
+	MemStats  runtime.MemStats `json:"memstats"`
+}
+
+func setSysInfoReply(sis *GetSysInfoReply) error {
+	sis.SysARCH = runtime.GOARCH
+	sis.SysOS = runtime.GOOS
+	sis.SysCPUS = runtime.NumCPU()
+	sis.Routines = runtime.NumGoroutine()
+	sis.GOVersion = runtime.Version()
+	sis.Hostname, _ = os.Hostname()
+
+	var memStats runtime.MemStats
+	runtime.ReadMemStats(&memStats)
+	sis.MemStats = memStats
+
+	return nil
+}
+
+// Get method
+func (s *GetSysInfoService) Get(r *http.Request, args *Args, reply *GetSysInfoReply) error {
+	return setSysInfoReply(reply)
+}