Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2024-12-23 16:58:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

updates to telemetry, separate telemetry commands, use json data

Browse Source
This commit is contained in:
sawka 2023-01-23 12:54:32 -08:00
parent ea897bf53c
commit f8c675c3e7
6 changed files with 156 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/main-server.go
View File

@ -407,9 +407,18 @@ func test() error {
}
func sendTelemetryWrapper() {
defer func() {
r := recover()
if r == nil {
return
}
log.Printf("[error] in sendTelemetryWrapper: %v\n", r)
debug.PrintStack()
return
}()
ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
defer cancelFn()
err := pcloud.SendTelemetry(ctx)
err := pcloud.SendTelemetry(ctx, false)
if err != nil {
log.Printf("[error] sending telemetry: %v\n", err)
}
@ -498,7 +507,7 @@ func main() {
go func() {
log.Printf("PCLOUD_ENDPOINT=%s\n", pcloud.GetEndpoint())
time.Sleep(1 * time.Minute)
time.Sleep(30 * time.Second)
for {
sendTelemetryWrapper()
// send new telemetry every 8-hours

5
db/migrations/000002_activity.up.sql
View File

@ -1,10 +1,7 @@
CREATE TABLE activity (
day varchar(20) PRIMARY KEY,
uploaded boolean NOT NULL,
numcommands int NOT NULL,
activeminutes int NOT NULL,
fgminutes int NOT NULL,
openminutes int NOT NULL,
tdata json NOT NULL,
tzname varchar(50) NOT NULL,
tzoffset int NOT NULL,
clientversion varchar(20) NOT NULL,

103
pkg/cmdrunner/cmdrunner.go
View File

@ -53,7 +53,7 @@ var RemoteSetArgs = []string{"alias", "connectmode", "key", "password", "autoins
var WindowCmds = []string{"run", "comment", "cd", "cr", "clear", "sw", "reset", "signal"}
var NoHistCmds = []string{"_compgen", "line", "history", "_killserver"}
var GlobalCmds = []string{"session", "screen", "remote", "set", "client"}
var GlobalCmds = []string{"session", "screen", "remote", "set", "client", "telemetry"}
var SetVarNameMap map[string]string = map[string]string{
"tabcolor": "screen.tabcolor",
@ -61,7 +61,6 @@ var SetVarNameMap map[string]string = map[string]string{
"anchor": "sw.anchor",
"focus": "sw.focus",
"line": "sw.line",
"telemetry": "client.telemetry",
}
var SetVarScopes = []SetVarScope{
@ -159,9 +158,14 @@ func init() {
registerCmdFn("line:purge", LinePurgeCommand)
registerCmdFn("client", ClientCommand)
registerCmdFn("client:set", ClientSetCommand)
registerCmdFn("client:show", ClientShowCommand)
registerCmdFn("telemetry", TelemetryCommand)
registerCmdFn("telemetry:on", TelemetryOnCommand)
registerCmdFn("telemetry:off", TelemetryOffCommand)
registerCmdFn("telemetry:send", TelemetrySendCommand)
registerCmdFn("telemetry:show", TelemetryShowCommand)
registerCmdFn("history", HistoryCommand)
registerCmdFn("_killserver", KillServerCommand)
@ -2085,7 +2089,7 @@ func KillServerCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (s
}
func ClientCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
return nil, fmt.Errorf("/client requires a subcommand: %s", formatStrs([]string{"set", "show"}, "or", false))
return nil, fmt.Errorf("/client requires a subcommand: %s", formatStrs([]string{"show"}, "or", false))
}
func boolToStr(v bool, trueStr string, falseStr string) string {
@ -2113,45 +2117,94 @@ func ClientShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (s
return update, nil
}
func ClientSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return nil, fmt.Errorf("cannot retrieve client data: %v\n", err)
}
var varsUpdated []string
if pk.Kwargs["telemetry"] != "" {
noTelemetry := !resolveBool(pk.Kwargs["telemetry"], true)
if clientData.ClientOpts.NoTelemetry != noTelemetry {
func TelemetryCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
return nil, fmt.Errorf("/telemetry requires a subcommand: %s", formatStrs([]string{"show", "on", "off", "send"}, "or", false))
}
func setNoTelemetry(ctx context.Context, clientData *sstore.ClientData, noTelemetryVal bool) error {
clientOpts := clientData.ClientOpts
clientOpts.NoTelemetry = noTelemetry
err = sstore.SetClientOpts(ctx, clientOpts)
clientOpts.NoTelemetry = noTelemetryVal
err := sstore.SetClientOpts(ctx, clientOpts)
if err != nil {
return nil, fmt.Errorf("error trying to update client telemetry: %v", err)
return fmt.Errorf("error trying to update client telemetry: %v", err)
}
log.Printf("client telemetry setting updated to %v\n", !noTelemetry)
err = pcloud.SendNoTelemetryUpdate(ctx, noTelemetry)
log.Printf("client no-telemetry setting updated to %v\n", noTelemetryVal)
err = pcloud.SendNoTelemetryUpdate(ctx, clientOpts.NoTelemetry)
if err != nil {
// ignore error, just log
log.Printf("[error] sending no-telemetry update: %v\n", err)
log.Printf("note that telemetry update has still taken effect locally, and will be respected by the client\n")
}
} else {
log.Printf("client telemetry setting unchanged, is %v\n", !noTelemetry)
return nil
}
func TelemetryOnCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return nil, fmt.Errorf("cannot retrieve client data: %v\n", err)
}
varsUpdated = append(varsUpdated, "telemetry")
if !clientData.ClientOpts.NoTelemetry {
return sstore.InfoMsgUpdate("telemetry is already on"), nil
}
if len(varsUpdated) == 0 {
return nil, fmt.Errorf("/client:set no updates, can set %s", formatStrs([]string{"telemetry"}, "or", false))
err = setNoTelemetry(ctx, clientData, false)
if err != nil {
return nil, err
}
err = pcloud.SendTelemetry(ctx, false)
if err != nil {
// ignore error, but log
log.Printf("[error] sending telemetry update (in /telemetry:on): %v\n", err)
}
return sstore.InfoMsgUpdate("telemetry is now on"), nil
}
func TelemetryOffCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return nil, fmt.Errorf("cannot retrieve client data: %v\n", err)
}
if clientData.ClientOpts.NoTelemetry {
return sstore.InfoMsgUpdate("telemetry is already off"), nil
}
err = setNoTelemetry(ctx, clientData, true)
if err != nil {
return nil, err
}
return sstore.InfoMsgUpdate("telemetry is now off"), nil
}
func TelemetryShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return nil, fmt.Errorf("cannot retrieve client data: %v\n", err)
}
var buf bytes.Buffer
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "telemetry", boolToStr(clientData.ClientOpts.NoTelemetry, "off", "on")))
update := sstore.ModelUpdate{
Info: &sstore.InfoMsgType{
InfoMsg: fmt.Sprintf("client updated %s", formatStrs(varsUpdated, "and", false)),
TimeoutMs: 2000,
InfoTitle: fmt.Sprintf("telemetry info"),
InfoLines: splitLinesForInfo(buf.String()),
},
}
return update, nil
}
func TelemetrySendCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (sstore.UpdatePacket, error) {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return nil, fmt.Errorf("cannot retrieve client data: %v\n", err)
}
force := resolveBool(pk.Kwargs["force"], false)
if clientData.ClientOpts.NoTelemetry && !force {
return nil, fmt.Errorf("cannot send telemetry, telemetry is off. pass force=1 to force the send, or turn on telemetry with /telemetry:on")
}
err = pcloud.SendTelemetry(ctx, force)
if err != nil {
return nil, fmt.Errorf("failed to send telemetry: %v", err)
}
return sstore.InfoMsgUpdate("telemetry sent"), nil
}
func formatTermOpts(termOpts sstore.TermOpts) string {
if termOpts.Cols == 0 {
return "???"

12
pkg/pcloud/pcloud.go
View File

@ -28,6 +28,7 @@ type NoTelemetryInputType struct {
type TelemetryInputType struct {
UserId string `json:"userid"`
ClientId string `json:"clientid"`
CurDay string `json:"curday"`
Activity []*sstore.ActivityType `json:"activity"`
}
@ -87,12 +88,12 @@ func doRequest(req *http.Request, outputObj interface{}) (*http.Response, error)
return resp, nil
}
func SendTelemetry(ctx context.Context) error {
func SendTelemetry(ctx context.Context, force bool) error {
clientData, err := sstore.EnsureClientData(ctx)
if err != nil {
return fmt.Errorf("cannot retrieve client data: %v", err)
}
if clientData.ClientOpts.NoTelemetry {
if !force && clientData.ClientOpts.NoTelemetry {
return nil
}
activity, err := sstore.GetNonUploadedActivity(ctx)
@ -103,7 +104,8 @@ func SendTelemetry(ctx context.Context) error {
return nil
}
log.Printf("sending telemetry data\n")
input := TelemetryInputType{UserId: clientData.UserId, ClientId: clientData.ClientId, Activity: activity}
dayStr := sstore.GetCurDayStr()
input := TelemetryInputType{UserId: clientData.UserId, ClientId: clientData.ClientId, CurDay: dayStr, Activity: activity}
req, err := makePostReq(ctx, "/telemetry", input)
if err != nil {
return err
@ -112,6 +114,10 @@ func SendTelemetry(ctx context.Context) error {
if err != nil {
return err
}
err = sstore.MarkActivityAsUploaded(ctx, activity)
if err != nil {
return fmt.Errorf("error marking activity as uploaded: %v", err)
}
return nil
}

36
pkg/sstore/dbops.go
View File

@ -1838,28 +1838,37 @@ func GetRIsForWindow(ctx context.Context, sessionId string, windowId string) ([]
return rtn, nil
}
func UpdateCurrentActivity(ctx context.Context, update ActivityUpdate) error {
func GetCurDayStr() string {
now := time.Now()
dayStr := now.Format("2006-01-02")
return dayStr
}
func UpdateCurrentActivity(ctx context.Context, update ActivityUpdate) error {
now := time.Now()
dayStr := GetCurDayStr()
txErr := WithTx(ctx, func(tx *TxWrap) error {
query := `SELECT day FROM activity WHERE day = ?`
if !tx.Exists(query, dayStr) {
query = `INSERT INTO activity (day, uploaded, numcommands, fgminutes, activeminutes, openminutes, tzname, tzoffset, clientversion, clientarch)
VALUES (?, 0, 0, 0, 0, 0, ?, ?, ?, ?)`
var tdata TelemetryData
query := `SELECT tdata FROM activity WHERE day = ?`
found := tx.GetWrap(&tdata, query, dayStr)
if !found {
query = `INSERT INTO activity (day, uploaded, tdata, tzname, tzoffset, clientversion, clientarch)
VALUES (?, 0, ?, ?, ?, ?, ?)`
tzName, tzOffset := now.Zone()
if len(tzName) > MaxTzNameLen {
tzName = tzName[0:MaxTzNameLen]
}
tx.ExecWrap(query, dayStr, tzName, tzOffset, scbase.PromptVersion, scbase.ClientArch())
tx.ExecWrap(query, dayStr, tdata, tzName, tzOffset, scbase.PromptVersion, scbase.ClientArch())
}
tdata.NumCommands += update.NumCommands
tdata.FgMinutes += update.FgMinutes
tdata.ActiveMinutes += update.ActiveMinutes
tdata.OpenMinutes += update.OpenMinutes
query = `UPDATE activity
SET numcommands = numcommands + ?,
fgminutes = fgminutes + ?,
activeminutes = activeminutes + ?,
openminutes = openminutes + ?,
SET tdata = ?,
clientversion = ?
WHERE day = ?`
tx.ExecWrap(query, update.NumCommands, update.FgMinutes, update.ActiveMinutes, update.OpenMinutes, scbase.PromptVersion, dayStr)
tx.ExecWrap(query, tdata, scbase.PromptVersion, dayStr)
return nil
})
if txErr != nil {
@ -1881,10 +1890,15 @@ func GetNonUploadedActivity(ctx context.Context) ([]*ActivityType, error) {
return rtn, nil
}
// note, will not mark the current day as uploaded
func MarkActivityAsUploaded(ctx context.Context, activityArr []*ActivityType) error {
dayStr := GetCurDayStr()
txErr := WithTx(ctx, func(tx *TxWrap) error {
query := `UPDATE activity SET uploaded = 1 WHERE day = ?`
for _, activity := range activityArr {
if activity.Day == dayStr {
continue
}
tx.ExecWrap(query, activity.Day)
}
return nil

20
pkg/sstore/sstore.go
View File

@ -122,16 +122,28 @@ type ActivityUpdate struct {
type ActivityType struct {
Day string `json:"day"`
Uploaded bool `json:"-"`
NumCommands int `json:"numcommands"`
ActiveMinutes int `json:"activeminutes"`
FgMinutes int `json:"fgminutes"`
OpenMinutes int `json:"openminutes"`
TData TelemetryData `json:"tdata"`
TzName string `json:"tzname"`
TzOffset int `json:"tzoffset"`
ClientVersion string `json:"clientversion"`
ClientArch string `json:"clientarch"`
}
type TelemetryData struct {
NumCommands int `json:"numcommands"`
ActiveMinutes int `json:"activeminutes"`
FgMinutes int `json:"fgminutes"`
OpenMinutes int `json:"openminutes"`
}
func (tdata TelemetryData) Value() (driver.Value, error) {
return quickValueJson(tdata)
}
func (tdata *TelemetryData) Scan(val interface{}) error {
return quickScanJson(tdata, val)
}
type ClientOptsType struct {
NoTelemetry bool `json:"notelemetry,omitempty"`
}