use of vote ch in client

1 files changed, 8 insertions, 4 deletions

diff --git a/internal/vote/vote.go b/internal/vote/vote.go
index 137beed..8ad5a37 100644
--- a/internal/vote/vote.go
+++ b/internal/vote/vote.go
@@ -26,16 +26,20 @@ func New(ids []string) (Vote, error) {
 	return v, nil
 }
 
-func NewFromJSON(bytes []byte) (v Vote, err error) {
-	if err = json.Unmarshal(bytes, &v); err != nil {
+func NewFromJSON(jsonStr string) (v Vote, err error) {
+	if err = json.Unmarshal([]byte(jsonStr), &v); err != nil {
 		return
 	}
 	v.ExpiresAt = time.Now().Add(Expiry)
 	return
 }
 
-func (v Vote) ToJSON() ([]byte, error) {
-	return json.Marshal(v)
+func (v Vote) ToJSON() (string, error) {
+	bytes, err := json.Marshal(v)
+	if err != nil {
+		return "", err
+	}
+	return string(bytes), nil
 }
 
 func (v Vote) Expired() bool {