mirror of
https://github.com/google/nomulus.git
synced 2025-06-27 06:44:51 +02:00
Add activity report generation code
This adds Bigquery API client code to generate the activity reports from our now standardSQL queries. The naming mirrors that of RDE (Staging generates the reports and uploads them to GCS). ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=164656344
This commit is contained in:
parent
2547313ef9
commit
477617eec9
25 changed files with 612 additions and 117 deletions
|
@ -56,15 +56,15 @@ public final class IcannReportingUploadAction implements Runnable {
|
|||
|
||||
private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
|
||||
|
||||
@Inject GcsUtils gcsUtils;
|
||||
@Inject IcannHttpReporter icannReporter;
|
||||
@Inject Retrier retrier;
|
||||
@Inject @Config("icannReportingBucket") String icannReportingBucket;
|
||||
@Inject @Parameter(RequestParameters.PARAM_TLD) String tld;
|
||||
@Inject @Parameter(IcannReportingModule.PARAM_YEAR_MONTH) String yearMonth;
|
||||
@Inject @Parameter(IcannReportingModule.PARAM_REPORT_TYPE) ReportType reportType;
|
||||
@Inject @Parameter(IcannReportingModule.PARAM_SUBDIR) Optional<String> subdir;
|
||||
@Inject @Config("icannReportingBucket") String reportingBucket;
|
||||
@Inject GcsUtils gcsUtils;
|
||||
@Inject IcannHttpReporter icannReporter;
|
||||
@Inject Response response;
|
||||
@Inject Retrier retrier;
|
||||
|
||||
@Inject
|
||||
IcannReportingUploadAction() {}
|
||||
|
@ -73,11 +73,9 @@ public final class IcannReportingUploadAction implements Runnable {
|
|||
public void run() {
|
||||
validateParams();
|
||||
String reportFilename = createFilename(tld, yearMonth, reportType);
|
||||
logger.infofmt("Reading ICANN report %s from bucket %s", reportFilename, reportingBucket);
|
||||
final GcsFilename gcsFilename =
|
||||
new GcsFilename(
|
||||
reportingBucket + "/" + (subdir.isPresent() ? subdir.get() : DEFAULT_SUBDIR),
|
||||
reportFilename);
|
||||
String reportBucketname = createReportingBucketName(icannReportingBucket, subdir, yearMonth);
|
||||
logger.infofmt("Reading ICANN report %s from bucket %s", reportFilename, reportBucketname);
|
||||
final GcsFilename gcsFilename = new GcsFilename(reportBucketname, reportFilename);
|
||||
checkState(
|
||||
gcsUtils.existsAndNotEmpty(gcsFilename),
|
||||
"ICANN report object %s in bucket %s not found",
|
||||
|
@ -103,12 +101,19 @@ public final class IcannReportingUploadAction implements Runnable {
|
|||
}
|
||||
|
||||
static String createFilename(String tld, String yearMonth, ReportType reportType) {
|
||||
// TODO(b/62585428): Change file naming date format to YYYY-MM for consistency with URL.
|
||||
// Report files use YYYYMM naming instead of standard YYYY-MM.
|
||||
// Report files use YYYYMM naming instead of standard YYYY-MM, per ICANN requirements.
|
||||
String fileYearMonth = yearMonth.substring(0, 4) + yearMonth.substring(5, 7);
|
||||
return String.format("%s-%s-%s.csv", tld, reportType.toString().toLowerCase(), fileYearMonth);
|
||||
}
|
||||
|
||||
static String createReportingBucketName(
|
||||
String reportingBucket, Optional<String> subdir, String yearMonth) {
|
||||
return subdir.isPresent()
|
||||
? String.format("%s/%s", reportingBucket, subdir.get())
|
||||
: String.format("%s/%s/%s", reportingBucket, DEFAULT_SUBDIR, yearMonth);
|
||||
|
||||
}
|
||||
|
||||
private void validateParams() {
|
||||
assertTldExists(tld);
|
||||
checkState(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue