// Save will store the organization to cached memory and disk. On successfull
// save the object will be automatically unlocked.
// NB: If error occure the unlocking of the object need to be done manually.
func (o *Organization) Save() (err error) {
return database.GetPureDB().Update(func(tx *bolt.Tx) (err error) {
// open the bucket
b := tx.Bucket([]byte(OrganizationBucketName))
// Checks if the bucket was opened, and creates a new one if not existing. Returns error on any other situation.
if b == nil {
return errors.New("Missing bucket")
}
buf := &bytes.Buffer{}
encoder := gob.NewEncoder(buf)
if err = encoder.Encode(o); err != nil {
return
}
data, err := ioutil.ReadAll(buf)
if err != nil {
return err
}
err = b.Put([]byte(o.Name), data)
if err != nil {
return err
}
o.Unlock()
return nil
})
}
// ListRegisteredOrganizations will list all the organizations registered in autograder.
func ListRegisteredOrganizations() (out []*Organization) {
out = make([]*Organization, 0)
keys := []string{}
database.GetPureDB().View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(OrganizationBucketName))
c := b.Cursor()
for k, _ := c.First(); k != nil; k, _ = c.Next() {
keys = append(keys, string(k))
}
return nil
})
for _, key := range keys {
org, err := NewOrganization(key, true)
if err != nil {
log.Println(err)
continue
}
out = append(out, org)
}
return
}
// ListAllMembers lists all members stored in the system.
func ListAllMembers() (out []*Member) {
out = make([]*Member, 0)
keys := []string{}
database.GetPureDB().View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(MemberBucketName))
if b == nil {
return errors.New("Unable to find bucket")
}
c := b.Cursor()
for k, _ := c.First(); k != nil; k, _ = c.Next() {
keys = append(keys, string(k))
}
return nil
})
for _, key := range keys {
m, err := NewMemberFromUsername(key, true)
if err != nil {
continue
}
out = append(out, m)
}
return
}
func (br *BuildResult) loadStoredData(lock bool) error {
return database.GetPureDB().View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(BuildBucketName))
if b == nil {
return errors.New("Bucket not found. Are you sure the bucket was registered correctly?")
}
data := b.Get([]byte(strconv.Itoa(br.ID)))
if data == nil {
return errors.New("No data in database")
}
buf := &bytes.Buffer{}
decoder := gob.NewDecoder(buf)
n, _ := buf.Write(data)
if n != len(data) {
return errors.New("Couldn't write all data to buffer while getting data from database. " + strconv.Itoa(n) + " != " + strconv.Itoa(len(data)))
}
err := decoder.Decode(br)
if err != nil {
return err
}
return nil
})
}
// Save will store the build results to the database.
func (br *BuildResult) Save() error {
return database.GetPureDB().Update(func(tx *bolt.Tx) error {
// open the bucket
b := tx.Bucket([]byte(BuildBucketName))
// Checks if the bucket was opened, and creates a new one if not existing. Returns error on any other situation.
if b == nil {
return errors.New("Missing bucket")
}
buf := &bytes.Buffer{}
encoder := gob.NewEncoder(buf)
if err := encoder.Encode(br); err != nil {
return err
}
data, err := ioutil.ReadAll(buf)
if err != nil {
return err
}
err = b.Put([]byte(strconv.Itoa(br.ID)), data)
if err != nil {
return err
}
return nil
})
}
// Delete will remove the group object.
func (g *Group) Delete() error {
for username := range g.Members {
user, err := NewMemberFromUsername(username, true)
if err != nil {
continue
}
courseopt := user.Courses[g.Course]
if courseopt.GroupNum == g.ID {
user.Lock()
courseopt.IsGroupMember = false
courseopt.GroupNum = 0
user.Courses[g.Course] = courseopt
if err = user.Save(); err != nil {
user.Unlock()
log.Println(err)
}
}
}
delete(InMemoryGroups, g.ID)
return database.GetPureDB().Update(func(tx *bolt.Tx) error {
return tx.Bucket([]byte(GroupsBucketName)).Delete([]byte(strconv.Itoa(g.ID)))
})
}
// loadData loads data from storage if it exists.
func (m *Member) loadStoredData(lock bool) (err error) {
err = database.GetPureDB().View(func(tx *bolt.Tx) error {
// locks the object directly in order to ensure consistent info from DB.
if lock {
m.Lock()
}
b := tx.Bucket([]byte(MemberBucketName))
if b == nil {
return errors.New("Bucket not found. Are you sure the bucket was registered correctly?")
}
data := b.Get([]byte(m.Username))
if data == nil {
return errors.New("No data in database")
}
buf := &bytes.Buffer{}
decoder := gob.NewDecoder(buf)
n, _ := buf.Write(data)
if n != len(data) {
return errors.New("Couldn't write all data to buffer while getting data from database. " + strconv.Itoa(n) + " != " + strconv.Itoa(len(data)))
}
err = decoder.Decode(m)
if err != nil {
return err
}
return nil
})
if err != nil {
if err.Error() == "No data in database" {
err = nil
}
}
if !m.accessToken.HasTokenInStore() {
m.accessToken.SetUsernameToTokenInStore(m.Username)
}
return
}
// GetNextBuildID will find the next available build ID.
// returns -1 on error
func GetNextBuildID() int {
nextid := -1
if err := database.GetPureDB().Update(func(tx *bolt.Tx) error {
// open the bucket
b := tx.Bucket([]byte(BuildBucketName))
// Checks if the bucket was opened, and creates a new one if not existing. Returns error on any other situation.
if b == nil {
return errors.New("Missing bucket")
}
var err error
data := b.Get([]byte(BuildLengthKey))
if data == nil {
nextid = 0
} else {
nextid, err = strconv.Atoi(string(data))
if err != nil {
return err
}
}
nextid++
data = []byte(strconv.Itoa(nextid))
err = b.Put([]byte(BuildLengthKey), data)
if err != nil {
return err
}
return nil
}); err != nil {
return -1
}
return nextid
}
// LoadStoredData fetches the organization data stored on disk or in cached memory.
func (o *Organization) LoadStoredData(lock bool) (err error) {
database.GetPureDB().View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(OrganizationBucketName))
if b == nil {
return errors.New("Bucket not found. Are you sure the bucket was registered correctly?")
}
data := b.Get([]byte(o.Name))
if data == nil {
return errors.New("No data in database")
}
buf := &bytes.Buffer{}
decoder := gob.NewDecoder(buf)
n, _ := buf.Write(data)
if n != len(data) {
return errors.New("Couldn't write all data to buffer while getting data from database. " + strconv.Itoa(n) + " != " + strconv.Itoa(len(data)))
}
err = decoder.Decode(o)
if err != nil {
return err
}
return nil
})
// locks the object directly in order to ensure consistent info from DB.
if lock {
o.Lock()
}
return
}