forked from urdt/ussd
append sessionid to the gdbm file
This commit is contained in:
parent
0c360c0cc4
commit
fee51edade
@ -107,7 +107,7 @@ func main() {
|
||||
|
||||
fp := path.Join(dp, sessionId)
|
||||
|
||||
ussdHandlers, err := ussd.NewHandlers(fp, &st)
|
||||
ussdHandlers, err := ussd.NewHandlers(fp, &st,sessionId)
|
||||
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "handler setup failed with error: %v\n", err)
|
||||
|
@ -24,7 +24,7 @@ import (
|
||||
var (
|
||||
scriptDir = path.Join("services", "registration")
|
||||
translationDir = path.Join(scriptDir, "locale")
|
||||
dbFile = path.Join(scriptDir, "vise.gdbm")
|
||||
//dbFile = path.Join(scriptDir, "userdata.gdbm")
|
||||
)
|
||||
|
||||
const (
|
||||
@ -65,8 +65,9 @@ type Handlers struct {
|
||||
accountService server.AccountServiceInterface
|
||||
}
|
||||
|
||||
func NewHandlers(dir string, st *state.State) (*Handlers, error) {
|
||||
db, err := gdbm.Open(dbFile, gdbm.ModeWrcreat)
|
||||
func NewHandlers(dir string, st *state.State, sessionId string) (*Handlers, error) {
|
||||
filename := path.Join(scriptDir, sessionId+"_userdata.gdbm")
|
||||
db, err := gdbm.Open(filename, gdbm.ModeWrcreat)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
@ -425,9 +426,12 @@ func (h *Handlers) Authorize(ctx context.Context, sym string, input []byte) (res
|
||||
res.FlagReset = append(res.FlagReset, flags["flag_account_authorized"])
|
||||
return res, nil
|
||||
}
|
||||
} else {
|
||||
res.FlagSet = append(res.FlagSet, flags["flag_incorrect_pin"])
|
||||
res.FlagReset = append(res.FlagReset, flags["flag_account_authorized"])
|
||||
}
|
||||
} else if errors.Is(err, gdbm.ErrItemNotFound) {
|
||||
//PIN not set yet
|
||||
return res, err
|
||||
} else {
|
||||
return res, err
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user