Merge 261d8f6 into dbdff0b

proto/wantslist.proto

@@ -2,6 +2,10 @@ syntax = "proto3";
 
 package wantslist;
 
+message Config {
+  repeated WantList lists = 1;
+}
+
 message WantListEntry {
   int32 want = 1;
   int32 index = 2; 
@@ -13,7 +17,9 @@ message WantList {
 }
 
 
-message AddWantListRequest {}
+message AddWantListRequest {
+  WantList add = 1;
+}
 
 message AddWantListResponse {}
 
@@ -23,5 +29,5 @@ message GetWantListResponse {}
 
 service WantService {
   rpc AddWantList(AddWantListRequest) returns (AddWantListResponse) {};
-  rpc GetWantList(GetWantListRequest) returns (GetWantListResponse) {};
+  //rpc GetWantList(GetWantListRequest) returns (GetWantListResponse) {};
 }

wantslist.go

@@ -11,17 +11,25 @@ import (
 	"google.golang.org/grpc"
 
 	pbg "github.com/brotherlogic/goserver/proto"
+	pb "github.com/brotherlogic/wantslist/proto"
+)
+
+const (
+	// KEY - where the wantslists are stored
+	KEY = "/github.com/brotherlogic/wantslist/config"
 )
 
 //Server main server type
 type Server struct {
 	*goserver.GoServer
+	config *pb.Config
 }
 
 // Init builds the server
 func Init() *Server {
 	s := &Server{
 		&goserver.GoServer{},
+		&pb.Config{},
 	}
 	return s
 }
@@ -36,8 +44,29 @@ func (s *Server) ReportHealth() bool {
 	return true
 }
 
+func (s *Server) save(ctx context.Context) {
+	s.KSclient.Save(ctx, KEY, s.config)
+}
+
+func (s *Server) load(ctx context.Context) error {
+	config := &pb.Config{}
+	data, _, err := s.KSclient.Read(ctx, KEY, config)
+
+	if err != nil {
+		return err
+	}
+
+	s.config = data.(*pb.Config)
+	return nil
+}
+
 // Mote promotes/demotes this server
 func (s *Server) Mote(ctx context.Context, master bool) error {
+	if master {
+		err := s.load(ctx)
+		return err
+	}
+
 	return nil
 }