aboutsummaryrefslogtreecommitdiff
path: root/challenge-259/luca-ferrari/pljava/src/main/java/Task1.java
blob: de464e56c0d06062c843360140c4b4338419027f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package PWC259;

/**
 * PL/Java implementation for PWC 259
 * Task 1
 * See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-259>
 *
 *
 * To compile on the local machine:

 $ export JAVA_HOME=/usr/lib/jvm/java-17-openjdk-amd64/  # if not already set
 $ mvn clean build
 $ scp target/PWC259-1.jar  luca@rachel:/tmp


 * To install into PostgreSQL execute:

 select sqlj.install_jar( 'file:///tmp/PWC259-1.jar', 'PWC259', true );

 select pwc259.task2_pljava();

 and then to redeploy:

 select sqlj.replace_jar( 'file:///tmp/PWC259-1.jar', 'PWC259', true );

*/

import org.postgresql.pljava.*;
import org.postgresql.pljava.annotation.Function;
import static org.postgresql.pljava.annotation.Function.Effects.IMMUTABLE;
import static org.postgresql.pljava.annotation.Function.OnNullInput.RETURNS_NULL;

import java.util.*;
import java.util.stream.*;
import java.sql.SQLException;
import java.util.logging.*;
import java.sql.ResultSet;
import java.sql.Date;

public class Task1 {

    private final static Logger logger = Logger.getAnonymousLogger();

    @Function( schema = "pwc259",
	       onNullInput = RETURNS_NULL,
	       effects = IMMUTABLE )
    public static final java.sql.Date task1_pljava( Date startDay, int how_many, Date[] holidays ) throws SQLException {
	logger.log( Level.INFO, "Entering pwc259.task1_pljava" );

	Calendar day = Calendar.getInstance();
	day.setTime( startDay );

	while ( how_many > 0 ) {
	    day.add( Calendar.DAY_OF_YEAR, 1 );

	    while ( day.get( Calendar.DAY_OF_MONTH ) == Calendar.SUNDAY
		    || day.get( Calendar.DAY_OF_MONTH ) == Calendar.SATURDAY )
		day.add( Calendar.DAY_OF_YEAR, 1 );

	    if ( holidays != null )
		for ( Date skip : holidays )
		    if ( skip.equals( day.getTime() ) )
			day.add( Calendar.DAY_OF_YEAR, 1 );

	    how_many--;
	}

	return new java.sql.Date( day.getTimeInMillis() );
    }
}