package main import ( "flag" "fmt" "net/http" "os" "strings" "e.coding.net/rta/public/saasapi" "e.coding.net/rta/public/saasapi/pkg/saashttp" "google.golang.org/protobuf/encoding/protojson" ) const ( getIdsMax = 100 ) type readParams struct { cfg *Config appid string userids []string saasHttp *saashttp.SaasClient } func RunRead(args ...string) error { fs := flag.NewFlagSet("read", flag.ExitOnError) cfgFile := paramConfig(fs) appid := paramAppid(fs) userids := paramUserids(fs) if err := fs.Parse(args); err != nil { fmt.Fprintln(os.Stderr, "command line parse error", "err", err) return err } // 切割字符串 idsSlice := strings.Split(*userids, ",") if fs.NArg() > 0 || len(idsSlice) == 0 || (len(idsSlice) == 1 && idsSlice[0] == "") || len(idsSlice) > getIdsMax { fs.PrintDefaults() return nil } cfg, err := LoadConfigFile(*cfgFile) if err != nil { fmt.Fprintln(os.Stderr, "load config file error", "err", err) return err } readParams := readParams{ cfg: cfg, userids: idsSlice, appid: *appid, saasHttp: &saashttp.SaasClient{ Client: &http.Client{}, ApiUrls: &cfg.ApiUrls, Auth: &cfg.Auth, }, } return doRead(readParams) } func doRead(readParams readParams) error { saasReq := &saasapi.SaasReq{ Cmd: &saasapi.SaasReq_Read{ Read: &saasapi.Read{}, }, } if readParams.appid != "" { saasReq.UseridType = saasapi.UserIdType_OPENID saasReq.Appid = readParams.appid } saasReadItems := []*saasapi.ReadItem{} for _, userid := range readParams.userids { saasReadItems = append(saasReadItems, &saasapi.ReadItem{ Userid: userid, }) } saasReq.Cmd.(*saasapi.SaasReq_Read).Read.ReadItems = saasReadItems res, err := readParams.saasHttp.Read(saasReq) if err != nil { fmt.Fprintln(os.Stderr, "submit read error", "err", err) return err } if res.GetCode() != saasapi.ErrorCode_SUCC { fmt.Fprintln(os.Stderr, protojson.Format(res)) return err } else { fmt.Println(protojson.Format(res)) } return nil }