40 lines
938 B
Go
40 lines
938 B
Go
|
package sqlite
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"go.signoz.io/signoz/pkg/query-service/model"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (mds *ModelDaoSqlite) GetIngestionKeys(ctx context.Context) ([]model.IngestionKey, *model.ApiError) {
|
||
|
|
ingestion_keys := []model.IngestionKey{}
|
||
|
|
err := mds.db.Select(&ingestion_keys, `SELECT * FROM ingestion_keys`)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return nil, &model.ApiError{Typ: model.ErrorInternal, Err: err}
|
||
|
|
}
|
||
|
|
return ingestion_keys, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (mds *ModelDaoSqlite) InsertIngestionKey(ctx context.Context, ingestion_key *model.IngestionKey) *model.ApiError {
|
||
|
|
_, err := mds.db.ExecContext(ctx, `
|
||
|
|
INSERT INTO ingestion_keys (
|
||
|
|
ingestion_key,
|
||
|
|
name,
|
||
|
|
key_id,
|
||
|
|
ingestion_url,
|
||
|
|
data_region
|
||
|
|
) VALUES (
|
||
|
|
?,
|
||
|
|
?,
|
||
|
|
?,
|
||
|
|
?,
|
||
|
|
?
|
||
|
|
)`, ingestion_key.IngestionKey, ingestion_key.Name, ingestion_key.KeyId, ingestion_key.IngestionURL, ingestion_key.DataRegion)
|
||
|
|
if err != nil {
|
||
|
|
return &model.ApiError{Typ: model.ErrorInternal, Err: err}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|