Set-Go_Perm.go

////////////////////////////////////////////////////////////////////////////
// Porgram: Set-Go_Perm.go
// Purpose: Demo saving the golang-set
// Authors: Tong Sun (c) 2016, All rights reserved
////////////////////////////////////////////////////////////////////////////

package main

import (
	"encoding/gob"
	"log"
	"os"

	set "github.com/deckarep/golang-set"
)

const persistName = "Set-Go_Perm.gob"

func main() {
	requiredClasses := set.NewSet()
	requiredClasses.Add("Cooking")
	requiredClasses.Add("English")
	requiredClasses.Add("Math")
	requiredClasses.Add("Biology")

	if err := SaveState(persistName, requiredClasses); err != nil {
		log.Fatal("SaveState failed:", err)
	}

}

func SaveState(persistName string, state interface{}) error {
	// create persistence file
	f, err := os.Create(persistName)
	if err != nil {
		return err
	}
	defer f.Close()

	// write persistemce file
	e := gob.NewEncoder(f)
	if err = e.Encode(state); err != nil {
		return err
	}
	return nil
}

func RestoreState(persistName string, state interface{}) error {
	// open persistence file
	f, err := os.Open(persistName)
	if err != nil {
		return err
	}
	defer f.Close()

	// read persistemce file
	e := gob.NewDecoder(f)
	if err = e.Decode(state); err != nil {
		return err
	}
	return nil
}