| // Copyright 2015 The Vanadium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| // client handles pulling data from syncbase. To be fleshed out when discovery is added. |
| |
| package client |
| |
| import ( |
| "fmt" |
| "strconv" |
| "strings" |
| |
| "hearts/img/uistate" |
| "hearts/syncbase/util" |
| |
| "v.io/v23/context" |
| "v.io/v23/discovery" |
| wire "v.io/v23/services/syncbase/nosql" |
| "v.io/v23/syncbase/nosql" |
| ldiscovery "v.io/x/ref/lib/discovery" |
| "v.io/x/ref/lib/discovery/plugins/mdns" |
| "v.io/x/ref/lib/signals" |
| _ "v.io/x/ref/runtime/factories/generic" |
| ) |
| |
| // Searches for new syncgroups being advertised, sends found syncgroups to sgChan |
| func ScanForSG(sgChan chan []string, ctx *context.T, quit chan bool) { |
| mdns, err := mdns.New("") |
| if err != nil { |
| ctx.Fatalf("Plugin failed: %v", err) |
| } |
| ds := ldiscovery.NewWithPlugins([]ldiscovery.Plugin{mdns}) |
| fmt.Printf("Start scanning...\n") |
| ch, err := ds.Scan(ctx, "") |
| if err != nil { |
| ctx.Fatalf("Scan failed: %v", err) |
| } |
| instances := make(map[string]string) |
| loop: |
| for { |
| select { |
| case update := <-ch: |
| sgNames := GetSG(instances, update) |
| if sgNames != nil { |
| sgChan <- sgNames |
| } |
| case <-signals.ShutdownOnSignals(ctx): |
| break loop |
| case <-quit: |
| break loop |
| } |
| } |
| } |
| |
| // Returns the addresses of any discovered syncgroups that contain croupier game information |
| func GetSG(instances map[string]string, update discovery.Update) []string { |
| switch u := update.(type) { |
| case discovery.UpdateFound: |
| found := u.Value |
| instances[string(found.Service.InstanceId)] = found.Service.InstanceName |
| fmt.Printf("Discovered %q: Instance=%x, Interface=%q, Addrs=%v\n", found.Service.InstanceName, found.Service.InstanceId, found.Service.InterfaceName, found.Service.Addrs) |
| if found.Service.InterfaceName == util.CroupierInterface { |
| return []string{found.Service.Attrs["settings_sgname"], found.Service.Addrs[0]} |
| } |
| case discovery.UpdateLost: |
| lost := u.Value |
| name, ok := instances[string(lost.InstanceId)] |
| if !ok { |
| name = "unknown" |
| } |
| delete(instances, string(lost.InstanceId)) |
| fmt.Printf("Lost %q: Instance=%x\n", name, lost.InstanceId) |
| } |
| return nil |
| } |
| |
| // Returns a watchstream of the data in the table |
| func WatchData(tableName, prefix string, u *uistate.UIState) (nosql.WatchStream, error) { |
| db := u.Service.App(util.AppName).NoSQLDatabase(util.DbName, nil) |
| resumeMarker, err := db.GetResumeMarker(u.Ctx) |
| if err != nil { |
| fmt.Println("RESUMEMARKER ERR: ", err) |
| } |
| return db.Watch(u.Ctx, tableName, prefix, resumeMarker) |
| } |
| |
| // Joins gamelog syncgroup |
| func JoinLogSyncgroup(ch chan bool, logName string, u *uistate.UIState) { |
| fmt.Println("Joining gamelog syncgroup") |
| u.IsOwner = false |
| app := u.Service.App(util.AppName) |
| db := app.NoSQLDatabase(util.DbName, nil) |
| logSg := db.Syncgroup(logName) |
| myInfoJoiner := wire.SyncgroupMemberInfo{8, false} |
| _, err := logSg.Join(u.Ctx, myInfoJoiner) |
| if err != nil { |
| fmt.Println("SYNCGROUP JOIN ERROR: ", err) |
| ch <- false |
| } else { |
| fmt.Println("Syncgroup joined") |
| // Set UIState GameID |
| tmp := strings.Split(logName, "-") |
| gameID, _ := strconv.Atoi(tmp[len(tmp)-1]) |
| u.GameID = gameID |
| ch <- true |
| } |
| } |
| |
| // Joins player settings syncgroup |
| func JoinSettingsSyncgroup(ch chan bool, settingsName string, u *uistate.UIState) { |
| fmt.Println("Joining user settings syncgroup") |
| app := u.Service.App(util.AppName) |
| db := app.NoSQLDatabase(util.DbName, nil) |
| settingsSg := db.Syncgroup(settingsName) |
| myInfoJoiner := wire.SyncgroupMemberInfo{8, false} |
| _, err := settingsSg.Join(u.Ctx, myInfoJoiner) |
| if err != nil { |
| fmt.Println("SYNCGROUP JOIN ERROR: ", err) |
| ch <- false |
| } else { |
| fmt.Println("Syncgroup joined") |
| ch <- true |
| } |
| } |
| |
| func NumInSG(logName string, u *uistate.UIState) int { |
| app := u.Service.App(util.AppName) |
| db := app.NoSQLDatabase(util.DbName, nil) |
| sg := db.Syncgroup(logName) |
| members, err := sg.GetMembers(u.Ctx) |
| if err != nil { |
| fmt.Println(err) |
| } |
| return len(members) |
| } |