Method QueryData

datasource/postgresql/postgresql.go:159–216

(ctx context.Context, query interface{})

}

func (p *PostgreSQL) QueryData(ctx context.Context, query interface{}) ([]models.DataResp, error) {
	postgresqlQueryParam := new(QueryParam)
	if err := mapstructure.Decode(query, postgresqlQueryParam); err != nil {
		return nil, err
	}

	postgresqlQueryParam.SQL = formatSQLDatabaseNameWithRegex(postgresqlQueryParam.SQL)
	if strings.Contains(postgresqlQueryParam.SQL, "$__") {
		var err error
		postgresqlQueryParam.SQL, err = macros.Macro(postgresqlQueryParam.SQL, postgresqlQueryParam.From, postgresqlQueryParam.To)
		if err != nil {
			return nil, err
		}
	}
	if postgresqlQueryParam.Database != "" {
		p.Shards[0].DB = postgresqlQueryParam.Database
	} else {
		db, err := parseDBName(postgresqlQueryParam.SQL)
		if err != nil {
			return nil, err
		}
		p.Shards[0].DB = db
	}

	timeout := p.Shards[0].Timeout
	if timeout == 0 {
		timeout = 60
	}
	timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(timeout)*time.Second)
	defer cancel()

	items, err := p.Shards[0].QueryTimeseries(timeoutCtx, &sqlbase.QueryParam{
		Sql: postgresqlQueryParam.SQL,
		Keys: types.Keys{
			ValueKey: postgresqlQueryParam.Keys.ValueKey,
			LabelKey: postgresqlQueryParam.Keys.LabelKey,
			TimeKey:  postgresqlQueryParam.Keys.TimeKey,
		},
	})

	if err != nil {
		logger.Warningf("query:%+v get data err:%v", postgresqlQueryParam, err)
		return []models.DataResp{}, err
	}
	data := make([]models.DataResp, 0)
	for i := range items {
		data = append(data, models.DataResp{
			Ref:    postgresqlQueryParam.Ref,
			Metric: items[i].Metric,
			Values: items[i].Values,
		})
	}

	// parse resp to time series data
	logger.Infof("req:%+v keys:%+v \n data:%v", postgresqlQueryParam, postgresqlQueryParam.Keys, data)

	return data, nil
}